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
Cloud • Information Technology • Security • Software • Cybersecurity
Lead escalation management for critical customer issues across assigned region. Build and execute cross-functional action plans, communicate status to executives, drive resolution, perform post-mortems, and recommend product/process improvements.
Top Skills:
Ai/MlCloud-NativePredictive AnalyticsSecure Service EdgeSentiment AnalysisZero Trust ExchangeZero Trust Network AccessZero Trust Sase
Cloud • Information Technology • Security • Software • Cybersecurity
Manage and expand Zscaler's global peering footprint by identifying strategic partners, coordinating with network engineering, legal, procurement, and negotiating peering agreements. Represent Zscaler at global peering forums and events, enable new peering connectivity, and design scalable network solutions to improve global connectivity for the Zero Trust platform.
Top Skills:
Ai ToolsBgpPrivate PeeringPublic Internet ExchangesRouter ConfigurationsZero Trust Exchange
Cloud • Information Technology • Security • Software • Cybersecurity
Lead design and governance of global sales compensation plans and AI-native incentive frameworks. Architect metrics for consumption and AI monetization, drive C-level decisions, build AI tools for modeling and forecasting, and partner with Finance, HR, IT, Product, and GTM to enable accurate, auditable payouts and change management for global sales teams.
What you need to know about the Los Angeles Tech Scene
Los Angeles is a global leader in entertainment, so it’s no surprise that many of the biggest players in streaming, digital media and game development call the city home. But the city boasts plenty of non-entertainment innovation as well, with tech companies spanning verticals like AI, fintech, e-commerce and biotech. With major universities like Caltech, UCLA, USC and the nearby UC Irvine, the city has a steady supply of top-flight tech and engineering talent — not counting the graduates flocking to Los Angeles from across the world to enjoy its beaches, culture and year-round temperate climate.
Key Facts About Los Angeles Tech
- Number of Tech Workers: 375,800; 5.5% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Snap, Netflix, SpaceX, Disney, Google
- Key Industries: Artificial intelligence, adtech, media, software, game development
- Funding Landscape: $11.6 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Strong Ventures, Fifth Wall, Upfront Ventures, Mucker Capital, Kittyhawk Ventures
- Research Centers and Universities: California Institute of Technology, UCLA, University of Southern California, UC Irvine, Pepperdine, California Institute for Immunology and Immunotherapy, Center for Quantum Science and Engineering
