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
Fintech • Financial Services
Lead technical architecture and roadmap for analytics engineering: design semantic/metrics layer, build AI-ready Snowflake data foundation, optimize dbt projects, mentor engineers, partner with Data Science/Engineering, and champion data governance, observability, and production readiness for AI and downstream analytics.
Top Skills:
Agent FrameworksCi/CdDbtFivetranKafkaObservability ToolsPythonRagSnowflakeSnowflake CortexSnowflake IntelligenceSQLStreaming/Cdc
Fintech • Financial Services
Design, build, and maintain scalable dimensional data models, semantic/metrics layers, and production data pipelines (dbt, Snowflake). Partner with Data Science, Data Engineering, BI, and DevOps to enable AI/ML, real-time analytics, governance, and data quality while mentoring junior engineers and enforcing best practices.
Top Skills:
CortexDatafoldDbtFivetranKafkaLlmPythonRagRedshiftSnowflakeSnowflake IntelligenceSQLStreamkapTableau
Fintech • Financial Services
Lead technical architecture and roadmap for analytics engineering, build and govern semantic/metrics layers, optimize dbt and Snowflake performance, mentor engineers, partner with Data Science and Data Engineering to deliver AI-ready data foundations and ensure data quality and governance.
Top Skills:
DbtSnowflakeSQL
What you need to know about the Toronto Tech Scene
Although home to some of the biggest names in tech, including Google, Microsoft and Amazon, Toronto has established itself as one of the largest startup ecosystems in the world. And with over 2,000 startups — more than 30 percent of the country's total startups — Toronto continues to attract new businesses. Be it helping entrepreneurs manage their finances, simplifying business operations by automating payroll or assisting pharmaceutical companies in launching new drugs, the city's tech scene is just getting started.
