P

Senior Systems Engineer, AI Infrastructure

Prime Intellect • United State
Remote Visa Sponsorship Relocation
Apply Now
AI Summary

Build and optimize the systems infrastructure behind large-scale RL and distributed training workloads for Prime Intellect's open superintelligence stack. Improve end-to-end training efficiency across compute, memory, networking, and scheduling layers while contributing to the prime-rl framework. Requires deep expertise in PyTorch, distributed training frameworks, and GPU architecture optimization.

Key Highlights
Contribute to the prime-rl framework for large-scale RL and distributed training
Design and implement low-level performance optimizations including kernels and communication paths
Work on distributed training systems spanning data, tensor, and pipeline parallel workloads

Looking to advance your Machine Learning & AI career with relocation support? Explore Machine Learning & AI Jobs with Relocation Packages that include comprehensive packages to help you move and settle in your new role.

Key Responsibilities
Build and optimize the systems infrastructure behind large-scale RL and distributed training workloads by contributing to the prime-rl framework
Improve end-to-end training efficiency across compute, memory, networking, and scheduling layers
Design and implement low-level performance optimizations, including kernels, communication paths, and runtime improvements
Work on distributed training systems spanning data, tensor, and pipeline parallel workloads
Help shape the architecture of the RL training stack, including async rollout and post-training systems
Contribute to open-source libraries and internal infrastructure used for frontier-scale model training
Collaborate closely with researchers and infrastructure engineers to translate bottlenecks into concrete systems improvements
Stay at the frontier of training systems, inference systems, compiler/runtime tooling, and hardware-aware optimization techniques
Technical Skills Required
PyTorch Distributed Training GPU Architecture

Discover our full range of relocation jobs with comprehensive support packages to help you relocate and settle in your new location.

Benefits & Perks
Cash Compensation Range of $150-350k including equity incentives
Flexible work arrangements with remote or in-person options
Visa sponsorship and relocation assistance for international candidates
Quarterly team off-sites, hackathons, conferences and learning opportunities

Interested in relocating to United State? Check out our comprehensive Relocation Jobs in United State page with detailed relocation packages and benefits.

Nice to Have
Experience writing or optimizing CUDA / Triton kernels
Experience with compiler or runtime optimization for ML systems
Experience working on RL training infrastructure, rollout systems, or asynchronous training pipelines
Experience with multi-node GPU clusters and high-performance networking
Contributions to open-source ML systems or infrastructure projects
Interest in publishing technical work or sharing insights through engineering blogs and technical writing

Job Description

Prime Intellect is building the open superintelligence stack: the infrastructure frontier AI labs build internally, made available to every ambitious AI team.
Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

Apply Now
See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources
Job Overview
Posted Date Oct 16, 2026
Employment Type Full-time
Experience Level Not Applicable
Location United State
Category Machine Learning
Company Prime Intellect
Mentioned Skills
ai cloudflare san datadog pytorch deepspeed

Similar Jobs

Explore other opportunities that match your interests

Forward Deployed Engineer - Databricks - Forensic Discovery & Financial Crime
Machine Learning
•
4h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Not Applicable

Deloitte

United State

Senior Machine Learning Engineer (Autonomous Driving AI)
Machine Learning
•
16h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Not Applicable

wayve

United State

Junior AI/ML Engineer
Machine Learning
•
20h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Entry level

Tech Consulting

United State

Subscribe our newsletter

New Things Will Always Update Regularly