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 • HR Tech • Information Technology • Productivity • Software • Automation
Lead production database troubleshooting and performance tuning across multi-tenant PostgreSQL/MariaDB fleets. Own incident RCAs, develop observability and automation, influence infrastructure decisions, mentor cross-functional engineers, and support large-scale web distributed applications and Linux-based environments in cloud and on-prem deployments.
Top Skills:
AnsibleApacheCi/CdCloud InfrastructureContainersJavaScriptJbossKubernetesLinuxMariadbPaasPostgresPythonSaaSServicenowShell ScriptingTomcatWeblogicWebsphere
.png)
Artificial Intelligence • Blockchain • Fintech • Financial Services • Cryptocurrency • NFT • Web3
The Senior Software Engineer will manage data systems, develop scalable pipelines, ensure data security, and build self-service applications for users at Coinbase.
Top Skills:
AirflowGoJavaKafkaPythonSparkSQL
Artificial Intelligence • Blockchain • Fintech • Financial Services • Cryptocurrency • NFT • Web3
Lead design and delivery of backend risk systems to detect and prevent fraud, manage credit and market risk, and protect users. Drive architecture for distributed, high-availability services, partner with Data Science/ML and product teams, build AI-native detection and response systems, mentor engineers, own operational excellence, and lead incident response and post-mortems.
Top Skills:
Event-Driven ArchitectureGenerative AiGoGraphQLJavaMicroservicesPythonRest ApisRuby
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.
