A

Senior Machine Learning Engineer

Atomi
Full-time
Remote
Australia
Join Atomi: Innovate, empower and inspire in the Edtech world.

Our mission is to help unlock the world’s potential by optimising the way we teach and learn. With a unique blend of technology and creativity, we bring learning to life, making complex concepts simple and accessible to students, educators, and parents alike.

We’re a team driven by our core values: a deep-rooted passion for learning, a relentless pursuit of quality at speed, humility in our growth, a commitment to pushing boundaries, empowering others fearlessly and championing the needs of our customers. Our culture is built on these values, ensuring we stay agile, innovative and empathetic to the evolving needs of the education community.

At Atomi, you’ll be encouraged to think creatively, work collaboratively and contribute to a mission where every piece of work is a step towards realising our collective potential in classrooms and beyond.

The opportunity

Atomi’s AI/ML team works to improve student outcomes and empower teachers with actionable AI powered insights. In particular, we’ve pioneered AI for instant marking and personalised feedback within interactive high school assessments, currently used in 700+ high schools across Australia and beyond.

In this role, you'll be leading the development of ML systems for student performance modelling and personalised recommendations and ensure that it continues to increase in capability, accuracy and operate reliably and safely under an increasing production scale.

What you'll do

    • Lead the development of ML systems for student performance prediction and personalised content recommendation
    • Design feature engineering pipelines for modelling student knowledge over time.
    • Development deep learning models capable to learn latent patterns from sparse interaction data
    • Optimise inference performance to meet production latency requirements
    • Implement monitoring and training workflows to combat data drift and maintain reliability
    • Contribute to ELT pipelines that generate quality data for ML workflows at scale
    • Design and execute online experiments to validate ML capabilities
    • Propose testing frameworks for hypothesis-driven ML development
    • Write production-ready code, participate in reviews, and build reusable components
    • Work with MLOps engineers to design, develop and deploy reliable ML systems
    • Collaborate with with product teams to deliver impactful user-centric features
    • Communicate findings to stakeholders and contribute to team knowledge sharing

What you'll bring

    • Expertise in deep learning (DL) techniques with proven experience building, optimising and deploying neural network architectures. Bonus for production experience with representation learning, sequence modelling (transformers, LSTMs), handling of data sparsity issues, and personalised recommendation systems.
    • Expertise in data/feature engineering from interaction streams.
    • Strong analytical skills with experience in statistical analysis and online controlled experiments. Bonus for familiarity with causal inference and bandit experiments.
    • Proficient in Python programming and version control systems (Git).
    • Hands-on experience using common DL frameworks (e.g. PyTorch, NVIDIA Merlin), ML orchestration tools (e.g. Metaflow) and relevant cloud MLOps services (e.g. AWS SageMaker).
    • Bonus for proficiency with modern data processing frameworks (e.g., Apache Spark, Beam, NVIDIA Merlin NVTabular) and experience with feature stores (e.g. Feast or SageMaker Feature Store) for managing ML features at scale.

Working at Atomi

    • Tailored flexibility: Enjoy work-life balance with hours that adapt to your needs, whether for university, family or personal time.
    • Generous leave: Experience unlimited paid leave options as a permanent team member.
    • Location Independence: Work from Sydney, interstate or wherever you find inspiration.
    • Growth and development: Use ongoing opportunities to improve your skills and expand your knowledge.
    • Holistic wellbeing: Benefit from a comprehensive employee assistance program subscription and additional wellbeing leave, supporting your best self.
    • Parental support: Enjoy 14 weeks of paid leave for new parents, with additional leave specifically for birthing parents.
    • Work with the best: We know you want to work alongside the best, so we work hard to hire only the best.
    • Regular social in-person and remote events: Including team sports competitions, trivia nights and themed annual events.
    • Atomi access: Enjoy free Atomi services for you and your family.
    • Onsite teams: Our Sydney office is ideally located in a central setting with cafes, shops and public transport nearby.
We’d like to understand who you are, your values and your professional journey. Please apply with your resume, a link to your portfolio and any insights you’d like to share.

We embrace diversity. Bringing together a variety of thoughts, backgrounds, and expressions is key to creating outstanding products and teams. We value many different skills and perspectives and how they can enhance Atomi and our culture. So, please apply even if you don’t meet every requirement listed.

Please note:
All applicants must hold Australian working rights.
Apply now
Share this job
Twitter Facebook Linkedin Email