Design, orchestrate, and optimize large-scale LLM pre-training across 1,000+ GPUs. Implement 3D parallelism, manage GPU clusters (SLURM/Kubernetes), optimize InfiniBand/RDMA networking and memory, and automate checkpointing and failure recovery for long training runs.
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
Artificial Intelligence • Information Technology • Machine Learning • Natural Language Processing • Productivity • Software • Generative AI
Build and operate high-throughput backend services for AI agent orchestration, calendar systems, real-time infrastructure, APIs, and data pipelines. Design scalable platform services, infrastructure as code, and highly available production systems using Go, PostgreSQL, Kubernetes, and cloud platforms. Collaborate with AI, product, and frontend teams to deliver AI-powered features, contribute to reliability and testing practices, and mentor junior engineers.
Top Skills:
AWSAzureCi/CdClaude CodeCursorGCPGithub CopilotGoInfrastructure As CodeKubernetesPostgres
.png){kind=logo}
AdTech • Cloud • Digital Media • Information Technology • News + Entertainment • App development
Leads the vision, strategy, roadmap, and lifecycle of NBCUniversal’s mobile product experience. Partners with design, engineering, analytics, production, and business teams to prioritize features, improve engagement, retention, onboarding, monetization, and product health. Drives research-informed decisions, launch readiness, stakeholder alignment, and continuous improvement while helping build the mobile product organization.
Top Skills:
A/B TestingConsole PlatformsCross-Platform Product DevelopmentMobile PlatformsPc PlatformsProduct Analytics
AdTech • Cloud • Digital Media • Information Technology • News + Entertainment • App development
Leads environment art for an in-game world, setting visual standards, mentoring artists, reviewing assets, collaborating with level design and art leadership, improving pipelines, planning workloads, solving technical issues, and documenting best practices. Requires 3D environment production experience for shipped games, proficiency with 3D software and world-building tools, and familiarity with modeling, texturing, lighting, shaders, materials, and procedural workflows.
Top Skills:
3Ds MaxBlenderHoudiniIn-Engine Lighting ToolsLevel-Editing ToolsMaterial GraphsMayaModoShader CreationSubstanceWorld-Building Tools
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
