ML Infra Engineer - Scale RL Pipelines (Paris)

United States Digital Space LLC; Paris (TX); Hybrid; USD 125,000 - 182,000; Full time

Benefits offered by this job

Competitive salary + equity

Hybrid work from Paris with relocation

Top-tier medical insurance in France

L&D budget and hardware/tools

Team off-sites twice a year

Job summary

White Circle is seeking an ML Infrastructure Engineer to join its AI safety team focused on policy enforcement and optimization for AI systems. The role requires hands-on experience with distributed RL/post-training systems, Python, PyTorch or JAX, and GPU debugging, with relocation to Paris for a hybrid setup.

You will build scalable pipelines, data control systems, and inference infrastructures, while developing agentic toolchains and multi-agent environments for robust model iteration and

Qualifications

• Hands-on experience designing and running distributed RL/post-training systems at scale.
• Strong Python (concurrency, async, multiprocessing, performance optimization) and PyTorch or JAX.
• Debugging distributed GPU workloads across CUDA, drivers, containers, NCCL, networking, storage, and checkpointing.
• Profiling across the stack (py-spy, PyTorch profiler, Nsight, perf, tracing).
• Inference stacks: vLLM, SGLang, TensorRT-LLM, Dynamo, or custom serving.
• Ability to connect system metrics to model behavior and learning dynamics.
• Relocation to Paris (hybrid) required.

Responsibilities

• Build scalable RL and post-training pipelines, including smoke tuning runs for quality testing and ablations.
• Design data control systems for rollouts, replay, filtering, evaluation, and policy updates.
• Tune training and inference end-to-end for throughput: networking, memory, scheduling, data loading, storage, checkpointing.
• Build infrastructure for model iteration (experiment runs, artifacts, evals, dashboards, reproducibility, cost visibility) and inference infrastructure for post-training and eval loops.
• Build agentic development environments: coding-agent harnesses, tool integrations, runtime sandboxes, multi-agent orchestration.

Skills

Python concurrency

PyTorch

JAX

Distributed systems

CUDA

Profiling tools

Kubernetes

C++

Tools

NCCL

Nsight

PyTorch profiler

TensorRT-LLM

Job description

White Circle is seeking an ML Infrastructure Engineer to join its AI safety team focused on policy enforcement and optimization for AI systems. The role requires hands-on experience with distributed RL/post-training systems, Python, PyTorch or JAX, and GPU debugging, with relocation to Paris for a hybrid setup.

You will build scalable pipelines, data control systems, and inference infrastructures, while developing agentic toolchains and multi-agent environments for robust model iteration and