At Mary Technology, we're tackling one of the legal industry's biggest challenges : Fact Chaos — the overwhelming disorder that arises from managing mountains of unstructured information.We're building intuitive, scalable tools that help lawyers spend less time untangling facts and more time on high-value thinking. From visualising case timelines to extracting key insights from documents, we're reimagining what legal work looks like.We're backed by $2.7M in funding, growing over 30% MoM, and working with some of the largest law firms in Australia and 50+ mid-tier firms.If you're excited by AI, legal tech, and solving real-world problems at scale—we want to hear from you.
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.
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.
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.
What We Value
Agency – You move decisively through ambiguity and take full ownership.
Taste – You have a sharp eye for quality across research investments, product, and engineering.
Craft Obsession – You're energised by solving hard problems and continuously levelling up your work.
Why Join Us?
Career growth – Join early and grow with us in a fast-scaling, high-impact environment.
Direct impact – Work closely with founders, legal experts, and engineers to shape the future of legal AI.
Equity – Share in our success with meaningful ownership in the company.
Purposeful work – Build technology that transforms how the legal industry works from the ground up.
#J-18808-Ljbffr