AI Student Performance Enhancement

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 Develop sophisticated feature engineering pipelines for temporal student-content interaction data Develop deep learning models for student performance prediction and personalised content recommendation whilst handling data sparsity, conducting thorough experimentation to improve accuracy while documenting methodologies and results Optimise inference performance through compression techniques Contribute to developing and maintaining scalable ELT pipelines generating high quality data for ML workflows Lead experimental design and implementation Help design and execute online tests to validate hypotheses on ML capabilities Develop testing frameworks for streamlining hypothesis driven ML development Write production-ready, testable code, participate in code reviews, build reusable libraries and contribute to the development of data, feature and ML services Collaborate with MLOps engineers and partner with software engineers and product designers/managers to ensure that the ML solutions are user-centric and reliably integrated in production 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), approaches to handle data sparsity issues, and personalised recommendation systems. Expertise in data/feature engineering of interaction data for sequence modelling. Strong analytical skills with experience in statistical analysis, online controlled experiments and data visualisation. 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. Proficient in SQL and data manipulation techniques. 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. Additional Information 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 on employer's website This employer gathers applications via their own applicant tracking system. You will be redirected to an external application form.