“FSDP” jobs in Canada

No matching filters. Try another name.

Results

• 3d

  Senior AI System Engineer - LLM Training System

  Vancouver, BC

  On-siteFull Time

  

  Huawei Technologies Canada Co., Ltd.

  3+ YOECandidates need a Bachelor's, Master's or PhD in a technical discipline and 3+ years in HPC, AI accelerator operators or distributed large-model training. Solid skill in AscendC or CUDA C/C++, C++ and Python, and experience with distributed training frameworks and profiling tools are required.

  AscendC, CANN, Triton, TileLang, MindSpeed, Megatron-LM

  27d

  Researcher - LLMs

  Montréal, QC

  On-siteContract

  

  Huawei Technologies Canada Co., Ltd.

  Candidates need a Master's or PhD in computer science or a related field, or equivalent practical experience. They need strong NLP and LLM knowledge, expert Python and PyTorch skills, and experience with LLM training and inference tools. English proficiency is essential.

  Python, PyTorch, Hugging Face Transformers, Accelerate, Megatron-LM, vLLM

  27d

  Co-op Researcher – LLM Post-Training & Agentic Coding

  Edmonton, AB

  On-siteInternship

  

  Huawei Technologies Canada Co., Ltd.

  Candidates need to be enrolled in a senior co-op program in computer science, machine learning or AI. They need strong Python and PyTorch skills, solid knowledge of large language models, and hands-on experience with ML research or systems.

  PyTorch, vLLM, SGLang, FSDP, DeepSpeed, Hugging Face

• 22d

  Senior AI Machine Learning Engineer

  Montréal, QC

  Full Time

  S

  SAP

  5+ YOECandidates need a PhD or MS in a related field, 5+ years delivering ML systems end to end, and hands-on work with PyTorch and PyTorch Geometric or DGL. Strong Python, distributed systems experience and cloud scaling on Azure or AWS are also required.

  PyTorch, PyTorch Geometric, Deep Graph Library, Azure, AWS

• 23d

  AI Systems Engineer – AI Model (Training & Inference)

  Markham, ON

  Full Time

  AI

  AMD, Inc.

  Candidates need hands-on experience shipping AI/ML infrastructure, training and running large models on GPUs, and building GPU kernels. A Bachelor's, Master's or PhD in computer or software engineering or computer science is preferred.

  AMD Instinct MI-series GPUs, RCCL, Kubernetes, PyTorch, vLLM, SGLang

• 23d

  Senior ML Researcher - Reconstruction

  Toronto, ON

  $250k–$300k/yrOn-siteFull Time

  P

  Peripheral

  Candidates need a Master's or PhD in a related field and a proven record leading 3D reconstruction or neural rendering research into production. Strong expertise in multi-view geometry, NeRFs, Gaussian Splatting, distributed GPU training and team leadership is required.

  PyTorch, CUDA, NVIDIA Isaac Sim, NVIDIA Omniverse, NVDEC/NVENC

• 24d

  Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)

  Toronto, ON

  $295k–$535k/yrRemoteFull Time

  

  Cohere: Cohere is the leading security-first enterprise AI company.

  Candidates need strong research and software engineering skills, hands-on experience training LLMs, and experience applying reinforcement learning to LLMs. A record of work on generalization, robustness or distribution shift, and rigorous experimental practice are also required.

  JAX, PyTorch, FSDP, Megatron

• 28d

  Ingénieur ou ingénieure de recherche en IA

  Toronto, ON

  

  Deloitte

  3+ YOEAu moins trois ans en ingénierie de l’apprentissage machine ou en recherche appliquée; Solides compétences en Python et en PyTorch, TensorFlow ou JAX; Expérience avec Hugging Face Transformers et la bibliothèque TRL

  Python, PyTorch, TensorFlow, JAX, Hugging Face, vLLM

  28d

  AI Research Engineer

  Toronto, ON

  

  Deloitte

  3+ YOECandidates need 3+ years in machine learning, data engineering or applied research, with strong Python and deep learning framework skills. Hands-on experience with Hugging Face, reinforcement learning fine-tuning, distributed GPU training and LLM serving is required.

  Hugging Face Transformers, Hugging Face TRL, PyTorch, vLLM, DeepSpeed, Weights & Biases

  28d

  Conseiller(ère) principal(e)- Ingénierie IA

  Montréal, QC

  

  Deloitte

  3+ YOEAu moins trois ans d'expérience en apprentissage machine ou recherche appliquée; Maîtrise de Python et de PyTorch, TensorFlow ou JAX; Expérience avec Hugging Face Transformers et TRL

  Hugging Face, PyTorch, TensorFlow, JAX, DeepSpeed, vLLM

  28d

  Conseiller(ère) principal(e) - Ingénierie IA

  Montréal, QC

  

  Deloitte

  3+ YOEAu moins trois ans en apprentissage machine, ingénierie des données ou recherche appliquée; Maîtrise de Python et de PyTorch, TensorFlow ou JAX; Expérience avec Hugging Face Transformers et TRL

  Hugging Face Transformers, TRL, DeepSpeed, vLLM, PyTorch, Weights & Biases

• 28d

  AI Systems Engineer – AI Model (Training & Inference)

  Markham, ON

  Full Time

  AM

  Advanced Micro Devices, Inc

  Candidates need hands-on experience shipping LLMs on real hardware and writing GPU kernels that improved production performance. They also need experience building distributed AI infrastructure, plus a relevant bachelor's, master's or PhD in computer or software engineering or computer science.

  HIP, Triton, MLIR, ROCm, RCCL, Kubernetes

• 29d

  Senior Research Engineer/Scientist

  Montréal, QC

  On-siteFull Time

  

  ServiceNow

  6+ YOE6+ ans d'expérience avec un baccalauréat, ou 4+ ans avec une maîtrise ou un doctorat; Expertise en post-entraînement des LLM et en apprentissage par renforcement (PPO, GRPO, DPO); Expérience en entraînement distribué (FS

  Python, FSDP, DeepSpeed, Megatron, vLLM, SGLang

• 29d

  Ingénieur.e sénior en apprentissage automatique distribué

  Montréal, QC

  

  LoiZéro

  3+ YOEDiplôme en informatique ou génie logiciel requis; maîtrise ou doctorat préféré; Plus de 3 ans d'expérience en frameworks d'entraînement ML distribué; Expérience avec Megatron, DeepSpeed, FSDP, vLLM ou similaires

  Megatron, DeepSpeed, HuggingFace Accelerate, FSDP, vLLM, verl

• 29d

  Senior / Staff ML Ops Engineer

  Toronto, ON

  HybridFull Time

  

  Waabi

  5+ YOEWaabi needs candidates with 5+ years of software or infrastructure engineering, excellent Python and experience building APIs and CLIs. Candidates also need practical AWS and infrastructure-as-code skills, plus PyTorch distributed training and experiment-tracking experience.

  Kubernetes, PyTorch, AWS, Terraform, Pulumi, Argo Workflows

• 29d

  Senior AI Compute Infrastructure Engineer

  Remote in Canada

  RemoteFull Time

  

  Kraken

  5+ YOECandidates need 5+ years of infrastructure engineering experience, including GPU or ML infrastructure in production. They also need Kubernetes and Linux skills, Python, experience with serving frameworks such as vLLM or Triton, and a track record of controlling compute costs.

  Kubernetes, Linux, Python, vLLM, Triton Inference Server, TensorRT

• 29d

  Applied Machine Learning Research Scientist

  Toronto, ON

  HybridFull Time

  

  Cerebras Systems

  4+ YOECandidates need a Bachelor's or Master's degree in a related field and 4+ years of ML experience. Strong Python and PyTorch skills, transformer knowledge and the ability to implement ideas from ML papers are required.

  PyTorch, Python

• 29d

  Senior Distributed ML Engineer

  Montréal, QC

  

  LawZero

  3+ YOECandidates need a relevant computer science degree and 3+ years designing distributed ML training frameworks, including recent use of Megatron, DeepSpeed, FSDP or vLLM. Experience with cloud platforms, Ray or SLURM, GPU profiling, Docker or Kubernetes, and deep learning research is also expected.

  Megatron, DeepSpeed, HuggingFace Accelerate, FSDP, vLLM, verl

13 companies