Research Engineer: End-to-End AI Training & Deployment

Mistral; Singapore; On-site; SGD 120,000 - 180,000; Full time

Job summary

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

You’ll work end-to-end with scientists, engineers, product, and customer-facing teams to ensure Forge projects ship, are maintainable, and trusted by clients. This Applied Science role spans algorithms, infrastructure, evals, and data.

Qualifications

• Strong Python engineering skills and experience with large codebases (testing, CI, ownership).
• Hands-on experience with PyTorch, JAX, or similar frameworks.
• Strong systems and infrastructure fundamentals.
• Experience with LLM training or post-training: fine-tuning, RL, distillation, evaluation, data pipelines.
• Excellent debugging skills in distributed systems and data issues.
• Clear communication with technical and non-technical stakeholders.
• High agency, low ego, and ability to operate in fast-moving environments.

Responsibilities

• Build and improve post-training and evaluation workflows (CPT/SFT/RL/distillation).
• Develop tools and pipelines for synthetic data generation, data curation, training, evaluation, and deployment.
• Debug and harden large-scale ML systems: distributed training, scheduling, checkpointing, observability.
• Improve the Forge codebase with clean APIs, tests, and documentation.
• Advance RL training stack for high-throughput, scalable post-training systems at frontier-model scale.
• Ensure Forge deployment is seamless across hardware, software stacks, cloud, and on-premises.
• Collaborate with researchers and infrastructure engineers to translate bottlenecks into system improvements.

Skills

Python engineering

Large codebases

Testing & CI

PyTorch

JAX

Distributed systems

Debugging distributed jobs

Technical communication

Tools

PyTorch

JAX

Kubernetes

SLURM

DeepSpeed

Job description

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

You’ll work end-to-end with scientists, engineers, product, and customer-facing teams to ensure Forge projects ship, are maintainable, and trusted by clients. This Applied Science role spans algorithms, infrastructure, evals, and data.