Research Engineer - Scalable ML & Production Tools

Mistral; Greater London; On-site; GBP 100,000 - 140,000; Full time

Benefits offered by this job

Healthcare coverage

Parental leave

Retirement plans

Relocation support

Wellness programs

Meal and transportation allowances

Job summary

Mistral, a leader in full-stack AI solutions, seeks a Research Engineer – ML track to build scalable learning systems powering open-weight models. You will work with Research Scientists to move prototypes toward production-grade components and integrate with our enterprise API.

The role emphasizes hands-on deep learning, distributed training, and robust tooling in Python, with experience in PyTorch/JAX/TensorFlow and systems like SLURM and Kubernetes.

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 (DeepSpeed / FSDP / SLURM / K8s).
• Experience in deep learning, NLP or LLMs; CUDA or data-pipeline chops.
• Strong software-design instincts: testing, code review, CI/CD.
• Self-starter, low-ego, collaborative.

Responsibilities

• Accelerate researchers by taking on heavy parts of large-scale ML pipelines and building robust tools.
• Interface cutting-edge research with production: integrate checkpoints, streamline evaluation, 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

PyTorch

JAX

TensorFlow

Distributed training

Python

CI/CD

Deep learning

NLP/LLMs

Education

Master’s or PhD in Computer Science

Tools

DeepSpeed

FSDP

SLURM

Kubernetes

Job description

Mistral, a leader in full-stack AI solutions, seeks a Research Engineer – ML track to build scalable learning systems powering open-weight models. You will work with Research Scientists to move prototypes toward production-grade components and integrate with our enterprise API.

The role emphasizes hands-on deep learning, distributed training, and robust tooling in Python, with experience in PyTorch/JAX/TensorFlow and systems like SLURM and Kubernetes.