Senior Software Engineer, Machine Learning
Posted on:
June 16, 2025
Job description
The Role
We’re looking for a Senior Machine Learning Software Engineer to drive forward both research and implementation. You’ll be working on cutting-edge models and infrastructure that transform how legal professionals interact with data.
You’ll fine-tune and deploy production-grade transformer models, build simulation environments for synthetic data, and design scalable inference systems — all while working closely with a high-performing, curious team.
Responsibilities
Your Impact
- Fine-tune and iterate on transformer-based models — LLMs, embedding models, cross-encoders, and other components of modern LLM pipelines.
- Build scalable, low-latency inference systems with monitoring, optimisation, and dynamic scaling.
- Design synthetic data pipelines and Gym-style simulation environments to support model evaluation.
- Lead experimental ML initiatives that drive product innovation and long-term value.
- Help shape the foundational systems for Legal AGI.
Job requirements
What We’re Looking For
- Experience training and iterating on production transformer models and LLM pipelines.
- Hands-on expertise with PyTorch, high-level ML frameworks, and inference engines (e.g. vLLM, SGLang, TensorRT).
- Profound grasp of the forefront of ML and LLMs, from weight quantisation to the nuances between MLA and GQA.
- Self-motivated learner who stays current with ML research and contributes to team knowledge sharing.
Apply now
When you apply, please mention you found the job on Aussie Intelligence. This encourages more tech companies to hire through our platform, creating more opportunities for you. Good luck with your application!
Apply now
