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 • Cloud • Computer Vision • Hardware • Internet of Things • Software
Serve as the primary post-implementation contact for top customers, craft joint success plans, run executive business reviews and workshops, mentor teammates, support French-speaking customers, and advise on customizing Samsara’s IoT platform to drive safety, efficiency, and sustainability.
Top Skills:
IotSamsara PlatformVehicle TelematicsVideo-Based Safety
Gaming
Lead product vision, roadmap, and outcomes for Hit It Rich Slots. Drive monetization and live-ops strategy, coordinate cross-functional teams and partners, define analytics standards, and own long-term player engagement, retention, and revenue growth.
Cloud • Fintech • Food • Information Technology • Software • Hospitality
Lead design and delivery of scalable ML systems across the full lifecycle for recommendations, forecasting, targeting, and personalization. Collaborate with engineers and product teams, set modeling and productionization standards, mentor data scientists, guide architecture and experimentation, and apply cloud and MLOps best practices to drive measurable product and business impact.
Top Skills:
AthenaAWSBedrockCi/CdDynamoDBGlueLlm Fine-TuningMl Ops FrameworksPythonPyTorchRlhfSagemakerScikit-LearnSQLTensorFlowVersion Control
What you need to know about the Ottawa Tech Scene
The capital city of Canada and the nation's fourth-largest urban area, Ottawa has proven a rapidly growing global tech hub. With over 1,800 tech companies, many of which are leaders in their sectors, the city's tech talent now makes up more than 13 percent of its total workforce. This growth is driven not only by the big players like UL Solutions and Dropbox, but also by a thriving startup ecosystem, as new businesses emerge to follow in the footsteps of those that came before them.
