Frontier AI Systems Engineer - Distributed Training (Remote)

Prime Intellect; San Francisco, Northern (CA, KY); Hybrid; USD 150,000 - 350,000; Full time

Job summary

Prime Intellect is building open frontier AI infrastructure, unifying training and deployment to enable frontier-scale models. We seek experts to architect and optimize distributed training systems and contribute to open-source components.

You will collaborate with researchers and infra engineers, optimize performance across compute, memory, and networking, and shape the RL training stack with async rollout and post-training systems.

Qualifications

• Strong systems engineering experience in AI/ML infrastructure.

Responsibilities

• Build and optimize the distributed training infrastructure behind our pre-training and large-scale RL training workloads.

Skills

Systems engineering

AI/ML infrastructure

PyTorch

Distributed training

Performance profiling

Tools

PyTorch Distributed

DeepSpeed

FSDP

Megatron

vLLM

Ray

CUDA

Triton

Job description

Prime Intellect is building open frontier AI infrastructure, unifying training and deployment to enable frontier-scale models. We seek experts to architect and optimize distributed training systems and contribute to open-source components.

You will collaborate with researchers and infra engineers, optimize performance across compute, memory, and networking, and shape the RL training stack with async rollout and post-training systems.