AI Research Engineer - ML Engineering
Core
Architecting and implementing scalable ML platforms and tools for autonomous decision-making in defence, focusing on distributed training and deep learning frameworks.
Role type
Senior IC machine-learning engineer (systems & infrastructure)
Builds
Scalable distributed training infrastructure and deep learning frameworks for defence AI applications
Domain
Defence / Machine Learning / Distributed Systems
Deliverable
production ML models
Required skills
Python, PyTorch, JAX, TensorFlow, distributed training, custom layer development, data strategy, GPU optimization
Preferred skills
NCCL, MPI, Slurm, Kubernetes, Ray, GPU architecture, large-scale multi-modal datasets
Technologies
PyTorch, JAX, TensorFlow, NCCL, MPI, Slurm, Kubernetes, Ray
Responsibilities
Extend deep learning frameworks for efficiency and ease of use; Scale infrastructure and tooling for faster distributed training; Design data strategy for large-scale datasets and efficient GPU utilization
Seniority
Senior, hands-on IC