Inactive
P

Prime Intellect

Open frontier-model stack for enterprise RL

Research Engineer - RL Infrastructure

Applied Machine LearningPosted on Mar 27
Full-timeHybrid

About the Job

Requirements

  • Strong systems engineering experience in AI/ML infrastructure, especially around large-scale model training or inference.
  • Deep familiarity with PyTorch and distributed training frameworks such as PyTorch Distributed, DeepSpeed, FSDP, Megatron, vLLM, Ray, or related tooling.
  • Experience optimizing training performance across kernels, memory movement, communication overhead, or parallelization strategy.
  • Hands-on experience with large-scale training techniques including data parallelism, tensor parallelism, and pipeline parallelism.
  • Strong understanding of GPU architecture, profiling, and performance debugging.
  • Ability to identify bottlenecks across the stack and drive improvements from first principles.
  • Comfort working in a fast-moving environment with ambiguous problems and high ownership.

Responsibilities

  • Build and optimize the systems infrastructure behind large-scale RL and distributed training workloads by contributing to the prime-rl framework.
  • Improve end-to-end training efficiency across compute, memory, networking, and scheduling layers.
  • Design and implement low-level performance optimizations, including kernels, communication paths, and runtime improvements.
  • Work on distributed training systems spanning data, tensor, and pipeline parallel workloads.
  • Help shape the architecture of the RL training stack, including async rollout and post-training systems.
  • Contribute to open-source libraries and internal infrastructure used for frontier-scale model training.
  • Collaborate closely with researchers and infrastructure engineers to translate bottlenecks into concrete systems improvements.
  • Stay at the frontier of training systems, inference systems, compiler/runtime tooling, and hardware-aware optimization techniques.

Desired Qualifications

  • Experience writing or optimizing CUDA/Triton kernels.
  • Experience with compiler or runtime optimization for ML systems.
  • Experience working on RL training infrastructure, rollout systems, or asynchronous training pipelines.
  • Experience with multi-node GPU clusters and high-performance networking.
  • Contributions to open-source ML systems or infrastructure projects.
  • Interest in publishing technical work or sharing insights through engineering blogs and technical writing.

About the Company

Prime Intellect

Open frontier-model stack for enterprise RL

Prime Intellect provides an open, end-to-end superintelligence stack for enterprises to train and continually improve their own AI agents. The platform includes scalable compute, large-scale reinforcement learning, environments, sandboxes, evaluations, and deployment tools, and it ships open frontier models to customers. It covers the full lifecycle from training to deployment plus post-training support, enabling agentic training at scale. Its differentiator is the open, end-to-end stack used across thousands of customers with strong backing from top investors, and its goal is to help organizations build and iteratively improve their AI agents using open frontier models.

Company size
51-200
Company stage
Series A
Total funding
$150.5M
Headquarters
San Francisco, CA
Founded
2024

Recently Posted Jobs

Explore more opportunities at Prime Intellect

See more jobs