Research Engineer - AI Systems & Production Pipelines

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

Job summary

Mistral is seeking a Research Engineer to transform customer requirements into reliable training and deployment workflows across CPT/SFT/RL/distillation. You will work end-to-end on model adaptation, evaluation, data, and infrastructure, bridging research experiments with production constraints.

This role sits in Applied Science and involves collaborating with scientists, engineers, product, and customer-facing teams to ensure Forge projects ship, are maintainable, and trusted by clients.

Qualifications

• Strong Python engineering skills and experience working in large codebases (testing, code review, CI, operational ownership).
• Hands-on experience with PyTorch, JAX, or similar.
• Strong systems and infrastructure fundamentals.
• Experience with LLM training or post-training: fine-tuning, RL, distillation, evaluation, and/or data pipelines.
• Excellent debugging skills in ambiguous systems (distributed jobs, data issues, quality regressions, infra failures).
• Clear communication with technical and non-technical stakeholders.
• High agency, low ego, and comfort in fast-moving, under-specified environments.

Responsibilities

• Build and improve post-training and evaluation workflows (CPT/SFT/RL/distillation), turning prototypes into repeatable Forge “recipes”.
• Develop tools and pipelines for synthetic data generation, data curation, training, evaluation, and deployment.
• Debug and harden large-scale ML systems: distributed training, scheduling/execution, checkpointing, observability, and reproducibility.
• Improve the Forge codebase via clear APIs, tests, documentation, and maintainable abstractions.
• Push the frontier of our RL training stack (e.g., high-throughput async rollout and scalable post-training systems at frontier-model scale).
• Make sure Forge deployment is seamless and adaptable to a diversity of clients (hardware access, software stack, cloud and on-premises, …).
• Partner with researchers and infrastructure engineers to translate bottlenecks into concrete system improvements.

Skills

Python

PyTorch

JAX

Systems

LLM training

Debugging

Tools

Kubernetes

SLURM

DeepSpeed

Job description

Mistral is seeking a Research Engineer to transform customer requirements into reliable training and deployment workflows across CPT/SFT/RL/distillation. You will work end-to-end on model adaptation, evaluation, data, and infrastructure, bridging research experiments with production constraints.

This role sits in Applied Science and involves collaborating with scientists, engineers, product, and customer-facing teams to ensure Forge projects ship, are maintainable, and trusted by clients.