Research Engineer, ML — Build Production-Grade AI Pipelines

Mistral AI; Greater London; On-site; GBP 120,000 - 180,000; Full time

Benefits offered by this job

Healthcare coverage

Relocation support

Wellness programs

Job summary

Mistral AI in the United Kingdom seeks a Research Engineer – ML track to build and optimise large-scale learning systems powering open-weight models. You’ll work with Research Scientists and choose to join either Platform RE Team or Embedded RE Team, delivering production-grade components for Le Chat and enterprise API.

Candidates typically hold a Master’s or PhD in Computer Science with 4+ years in ML codebases, and experience in PyTorch/JAX/TensorFlow and distributed training.

Qualifications

• Master’s or PhD in Computer Science or equivalent proven track record.
• 4+ years working on large-scale ML codebases.
• Hands-on with PyTorch, JAX or TensorFlow; comfortable with distributed training.
• Experience in deep learning, NLP or LLMs; CUDA or data-pipeline chops a bonus.
• Strong software-design instincts: testing, code review, CI/CD.
• Self-starter, low-ego, collaborative.

Responsibilities

• Accelerate researchers by handling heavy parts of large-scale ML pipelines and building robust tools.
• Interface cutting-edge research with production: integrate checkpoints, evaluate, and expose APIs.
• Conduct experiments on the latest deep-learning techniques (sparsified 70B+ runs, distributed training on thousands of GPUs).
• Design, implement and benchmark ML algorithms; write clear, efficient code in Python.
• Deliver prototypes that become production-grade components for Le Chat and our enterprise API.

Skills

Large-scale ML codebases

Distributed training

Research collaboration

Python programming

Education

Master’s or PhD in Computer Science

Tools

PyTorch

JAX

TensorFlow

DeepSpeed

FSDP

SLURM

Kubernetes

Python

Job description

Mistral AI in the United Kingdom seeks a Research Engineer – ML track to build and optimise large-scale learning systems powering open-weight models. You’ll work with Research Scientists and choose to join either Platform RE Team or Embedded RE Team, delivering production-grade components for Le Chat and enterprise API.

Candidates typically hold a Master’s or PhD in Computer Science with 4+ years in ML codebases, and experience in PyTorch/JAX/TensorFlow and distributed training.