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
.png){kind=logo}
AdTech • Cloud • Digital Media • Information Technology • News + Entertainment • App development
Create high-quality 3D character gameplay animations for navigation, interactions, conversations, traversal, and other gameplay systems. Export, implement, and test animations in the game engine; troubleshoot animation issues; collaborate with animators, designers, and gameplay programmers; participate in reviews; and mentor junior animators. The role requires strong knowledge of body mechanics, biped locomotion, motion capture, keyframe animation, gameplay systems, and animation workflows, with experience shipping an AAA game.
Top Skills:
Animation Blend SystemsAnimation GraphsKeyframe AnimationMayaMotion CaptureMotion MatchingMotionbuilderRiggingRoot MotionUnreal Engine
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Participate in a 15-week full-time software engineering internship in Vancouver, Burnaby, or Richmond. Build and ship product features, contribute code to Atlassian products, apply data structures and algorithms, and learn full-lifecycle development through mentorship and collaboration with senior engineers. Work may involve cloud initiatives and AI-driven features.
Top Skills:
CC++JavaPython
Cloud • Security • Software • Cybersecurity • Automation
Own and evolve GitLab’s authorization systems across its Ruby on Rails monolith and next-generation Rust policy engine. Design fine-grained permissions for users, tokens, roles, and AI agents; secure GraphQL and REST APIs; lead feature-flagged rollouts and migrations; improve performance and reliability; and collaborate across authentication, platform, AI, and modular-service teams in a distributed asynchronous environment.
Top Skills:
CedarGoGraphQLGrpcProtocol BuffersRestRubyRuby On RailsRustYamlZanzibar-Style Authorization
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.
