Distributed RL Research Engineer - Flexible Work, SF Office

Prime Intellect, Inc.; United States; Hybrid; USD 150,000 - 350,000; Full time

Job summary

Prime Intellect builds the open superintelligence stack, delivering full-stack infrastructure for frontier-scale AI post-training and deployment. The company has raised $150M from top investors and pursues open-source AI to accelerate science.

We are seeking engineers to optimize large-scale RL systems, improve training efficiency, and design low-level kernels and runtimes. Flexible work arrangements include SF office options and visa sponsorship for qualified candidates.

Qualifications

• Deep PyTorch and distributed training framework expertise.
• Experience with GPU architecture, profiling, and performance debugging.

Responsibilities

• Optimize large-scale RL and distributed training systems.
• Improve training efficiency across compute, memory, networking, and scheduling.
• Design low-level performance optimizations like kernels and runtime.

Skills

Deep PyTorch

Distributed training

GPU architecture

Profiling

Performance debugging

Tools

CUDA

Profiling tools

Job description

Prime Intellect builds the open superintelligence stack, delivering full-stack infrastructure for frontier-scale AI post-training and deployment. The company has raised $150M from top investors and pursues open-source AI to accelerate science.

We are seeking engineers to optimize large-scale RL systems, improve training efficiency, and design low-level kernels and runtimes. Flexible work arrangements include SF office options and visa sponsorship for qualified candidates.