Frontier AI Systems Engineer — Remote, Equity, Impact

Prime Intellect, Inc.; United States; Remote; USD 150,000 - 350,000; Full time

Benefits offered by this job

Remote work options

Visa sponsorship

Relocation assistance

Quarterly off-sites

Learning opportunities

Job summary

Prime Intellect, Inc. is building the open frontier AI stack, a full-stack platform for post-training at frontier scale. We are seeking engineers to design and optimize distributed training infrastructure and collaborate with researchers and infrastructure engineers.

You will work on kernel and runtime optimizations, multi-node GPU clusters, and open-source ML systems, with remote-friendly, SF-office options and visa sponsorship for international candidates.

Qualifications

• Strong systems engineering experience in AI/ML infrastructure, especially around large-scale model training or inference.
• Deep familiarity with PyTorch and distributed training frameworks such as PyTorch Distributed, DeepSpeed, FSDP, Megatron, vLLM, Ray, or related tooling.
• Experience optimizing training performance across kernels, memory movement, communication overhead, or parallelization strategy.
• Hands-on experience with large-scale training techniques including data parallelism, tensor parallelism, and pipeline parallelism.

Responsibilities

• Build and optimize the distributed training infrastructure behind pre-training and large-scale RL workloads.
• 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.
• 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 with researchers and infra engineers to translate bottlenecks into system improvements.
• Stay at the frontier of training/inference systems, compiler/runtime tooling, and hardware-aware optimization.

Skills

Systems engineering

AI/ML infrastructure

PyTorch

Distributed training

CUDA / Triton kernels

Performance debugging

GPU architectures

Open-source contributions

Tools

PyTorch Distributed

DeepSpeed

FSDP

Megatron

vLLM

Ray

Job description

Prime Intellect, Inc. is building the open frontier AI stack, a full-stack platform for post-training at frontier scale. We are seeking engineers to design and optimize distributed training infrastructure and collaborate with researchers and infrastructure engineers.

You will work on kernel and runtime optimizations, multi-node GPU clusters, and open-source ML systems, with remote-friendly, SF-office options and visa sponsorship for international candidates.