Remote Jobs RockRemote Jobs Rock

Research Engineer, Code Agents Infra

πŸ“… Aug 4
Distributed SystemsCloud InfrastructureMlopsKubernetes
Apply Now

πŸ“œ Description

  • Design and operate sandboxing infrastructure for over 1 million concurrent isolated environments running untrusted code.
  • Build scalable pipelines for synthetic code generation, agent trajectories, rollouts, and self-play data collection.
  • Optimize agent training codebases and distributed runtimes to reduce rollout overhead, improve GPU utilization, and remove scaling bottlenecks.
  • Reduce sandbox cold starts with warm pools, snapshot and restore technology, and optimized image delivery.
  • Develop Kubernetes controllers and queuing systems to route short-lived workloads across multi-cluster hardware fleets.
  • Maintain isolation, availability, telemetry, and automated recovery for transient jobs; participate in on-call rotations.

πŸ› οΈ Requirements

  • 4+ years of experience in Systems Engineering, Distributed Systems, Cloud Infrastructure, or MLOps supporting LLM/RL workloads.
  • Data & Pipeline Engineering: Proven experience building high-throughput data processing and generation pipelines for large-scale datasets (e.g., Ray, Spark, custom distributed queues).
  • Deep experience with Kubernetes & Container Tech: Strong expertise writing custom K8s operators/controllers, managing Linux cgroups/namespaces, and optimizing Docker image layers and distribution systems.
  • High-Performance Software Engineering: Advanced proficiency in Python, Go, C++ or Rust, with a track record of profiling and optimizing high-performance ML or backend systems codebases.
  • Sandboxing & Isolation Technologies: Hands-on experience with lightweight virtualization, container runtimes, or WASM (e.g., Docker, gVisor, Firecracker).
  • Queueing & Scheduling: Deep familiarity with task queue systems, resource schedulers, and low-latency queuing architectures for high-volume, short-lived workloads.
  • Comfort with Ambiguity: Passion for working directly alongside AI researchers to rapidly turn frontier agent ideas into scalable, production-grade infrastructure.

✨ Benefits

  • Healthcare coverage
  • Parental leave
  • Retirement plans
  • Relocation support
  • Wellness programs
  • Meal allowances
  • Transportation allowances
Full job description

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.

  • Low-Latency Orchestration & Warm Pooling: Reduce sandbox cold-start times to sub-second levels using container warm pools, snapshot/restore technology (e.g., CRIU, microVMs), and optimized image delivery layers across hybrid clusters.

  • Multi-Cluster Queueing & Resource Allocation: Implement Kubernetes-native custom controllers, CRDs, and queuing systems to dynamically route short-lived evaluation, synthetic data, and agent execution tasks across diverse hardware fleets.

  • Isolation, Security & Security Boundary: Ensure strict multi-tenant network and process isolation for untrusted agent code using container/sandboxing runtimes (e.g., gVisor, Firecracker) and default-deny network postures.

  • Operational Excellence: Maintain high availability, telemetry, and automated self-healing across millions of transient jobs while participating in on-call rotations for critical agent training and execution pipelines.

What We're Looking For

  • 4+ years of experience in Systems Engineering, Distributed Systems, Cloud Infrastructure, or MLOps supporting LLM/RL workloads.

  • Data & Pipeline Engineering: Proven experience building high-throughput data processing and generation pipelines for large-scale datasets (e.g., Ray, Spark, custom distributed queues).

  • Deep experience with Kubernetes & Container Tech: Strong expertise writing custom K8s operators/controllers, managing Linux cgroups/namespaces, and optimizing Docker image layers and distribution systems.

  • High-Performance Software Engineering: Advanced proficiency in Python, Go, C++ or Rust, with a track record of profiling and optimizing high-performance ML or backend systems codebases.

  • Sandboxing & Isolation Technologies: Hands-on experience with lightweight virtualization, container runtimes, or WASM (e.g., Docker, gVisor, Firecracker).

  • Queueing & Scheduling: Deep familiarity with task queue systems, resource schedulers, and low-latency queuing architectures for high-volume, short-lived workloads.

  • Comfort with Ambiguity: Passion for working directly alongside AI researchers to rapidly turn frontier agent ideas into scalable, production-grade infrastructure.

What We Offer

We offer a comprehensive benefits package designed to support your well-being, growth, and work-life balance. Benefits vary by country and may include healthcare coverage, parental leave, retirement plans, relocation support, wellness programs, meal and transportation allowances, and other location-specific perks.

For the most up-to-date details on benefits available in your location, please refer to our Benefits page.

Privacy Policy

Your privacy matters to us. You can learn more about how we handle your personal data in our Applicant Privacy Policy.

Anthropic

Security Engineer / Research Engineer, RL Security

πŸ•’ 20 days ago
AnthropicπŸ‘₯ 10,000+ employees🏒 Research Services🀝 B2B🧠 AI-native

Build and operate security controls for large-scale reinforcement learning systems, reviewing code and designs, developing shared tooling and observability, and balancing security with training correctness and performance.

PythonRustC++Go
Harvey

Research Engineer, Infrastructure

πŸ•’ 7 days ago
HarveyπŸ‘₯ 10,000+ employees🏒 Software Development🀝 B2B🧠 AI-native

Build and own infrastructure for post-training open-weight models, including distributed training and rollouts, agent environments, evaluation tools, and reliable production integration in close partnership with researchers.

PythonPytorchJAXML Infrastructure
Harvey

Research Engineer, Infrastructure

πŸ•’ 7 days ago
HarveyπŸ‘₯ 10,000+ employees🏒 Software Development🀝 B2B🧠 AI-native

Build and scale the infrastructure for post-training open-weight models, including distributed training and rollouts, agent environments, evaluation tools, and reliable production serving in close partnership with researchers.

PythonPytorchJAXML Infrastructure
Harvey

Research Engineer, Infrastructure

πŸ•’ 7 days ago
HarveyπŸ‘₯ 10,000+ employees🏒 Software Development🀝 B2B🧠 AI-native

Build and operate the infrastructure behind legal-model post-training, including distributed training and rollouts, agent environments, evaluation tools, and reliable production serving in close partnership with researchers.

PythonPytorchJAXML Infrastructure
Anthropic

Research Engineer, Cybersecurity RL

πŸ•’ yesterday
AnthropicπŸ‘₯ 10,000+ employees🏒 Research Services🀝 B2B🧠 AI-native

Develop and implement reinforcement learning approaches for defensive cybersecurity, designing environments, running experiments and evaluations, and delivering work into production training runs for safer AI models.

Cybersecurity ResearchMachine LearningSoftware EngineeringReinforcement Learning

Trusted by Remote Workers