Job Description
About Us
n
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.
n
The Role
n
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.
n
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.
n
Your Impact
n
n
Fine-tune and iterate on transformer-based models — LLMs, embedding models, cross-encoders, and other components of modern LLM pipelines.
n
Build scalable, low-latency inference systems with monitoring, optimisation, and dynamic scaling.
n




Design synthetic data pipelines and Gym-style simulation environments to support model evaluation.
n
Lead experimental ML initiatives that drive product innovation and long-term value.
n
Help shape the foundational systems for Legal AGI.
n
n
What We're Looking For
n
n
Experience training and iterating on production transformer models and LLM pipelines.
n
Hands-on expertise with PyTorch, high-level ML frameworks, and inference engines (e.g. vLLM, SGLang, TensorRT).
n
Profound grasp of the forefront of ML and LLMs, from weight quantisation to the nuances between MLA and GQA.
n
Self-motivated learner who stays current with ML research and contributes to team knowledge sharing.
n
n
What We Value
n
n
Agency – You move decisively through ambiguity and take full ownership.
n
Taste – You have a sharp eye for quality across research investments, product, and engineering.
n
Craft Obsession – You're energised by solving hard problems and continuously levelling up your work.
n
n
Why Join Us?
n
n
Career growth – Join early and grow with us in a quick-scaling, high-impact environment.
n
Direct impact – Work closely with founders, legal experts, and engineers to shape the future of legal AI.
n
Equity – Share in our success with meaningful ownership in the company.
n
Purposeful work – Build technology that transforms how the legal industry works from the ground up.
n
#J-*****-Ljbffr

📌 Senior Software Engineer, Machine Learning (New South Wales)
🏢 Mary Technology
📍 New South Wales