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.

We’re looking for an experienced and passionate Senior Machine Learning Engineer to join our innovative AI/ML team to take on the challenge of improving student outcomes and empowering teachers with creative AI features. Our team is at the forefront of AI features like instant marking and personalised feedback within interactive assessments, currently used in over 700 high schools across Australia and beyond.

In this role, you’ll lead our efforts to advance our AI marking system while ensuring it continues to increase in capability and accuracy and operate reliably and safely to support our ambitious production and business goals.

What you'll do

• You'll lead efforts to optimise our AI Marking system, expanding its capabilities to a wider range of more complex assessment and question types.
• You'll design, develop, and deploy new machine learning solutions to improve how we mark and deliver personalised feedback to a growing number of users.
• Day-to-day you'll with data scientists to experiment with the latest techniques in NLP and AI and maximise the accuracy and robustness of the system.
• You'll help refine our sophisticated data strategies, including active learning pipelines, to enhance machine learning effectiveness.
• You'll leverage (and improve) existing monitoring processes to identify performance, scalability and reliability opportunities. Champion best practices in deployment and maintenance, ensuring our AI solutions are effective, resilient, and cost-efficient.
• You'll collaborate with data scientists, software engineers, domain experts and product managers to integrate AI solutions seamlessly, as well as to identify business and growth opportunities.
• You'll stay abreast of emerging trends and technologies in AI and machine learning, particularly in the context of educational technology, to drive innovation within our products.

What you'll bring to Atomi

• You'll have a strong foundation in machine learning, NLP and testing principles.
• You’ve got first-hand experience building, fine-tuning and deploying deep learning models on the cloud. (Bonus points if you’re familiar with LLMs, Parameter-Efficient Fine-Tuning, and NLP behavioural/adversarial testing.)
• You have proven experience with data processing, storage solutions, and implementing scalable data pipelines.
• You’re proficient in software development best practices, including version control with Git, continuous integration and deployment pipelines.
• You have a strong understanding of computer science/engineering fundamentals and first principles covering system design, data structures, architecture, and design patterns.
• You have expertise with Python coding, common tools in the ML ecosystem (e.g. PyTorch) and cloud services (e.g. AWS). (Bonus points if you’re also familiar with SQL, Pydantic, FastAPI, the OpenAI SDK and a ML pipeline tool (e.g. Metaflow)
• You have the ability to tackle complex problems, dealing with both the intricacies of language processing and the challenges of applying AI in real-world settings.
• You'll be enthusiastic working in an evolving scale-up environment
• You are an excellent communicator with all types of stakeholders across the business

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 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.