Distributed Training Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Periodic Labs · 2 months ago

Distributed Training Engineer

positionMenlo Park, CA
timeFull-time
remoteOnsite
seniorityMid, Senior Level

Periodic Labs is an AI and physical sciences lab focused on building advanced models for scientific discoveries. The Distributed Training Engineer will optimize and develop large-scale distributed LLM training systems, collaborating closely with researchers to enhance AI scientific research workflows and contribute to open-source training frameworks.

Artificial Intelligence (AI)Foundational AIGenerative AIMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

You will optimize, operate and develop large-scale distributed LLM training systems that power AI scientific research
You will work closely with researchers to bring up, debug, and maintain mid-training and reinforcement learning workflows
You will build tools and directly support frontier-scale experiments to make Periodic Labs the world’s best AI + science lab for physicists, computational materials scientists, AI researchers, and engineers
You will contribute open-source large scale LLM training frameworks

Qualification

Distributed training frameworksLarge-scale LLM trainingTraining on clustersOptimizing training throughputCollaboration with researchers

Required

Training on clusters with ≥5,000 GPUs
5D parallel LLM training
Distributed training frameworks such as Megatron-LM, FSDP, DeepSpeed, TorchTitan
Optimizing training throughput for large scale Mixture-of-Expert models

Company

Periodic Labs

twittertwittertwitter
company-logo
Periodic Labs develops artificial intelligence systems that simulate and predict the properties of materials using machine learning.
dateFounded in 2025
location
San Francisco, California, USA
people-size11-50 employees
web-link
https://periodic.com/

H1B Sponsorship

Periodic Labs has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)

Funding

Current Stage
Early Stage
Total Funding
$300M
2025-09-30Seed· $300M

Recent News

Business Insider
12 executives, researchers, others who left OpenAI in 2025 — mostly to Meta Superintelligence Lab
2025-12-26
IndiaTimes
Shengjia Zhao, Jason Wei and list of other executives, AI researchers who quit OpenAI in 2025
2025-12-26
Inside HPC & AI News | High-Performance Computing & Artificial Intelligence
Genesis Mission: DOE Announces Agreements with 24 Organizations
2025-12-20
Company data provided by crunchbase