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
Cloud • Security • Software • Cybersecurity • Automation
Build and maintain backend services and APIs using Ruby and GraphQL to improve source code workflows. Ensure reliability, performance, testing, observability, and on-call support. Collaborate cross-functionally and use AI agents responsibly to accelerate delivery.
Top Skills:
Ai Coding AgentsBackground ProcessingCachingGitGitalyGraphQLMonorepoObservabilityRubySQL
Cloud • Security • Software • Cybersecurity • Automation
Build and maintain backend services and APIs for source code workflows using Ruby and GraphQL. Improve performance, reliability, caching, and observability. Collaborate cross-functionally, write tests, participate in Tier 2 on-call, and use AI coding agents where appropriate.
Top Skills:
GitGitalyGraphQLRubySQL
Big Data • Fintech • Mobile • Payments • Financial Services
Lead and develop a team of Technical Account Managers supporting SMB merchants post-sale. Drive technical integrations, troubleshoot escalations, set OKRs, forecast hiring, and advocate cross-functionally to increase merchant adoption and product optimization.
Top Skills:
Rest ApisWeb Development
What you need to know about the Calgary Tech Scene
Employees can spend up to one-third of their life at work, so choosing the right company is crucial, not just for the job itself but for the company culture as well. While startups often offer dynamic culture and growth opportunities, large corporations provide benefits like career development and networking, especially appealing to recent graduates. Fortunately, Calgary stands out as a hub for both, recognized as one of Startup Genome's Top 100 Emerging Ecosystems, while also playing host to a number of multinational enterprises. In Calgary, job seekers can find a wide range of opportunities.
