Distributed AI Training Research Engineer (Remote)

Prime Intellect; United States; Hybrid; USD 110,000 - 150,000; Full time

Benefits offered by this job

Competitive compensation including equity incentives

Flexible work arrangements

Visa sponsorship and relocation assistance

Quarterly team off-sites and hackathons

Opportunity to work with a mission-driven team

Job summary

A cutting-edge AI company is seeking a Research Engineer specializing in Distributed Training. The candidate will be responsible for leading research to develop a decentralized AI training orchestration solution while optimizing workloads for performance and cost. Ideal applicants will have a strong AI/ML engineering background, expertise in distributed training frameworks, and a passion for advancing AI capabilities. The company offers competitive compensation, flexible work arrangements, and opportunities for professional growth.

Qualifications

• Strong background in AI/ML engineering with experience in end-to-end pipelines for large-scale AI models.
• Expertise in distributed training techniques, frameworks like PyTorch Distributed, and tools like Ray.
• Solid understanding of MLOps best practices including model versioning and CI/CD pipelines.

Responsibilities

• Lead research to build a decentralized training orchestration solution.
• Optimize AI workloads using recent compute & memory optimization techniques.
• Contribute to open-source libraries for distributed model training.
• Publish research in top-tier AI conferences such as ICML & NeurIPS.
• Distill technical outcomes into approachable blogs for customers.
• Stay up-to-date with advancements in AI/ML infrastructure.

Job description

A cutting-edge AI company is seeking a Research Engineer specializing in Distributed Training. The candidate will be responsible for leading research to develop a decentralized AI training orchestration solution while optimizing workloads for performance and cost. Ideal applicants will have a strong AI/ML engineering background, expertise in distributed training frameworks, and a passion for advancing AI capabilities. The company offers competitive compensation, flexible work arrangements, and opportunities for professional growth.