Research Engineer - AI Systems & Production Pipelines

Mistral AI; Greater London; On-site; GBP 110,000 - 170,000; Full time

Benefits offered by this job

Healthcare coverage

Parental leave

Relocation support

Wellness programs

Meal and transportation allowances

Job summary

Mistral AI in London seeks a Research Engineer for the Forge team to translate customer requirements into reliable training and deployment workflows across CPT/SFT/RL/distillation, evaluation, data, and infrastructure.

This role sits in Applied Science and collaborates with scientists, engineers, product, and customer-facing teams to ensure Forge projects ship, are maintainable, and trustworthy. Interview focus varies; strong Python, PyTorch/JAX, and infra fundamentals are valuable.

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

Strong Python engineering skills

PyTorch or JAX

Systems and infrastructure foundations

LLM training / post-training (fine-tun

Debugging distributed systems

Clear communication with stakeholders

High agency, low ego

Job description

Mistral AI in London seeks a Research Engineer for the Forge team to translate customer requirements into reliable training and deployment workflows across CPT/SFT/RL/distillation, evaluation, data, and infrastructure.

This role sits in Applied Science and collaborates with scientists, engineers, product, and customer-facing teams to ensure Forge projects ship, are maintainable, and trustworthy. Interview focus varies; strong Python, PyTorch/JAX, and infra fundamentals are valuable.