“PyTorch Profiler” jobs in Canada – Page 5

No matching filters. Try another name.

Results

• 29d

  Senior / Staff ML Training Optimization Engineer

  Toronto, ON

  HybridFull Time

  

  Waabi

  4+ YOECandidates need a Bachelor's, MS or PhD in computer science, robotics or a similar field, plus at least 4 years of industry experience. Strong coding in Python, C++ or Rust, deep learning experience with PyTorch or JAX, and skill profiling CPU and GPU code are required.

  PyTorch, JAX, CUDA, NVIDIA Nsight, PyTorch Profiler, Bazel

  29d

  Senior / Staff ML Onboard Optimization Engineer

  Toronto, ON

  HybridFull Time

  

  Waabi

  6+ YOEWaabi needs candidates with a Bachelor's, Master's or PhD in Computer Science, Robotics or a similar field and at least 6 years of industry experience. They should code well in Python, C++ or Rust, know deep learning frameworks like PyTorch, and be skilled at profiling CPU and GPU code.

  TensorRT, modelopt, CUDA, PyTorch, PyTorch Profiler, NVIDIA Nsight

• 29d

  Hardware, Edge ML Engineer

  Richmond Hill, ON

  $170k–$190k/yrHybridFull Time

  

  Mecka

  5+ YOECandidates need 5+ years in machine learning or computer vision, including deploying models on embedded or limited hardware. Strong Python and C++ skills, PyTorch experience and embedded Linux experience are required.

  PyTorch, C++, Python, Embedded Linux

• 29d

  ML Runtime and Kernel Engineer - Core ML

  Toronto, ON

  HybridFull Time

  

  Cerebras Systems

  Cerebras Systems needs candidates with a background in computer science, computer engineering or electrical engineering, plus experience building high-performance systems, ML systems, compilers or kernels. Strong C++ and Python skills, and familiarity with PyTorch or JAX, are required.

  C++, Python, PyTorch, JAX, CUDA, Triton

  29d

  Staff GPU Inference SDET

  Toronto, ON

  HybridFull Time

  

  Cerebras Systems

  8+ YOECerebras Systems needs 8+ years of software test or infrastructure quality experience. Candidates must have hands-on GPU cluster bring-up experience, knowledge of LLM serving engines, expert Python and strong Kubernetes, Slurm or Ray skills.

  Python, Kubernetes, Slurm, Ray, Prometheus, Grafana

  29d

  Staff Software Engineer, GPU Inference

  Toronto, ON

  HybridFull Time

  

  Cerebras Systems

  8+ YOECandidates need 8+ years of software engineering experience with ownership of complex production systems. Strong C++ and Python, LLM inference serving experience (such as vLLM or TensorRT-LLM) and GPU performance knowledge are required.

  vLLM, ROCm, HIP, PyTorch, Kubernetes, AMD Instinct GPUs

  29d

  Software Engineer, GPU Inference

  Toronto, ON

  On-siteFull Time

  

  Cerebras Systems

  5+ YOECerebras Systems needs candidates with 5+ years of software engineering experience, including production inference work for large models. Strong C++ and Python, hands-on serving framework experience such as vLLM, and deep GPU performance knowledge are required.

  vLLM, ROCm, PyTorch, Kubernetes, AMD Instinct GPUs, RCCL

  29d

  Senior Runtime Engineer

  Canada

  Full Time

  

  Cerebras Systems

  3+ YOECandidates need 3+ years building high-performance or distributed systems software, strong C/C++ skills, and experience with multi-threading and memory management. A Bachelor's, Master's, or equivalent experience in Computer Science, Electrical Engineering, or a related field is required.

  C, C++, PyTorch, Python

  29d

  Full Stack LLM Engineer

  Toronto, ON

  HybridFull Time

  

  Cerebras Systems

  Candidates need a degree in computer science, engineering or a related field, plus experience with deep learning frameworks, C/C++ low-level optimization and compiler development such as LLVM or MLIR. Strong debugging and optimization skills, including work on NP-hard problems, are also expected.

  PyTorch, TensorFlow, LLVM, MLIR, C++

• 29d

  Machine Learning Specialist

  Montréal, QC

  

  DRW

  5+ YOECandidates need a bachelor's or master's degree in computer science or software engineering, plus 5+ years building software for machine learning systems. Strong programming skills, deep learning knowledge and production MLOps experience are essential.

  TensorFlow, PyTorch, TensorRT, ONNX, CUDA, OpenCL

• 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

• 29d

  Member of Technical Staff, Agentic Environments

  Remote in Canada

  $295k–$535k/yrRemoteFull Time

  

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

  Candidates need extremely strong software engineering skills, proficiency in Python and ML frameworks like JAX or PyTorch, and experience debugging large-scale distributed training. Experience with Kubernetes, Ray and ML or LLM academic research is also required.

  Python, PyTorch, JAX, Kubernetes, Ray

  29d

  Member of Technical Staff, Integration/RL Team (Research Engineer)

  Toronto, ON +1

  $295k–$535k/yrRemoteFull Time

  

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

  Candidates need very strong software engineering skills, proficiency in Python and ML frameworks such as JAX, PyTorch or XLA/MLIR, and experience debugging large-scale distributed training. Test-driven development and clean code habits are valued. Experience with Kubernetes, Ray, post-training and ML or RL research are bonus qualifications.

  Python, JAX, PyTorch, XLA/MLIR, Kubernetes, Ray

  29d

  Member of Technical Staff, Training Performance Engineer

  Toronto, ON +1

  $295k–$535k/yrHybridFull Time

  

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

  Candidates need extremely strong software engineering skills, proficiency in Python and ML frameworks such as JAX, PyTorch and XLA/MLIR, and experience writing GPU kernels with CUDA or Triton. Experience with large-scale distributed training and transformer sequence models is also required.

  CUDA, Triton, PyTorch, JAX, XLA, MLIR

6 companies