Operate and optimize large-scale LLM pre-training on 1,000+ GPU clusters using PyTorch, DeepSpeed, or Megatron-LM. Improve networking (InfiniBand/RDMA), memory management, checkpointing, and failure recovery. Manage SLURM/Kubernetes GPU clusters and apply systems engineering (C++, CUDA, Python) and 3D parallelism techniques.
We are seeking a highly skilled LLM Pre-training & Distributed Systems Engineer. This role is essential for orchestrating large-scale machine learning training runs and optimizing distributed infrastructure. The ideal candidate will have a deep understanding of GPU clusters and extensive experience in system engineering to ensure efficient and reliable training processes.
Responsibilities:
- Orchestrate distributed training runs across 1,000+ GPUs using PyTorch, DeepSpeed, or Megatron-LM.
- Optimize networking (InfiniBand/RDMA) and memory management to prevent out-of-memory errors.
- Automate checkpointing and failure recovery during month-long training runs.
Required Skills:
- Deep expertise in 3D parallelism (Data, Tensor, Pipeline).
- Experience managing SLURM or Kubernetes-based GPU clusters.
- Strong systems engineering background (C++, CUDA, Python).
Similar Jobs
.png)
Artificial Intelligence • Blockchain • Fintech • Financial Services • Cryptocurrency • NFT • Web3
Own exception management and operational readiness for listed derivatives and crypto products. Lead UAT, playbooks, cutovers, monitoring, incident triage, root-cause analysis, and automation while partnering with Product, Engineering, Risk, Compliance, and Accounting.
Top Skills:
BlockchainCryptocurrencyGenerative Ai
Artificial Intelligence • Healthtech • Logistics • Social Impact • Software • Telehealth
Support in-home clinical leadership by managing workforce and payroll compliance, tracking attendance and training, preparing operational reports, maintaining SOPs and dashboards, routing non-clinical issues, and participating in audits to ensure HIPAA and policy adherence.
Top Skills:
Ehr SystemsGoogle DocsGoogle SheetsGoogle SlidesRipplingSlack
Artificial Intelligence • Machine Learning • Natural Language Processing • Software • Conversational AI
Lead and scale the People function by managing HRBPs and People Ops, executing the people roadmap, ensuring HRIS and compliance, driving performance management and talent development, handling complex employee relations, and promoting AI-native HR practices across a distributed, remote-first organization.
Top Skills:
Ai ToolsCloud ApisFoundation ModelsHrisSpeech-To-TextText-To-Speech
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
