ML Research Engineer: Large-Scale Systems

Mistral AI; Paris; On-site; EUR 90,000 - 130,000; Full time

Benefits offered by this job

Healthcare coverage

Parental leave

Relocation support

Wellness programs

Job summary

Mistral is hiring a Research Engineer – ML track to build and optimize large-scale learning systems powering open-weight models, in collaboration with Research Scientists.

Join Platform or Embedded RE teams to advance training frameworks, data pipelines, and scalable code; you’ll apply PyTorch/JAX/TensorFlow and run distributed training across GPUs.

This role emphasizes strong software design, collaboration, and contributions from a world-spanning team.

Qualifications

• Master’s or PhD in CS or equivalent.
• 4+ years working on large-scale ML codebases.
• Hands-on with PyTorch, JAX or TensorFlow; distributed training.
• Experience in deep learning, NLP or LLMs; CUDA or data pipelines helpful.
• 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 and APIs.
• Conduct experiments on the latest deep-learning techniques and benchmarks.
• Design, implement and benchmark ML algorithms; write clear, efficient Python code.
• Deliver prototypes that become production-grade components for internal platforms.

Skills

Distributed training

Deep learning

NLP/LLMs

CI/CD

Collaborative

Education

Master’s or PhD in Computer Science

Tools

PyTorch

JAX

TensorFlow

DeepSpeed

SLURM

Kubernetes

CUDA

Job description

Mistral is hiring a Research Engineer – ML track to build and optimize large-scale learning systems powering open-weight models, in collaboration with Research Scientists.

Join Platform or Embedded RE teams to advance training frameworks, data pipelines, and scalable code; you’ll apply PyTorch/JAX/TensorFlow and run distributed training across GPUs.

This role emphasizes strong software design, collaboration, and contributions from a world-spanning team.