Research Engineer - RL Infrastructure

Prime Intellect

Remote regions

Global

Salary range

$150,000–$350,000/yr

Benefits

Similar Jobs

See all

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.

Requirements:

  • Strong systems engineering in AI/ML infrastructure.
  • Deep PyTorch and distributed training framework expertise.
  • Experience with GPU architecture, profiling, and performance debugging.

Benefits:

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

Apply for This Position