Research Engineer - RL Infrastructure (Remote)

About the Role

We're seeking a Research Engineer - RL Infrastructure to join our innovative team at Prime Intellect. In this remote position, you'll be instrumental in building the open superintelligence stack, enabling researchers and enterprises to create, train, and deploy reinforcement learning models at scale. If you're passionate about shaping the future of AI and enjoy working on cutting-edge technology, this Research Engineer - RL Infrastructure remote role is perfect for you.

What You'll Do

• Design and build scalable RL training infrastructure, including async trainers and environment orchestration across large GPU clusters.
• Optimize performance, cost, and resource utilization of RL workloads using state-of-the-art compute and memory optimization techniques.
• Contribute to our open-source libraries and frameworks for distributed RL training.
• Publish research at top-tier venues such as ICML and NeurIPS.
• Write clear, approachable technical content that distills complex systems work for customers and the broader community.
• Stay current with advances in RL systems, distributed training, and ML infrastructure, proactively identifying opportunities to enhance our platform.

Requirements

• Strong background in ML engineering with hands-on experience building and scaling RL or large model training pipelines end-to-end.
• Deep expertise in distributed training techniques and frameworks (e.g., PyTorch Distributed, DeepSpeed, vLLM, Ray).
• Experience with RL-specific infrastructure, including environment management and online/async training loops.
• Solid understanding of MLOps best practices, including experiment tracking and CI/CD.
• Passion for advancing open, scalable RL infrastructure and democratizing access to frontier AI capabilities.

Nice to Have

• Familiarity with additional ML frameworks or languages.
• Experience in publishing research papers in AI-related fields.

What We Offer

• Competitive compensation including equity, aligning your success with Prime Intellect's growth and impact.
• Flexible work arrangements — remote or in-person at our San Francisco office.
• Visa sponsorship and relocation assistance for international candidates.
• Quarterly team offsites, hackathons, conferences, and learning opportunities.
• A talented, hard-working, mission-driven team united by a shared passion for accelerating AI research.