Research Engineer - RL Infrastructure (Fully Remote)

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

• 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.

Requirements:

• Deep PyTorch and distributed training framework expertise.
• Experience with GPU architecture, profiling, and performance debugging.
• $150-350k cash compensation with equity.
• Flexible work arrangements and SF office option.
• Visa sponsorship and relocation assistance.

Prime Intellect

Prime Intellect builds the open superintelligence stack, providing full-stack infrastructure for frontier-scale AI post-training and deployment. The company has raised $150M from top investors and is mission-driven, focused on open-source AI and accelerating science.