Research Engineer - RL Infrastructure

What this role involves

Prime Intellect is hiring a Research Engineer to build and optimize systems infrastructure for large-scale RL and distributed training. The role involves contributing to the prime-rl framework, improving training efficiency, and implementing low-level performance optimizations. Ideal candidates have strong systems engineering experience in AI/ML infrastructure and familiarity with distributed training frameworks.

Skills this posting asks for

• pytorch
• pytorch distributed
• deepspeed
• fsdp
• megatron
• vllm
• ray
• cuda
• triton
• distributed training
• data parallelism
• tensor parallelism
• pipeline parallelism
• gpu architecture
• performance profiling
• kernel optimization
• memory optimization
• communication optimization
• compiler optimization
• runtime optimization
• rl infrastructure
• async rollout
• multi-node gpu clusters
• high-performance networking

From the employer’s posting

OWN YOUR INTELLIGENCE Prime Intellect is building the open superintelligence stack: the infrastructure frontier AI labs build internally, made available to every ambitious AI team. Our platform, Lab, unifies compute, environments, evaluations, secure sandboxes, high-performance training, and…