Machine Learning Research Engineer

Requirements

• PhD in technical subject with major engineering component and exposure to AI/ML, or BSc, MSc and 2+ years of specific experience working on ML model development:
,
• Strong general engineering experience, as evidenced by exposure to one or more of:
,
• Software design / algorithms, especially for deep learning frameworks
,
• Modern ML frameworks such as JAX, PyTorch or TensorFlow
,
• Distributed systems and runtimes
,
• Compilers (e.g. XLA, Triton, CUDA, Pallas, …)
,
• Large scale model training and serving infrastructure
,
• Experience in navigating complex research codebases
,
• Databases and data processing pipelines
,
• Numerical methods, simulation, optimisation
,
• Strong fundamentals in mathematics, statistics, linear algebra
,
• Experience with the full ML research and development lifecycle
,
• Strong understanding of ML theory and applications
,
• Strong understanding...