ashby
Research Engineer, Code Agents Infra
Mistral AI
Palo Alto, USA
Full-time
Mid · 4+ years experience
Hybrid
Discovered 2 weeks ago
cplusplusdockerkuberneteslinuxpythonpytorch
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
cplusplusdockerkubernetes
Key Skills for This Role
cplusplusdockerkuberneteslinuxpythonpytorch
Full Job Posting
About Mistral
Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. We partner with enterprises tackling the hardest problems—across high-stakes industries like finance, manufacturing, defense, healthcare, and the public sector—co-creating customized AI systems that they can run on their terms.
We are a dynamic, collaborative team passionate about AI and its potential to transform society. Our diverse workforce thrives in competitive environments and is committed to driving innovation. Our teams are distributed between Europe, North America, Asia and the Middle East. We are creative, low-ego and team-spirited.
The Role
This role focuses on building and operating the end-to-end execution, training, and data infrastructure that powers Mistral’s agentic models and coding assistants. You will be a core contributor to our agent research stack: designing scalable systems for synthetic data generation, building ultra-fast training and RL execution environments, and maintaining high-throughput execution engines.
You will tackle the engineering challenges at every step of the agent lifecycle: from orchestrating 1M+ concurrent and short-lived sandboxes for untrusted code execution to optimizing agent training codebases, distributed trajectory collection pipelines, and dataset processing workflows across massive hybrid and multi-cloud clusters.
What You Will Do
- Large-Scale Sandboxing Infrastructure: Design, deploy, and operate our high-throughput sandboxing platform, executing LLM-generated untrusted code across over 1 million isolated environments concurrently for model evaluation and interactive RL environments.
- Agent Data Generation Pipelines: Architect and scale high-throughput pipelines for synthetic code generation, agent trajectories, rollouts, and self-play data collection to power post-training and RL loops.
- Training Codebase & Systems Optimization: Optimize agent training codebases and distributed execution runtimes (PyTorch, Ray, SLURM/Kubernetes) to minimize multi-step rollout overhead, improve GPU utilization, and eliminate scaling bottlenecks.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Mistral AI
Applied AI Engineer, Australia
Sydney, AUS
Discovered TodayFull-time
AI Deployment Strategist, Australia
Sydney, AUS
Discovered TodayFull-time
Enterprise Sales Manager
San Francisco, USA
Discovered YesterdayFull-time
Applied AI Engineer, AI for Mistral
Paris, USA
Discovered 5 days agoFull-time
Brand Creative Director
Paris, USA
Discovered 5 days agoFull-time
Technical Program Manager, Science Operations
Paris, USA
Discovered 5 days agoFull-time
Compute Solution Architect
New York City, USA
Discovered 6 days agoFull-time
Technical Recruiter, APAC
Melbourne, AUS
Discovered 6 days agoFull-time
