Research Engineer, AI Systems & Post-Training Pipelines

Lindus Health; Palo Alto (CA); On-site; USD 150,000 - 230,000; Full time

Benefits offered by this job

Healthcare coverage

Relocation support

Job summary

Mistral is seeking a Research Engineer on Forge to translate customer requirements into reliable training and deployment workflows. You will work end-to-end across model adaptation, post-training, evaluation, data, and infrastructure, collaborating with scientists, engineers, product, and customer-facing teams to ship Forge projects.

This role sits in Applied Science and emphasizes scalable systems, maintainability, and the ability to improve the entire stack for real client outcomes in a

Qualifications

• Strong Python engineering skills and work in large codebases with testing and CI.
• Hands-on experience with PyTorch, JAX, or similar frameworks.
• Solid systems and infrastructure fundamentals and debugging ability.
• Experience with LLM training or post-training: fine-tuning, RL, distillation, data pipelines.
• Clear communication with technical and non‑technical stakeholders.

Responsibilities

• Build and improve post-training and evaluation workflows, turning prototypes into repeatable Forge recipes.
• Develop 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 APIs, tests, documentation, and maintainable abstractions.
• Advance RL training stack with high-throughput rollout and scalable post-training systems.
• Ensure Forge deployment is adaptable to diverse clients, hardware, cloud, and on-premises.
• Collaborate with researchers and infrastructure engineers to translate bottlenecks into system improvements.

Skills

Python engineering

PyTorch

JAX

Large codebases

CI and testing

Distributed training

Debugging complex systems

Communication with stakeholders

Tools

SLURM

Ray

Kubernetes

Kueue

Karpenter

Skypilot

DeepSpeed

Megatron

Job description

Mistral is seeking a Research Engineer on Forge to translate customer requirements into reliable training and deployment workflows. You will work end-to-end across model adaptation, post-training, evaluation, data, and infrastructure, collaborating with scientists, engineers, product, and customer-facing teams to ship Forge projects.

This role sits in Applied Science and emphasizes scalable systems, maintainability, and the ability to improve the entire stack for real client outcomes in a