Staff Software Engineer, AI/ML
Own and scale our ML systems from the ground up. As a senior technical leader, you’ll architect, deploy, and optimise core models that transform messy scientific data into structured, usable knowledge powering real-world applications.
Responsibilities
- Design and lead implementation of ML architectures across NLP, search, and extraction tasks
- Optimise performance of large-scale pipelines, models, and systems in production
- Partner with product, infra, and domain teams to align ML efforts with user needs
- Set engineering standards for experimentation, testing, and deployment
- Mentor engineers and contribute to hiring, technical direction, and team development
- Identify and evaluate emerging techniques in LLMs, embeddings, and applied AI
technical requirements
- Deep understanding of NLP, transformers, embeddings, and distributed training
- Strong engineering background - ability to architect, debug, and optimise real systems
- Proficient in Python, PyTorch/TensorFlow, Docker, and cloud infra (AWS preferred)
- Experience with data pipelines, model serving, and CI/CD for ML
- (Bonus) Familiarity with vector search, RAG, or scientific/chemical domains
personal attributes
- Strong ownership mentality - comfortable setting direction and executing at pace
- Pragmatic technical leader who can balance speed, quality, and scalability
- Excellent communicator and collaborator across functions
- Passion for solving technically complex, high-value problems
- Low ego, high standards, and a bias toward action
what you’ll get
- High-leverage role owning foundational systems in a product-led AI company
- Work directly with experienced founders, engineers, and scientists
- Remote-first setup with flexibility and optional office in Sydney
- Accelerated scope and growth in a small, high-performing team
- Solve technical challenges with direct industry impact
- Competitive comp, autonomy, and the ability to shape core tech direction
NOTES
- We are a remote-first company.
- Preferred timezone: Australia.
Join our team
Oops! Something went wrong while submitting the form.
We teach
computers science.
OUR WHY
We are building chemistry-specific intelligence. Chemistry touches nearly every aspect of our lives and the problem we're trying to solve has significant scope and magnitude. It will directly affect people's lives for the better if we're successful. We are doing something that has never been done before - operating at a unique intersection of the physical and digital world.
