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
Big Data • Fintech • Mobile • Payments • Financial Services
Own the vision and roadmap for Affirm’s consumer loyalty program, launching rewards, tiers, engagement loops, and redemption experiences. Lead program economics, experimentation, incrementality, LTV measurement, and reward optimization. Partner with Risk, Credit, Legal, Compliance, Finance, Marketing, engineering, and design to deliver compliant products and drive repeat usage, retention, share of wallet, and incremental GMV.
Top Skills:
AmplitudeClaude CodeCursorSQL
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Architect and own marketing data infrastructure, Databricks pipelines, Gold Layer tables, semantic models, and AI-enabled data products. Lead complex cross-team initiatives, translate ambiguous business needs into scalable solutions, ensure data quality and reliability, and mentor engineers and analysts. Build, evaluate, and monitor production LLM and agent systems while automating marketing and go-to-market workflows with AI. Partner with BI, Data Science, Marketing, Sales, and R&D teams to support self-service analytics and trusted business KPIs.
Top Skills:
Ai AgentsBigQueryClaude CodeCursorData WarehousesDatabricksDbtEltETLLakehouse ArchitecturesLlmsPythonSalesforceSemantic LayersSnowflakeSQLStreamlit
4 Hours Ago
Easy Apply
Easy Apply
Big Data • Fintech • Mobile • Payments • Financial Services
Leads Marketing and Consumer Privacy Engineering teams, developing engineers and establishing a shared operating model. Translates business and product priorities into engineering plans across logged-out web, experimentation, marketing integrations, attribution, measurement, consent, privacy experiences, and consumer data collection. Partners with Product, Marketing, Privacy, Legal, and Engineering stakeholders to deliver reliable, scalable, privacy-aware systems while balancing customer needs, quality, speed, and risk.
Top Skills:
AnalyticsAttributionAustralian Privacy ActCcpa/CpraConsent ManagementCookiesGdprInstrumentationMarketing TechnologyMeasurement SystemsTaggingWeb Experimentation
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.
