Senior Engineer ML Engineer
You will validate the ML stack that connects accelerator hardware with trusted AI performance. You will run models, build benchmarking pipelines, create tests for low-level ML behavior, analyse numerical precision, quantisation, attention, distributed execution, and model subgraphs, and debug model and system performance in Linux.
Responsibilities
- Validate the ML stack across frameworks, models, and execution environments
- Run open source models
- Build automated benchmarking pipelines
- Create targeted tests for low-level ML behavior
- Test numerical precision, quantisation, attention mechanisms, distributed execution, and model subgraphs
- Identify regressions, correctness issues, and performance limits
- Analyse benchmark and experiment results
- Debug model behavior and system performance in Linux
Requirements
- Experience in machine learning or ML-adjacent software engineering
- Understanding of neural networks, training, inference, numerical precision, and performance trade-offs
- Experience with PyTorch, TensorFlow, JAX, Triton, or similar ML tools
- Strong Python skills
- Experience designing, running, and analysing ML benchmarks or model experiments
- Linux debugging skills
Benefits
- Unlimited annual leave
- Up to 5% matched pension
- Phantom equity
- Flexible work arrangements
- Free food
- On-site barista
- Health cash plan
- Income protection
- Life assurance
- Private medical insurance
- Dental plan