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
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Drive adoption and scale of new Samsara product lines across Mid-Market accounts. Partner with AEs/ADRs to develop pipeline, lead pilots, influence GTM and product roadmap, and build scalable sales assets.
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Partner with Account Executives to scale new Samsara product lines across Mid-Market accounts. Drive pipeline and pilot strategies, execute high-value deals, influence GTM and product roadmap, and build repeatable assets to accelerate adoption and sales across the segment.
Top Skills:
Ai MulticamAppsAsset TagsDriver WorkflowsEquipment MonitoringIotMaintenanceRoutingSaaSTofuVehicle TelematicsVideo-Based Safety
Artificial Intelligence • Legal Tech
Provide high-trust executive support to the General Counsel: manage a complex multi-timezone calendar, prepare meeting materials, coordinate travel, triage communications, support recruiting/onboarding, track projects in Notion/Asana, coordinate outside counsel and vendors, handle HR/compensation administration, manage documents in Google Drive, and process expenses. Must handle confidential legal and personnel matters with discretion.
Top Skills:
AsanaAshby (Ats)DocsDriveGc AiGmail)Google Workspace (CalendarGranolaHubspotNotionReclaim.AiSheetsSlackZoom
What you need to know about the Colorado Tech Scene
With a business-friendly climate and research universities like CU Boulder and Colorado State, Colorado has made a name for itself as a startup ecosystem. The state boasts a skilled workforce and high quality of life thanks to its affordable housing, vibrant cultural scene and unparalleled opportunities for outdoor recreation. Colorado is also home to the National Renewable Energy Laboratory, helping cement its status as a hub for renewable energy innovation.
Key Facts About Colorado Tech
- Number of Tech Workers: 260,000; 8.5% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Lockheed Martin, Century Link, Comcast, BAE Systems, Level 3
- Key Industries: Software, artificial intelligence, aerospace, e-commerce, fintech, healthtech
- Funding Landscape: $4.9 billion in VC funding in 2024 (Pitchbook)
- Notable Investors: Access Venture Partners, Ridgeline Ventures, Techstars, Blackhorn Ventures
- Research Centers and Universities: Colorado School of Mines, University of Colorado Boulder, University of Denver, Colorado State University, Mesa Laboratory, Space Science Institute, National Center for Atmospheric Research, National Renewable Energy Laboratory, Gottlieb Institute
