“JAX” jobs in Canada – Page 4

No matching filters. Try another name.

Results

• 1mo

  Research Scientist, Neural Reconstruction

  Toronto, ON

  HybridFull Time

  

  Waabi

  Candidates need a PhD in computer vision, machine learning, robotics or a related field, or equivalent research experience. Expert Python and PyTorch or JAX skills, distributed training experience, and a record of team collaboration are required.

  Research, Machine learning, Robotics, Python, PyTorch, Communication skills

  1mo

  Research Engineer, World Models

  Toronto, ON

  RemoteFull Time

  

  Waabi

  Candidates need a Master's degree in computer vision, machine learning, robotics or a related field, or equivalent industry experience. Strong Python and PyTorch or JAX skills, distributed training experience and hands-on work on generative or predictive models of the physical world are required.

  PyTorch, JAX, Kubeflow, Ray, ONNX, TensorRT

• 1mo

  Senior Machine Learning Engineer, AI Generation Engine

  Remote in Canada

  $140k–$210k/yrRemoteFull Time

  

  SandboxAQ: SandboxAQ is a B2B company delivering AI solutions that address some of the world's greatest challenges.

  5+ YOECandidates need a BS in Software Engineering, Computer Science or a related field, and 5+ years of postgraduate software experience. They need strong Python skills with PyTorch, TensorFlow, JAX, NumPy or Pandas, and proven experience taking ML models to scalable production.

  PyTorch, TensorFlow, JAX, NumPy, Pandas, Weights & Biases

  1mo

  Staff Machine Learning Engineer, AI Generation Engine

  Remote in Canada

  $172k–$258k/yrRemoteFull Time

  

  SandboxAQ: SandboxAQ is a B2B company delivering AI solutions that address some of the world's greatest challenges.

  8+ YOECandidates need a computer science or software engineering background, 8+ years of postgraduate software experience, and a record of building scalable ML systems. Strong Python skills with PyTorch, TensorFlow, JAX, NumPy and Pandas, plus MLOps practice, are essential.

  PyTorch, TensorFlow, JAX, NumPy, Pandas, Weights & Biases

• 1mo

  Senior Machine Learning Engineer - (Biometrics)

  Montréal, QC

  J

  Jumio

  5+ YOECandidates need 5+ years of machine learning experience, with at least 3 years in biometrics or face analysis. Strong Python, PyTorch, cloud deployment on AWS, and knowledge of algorithmic bias are required.

  PyTorch, TensorFlow, JAX, Airflow, OpenCV, Pillow

  1mo

  Senior Machine Learning Engineer

  Montréal, QC

  J

  Jumio

  5+ YOECandidates need 5+ years of industry machine learning experience focused on computer vision, strong Python and PyTorch skills, and experience with ML pipelines, Airflow and GPU-based training on AWS. Awareness of algorithmic bias is expected.

  PyTorch, TensorFlow, JAX, Airflow, AWS, TensorRT

  1mo

  Machine Learning Engineer - (Computer Vision)

  Montréal, QC

  J

  Jumio

  Candidates need deep computer vision and biometrics experience, especially face recognition, plus strong Python and production ML engineering skills. Experience with AWS cloud deployment, multi-GPU training and fairness measurement is also needed.

  PyTorch, TensorFlow, JAX, Airflow, OpenCV, Pillow

• 1mo

  Machine Learning Engineer - Enterprise

  Toronto, ON

  $150k–$400k/yrOn-siteFull Time

  

  Boson AI

  Candidates need a Bachelor's or Master's in computer science, ML, AI or a related field, or equivalent experience. Boson AI requires experience with large language or multimodal models, search systems, Python-style programming and ML frameworks, plus a public GitHub record.

  PyTorch, JAX, GitHub

• 1mo

  Senior Research Scientist

  Toronto, ON

  RemoteFull Time

  

  adaption

  4+ YOECandidates need a PhD or equivalent research experience in computer science, 4-5+ years in industry and a publishing record at ICML, NeurIPS, ACL, ICLR or EMNLP. They need deep expertise in model efficiency, real-time alignment or algorithmic optimization, and strong Python skills with PyTorch, JAX or TensorFlow.

  PyTorch, JAX, TensorFlow, Python

  1mo

  Research Scientist

  Toronto, ON

  RemoteFull Time

  

  adaption

  4+ YOECandidates need a PhD or equivalent research experience in computer science and 4-5+ years of industry experience. They should have deep expertise in model efficiency, real-time alignment or algorithmic optimization, and strong Python skills.

  PyTorch, JAX, TensorFlow, Python

  1mo

  Modelling Resident

  Toronto, ON

  RemoteContract

  

  adaption

  Candidates need a degree or equivalent research or engineering experience in computer science, strong Python skills, and experience with deep learning frameworks such as PyTorch, JAX or TensorFlow. Familiarity with RLHF and fine-tuning and a project in model efficiency, synthetic data or real-time alignment are valued.

  PyTorch, JAX, TensorFlow, Python

• 1mo

  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

• 1mo

  Senior ML Research Scientist

  Montréal, QC

  

  LawZero

  5+ YOECandidates need an advanced degree in a relevant field, ideally a PhD, and 5+ years of deep learning research, including frontier models. Strong Python, PyTorch, TensorFlow or JAX skills and clear written and verbal communication are also required.

  PyTorch, TensorFlow, JAX, Python

  1mo

  Senior ML Research Developer

  Montréal, QC

  

  LawZero

  5+ YOECandidates need a computer science-related degree, an MSc or higher in machine learning or equivalent experience, and 5+ years of industry experience building complex ML workflows on high performance computing. Cloud, containerization, data infrastructure and deep learning research experience are also expected.

  PyTorch, TensorFlow, JAX, AWS, GCP, Azure

• 1mo

  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

  1mo

  Machine Learning Intern/Co-op (Winter 2027)

  Remote in Canada

  RemoteInternship

  

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

  Cohere needs candidates currently enrolled in a post-secondary program and available for a full-time 3-6 month term. Required skills include Python, ML frameworks such as TensorFlow and JAX, distributed training, and Transformer-based models.

  Python, TensorFlow, TF-Serving, JAX, XLA/MLIR, CUDA

  1mo

  Research Internship (Winter 2027)

  Toronto, ON +1

  RemoteInternship

  

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

  Candidates need to be pursuing or completing a PhD in machine learning, NLP or a related field, with experience in distributed training, transformers and Python, and strong communication skills. Cohere will also consider exceptional non-PhD candidates.

  PyTorch, JAX, TensorFlow, Python, C++

  1mo

  Senior Member of Technical Staff, Safety and Security for Agents

  Toronto, ON +1

  $295k–$535k/yrHybridFull Time

  

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

  Candidates need strong statistical skills, extremely strong software engineering skills, and hands-on experience training LLMs on distributed infrastructure. Experience with human annotation, dataset quality analysis, Python and ML frameworks, plus top-tier publications, is expected.

  Python, PyTorch, TensorFlow, JAX

  1mo

  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

  1mo

  Member of Technical Staff, Data Analysis and Evaluation

  Toronto, ON +1

  $295k–$535k/yrRemoteFull Time

  

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

  Cohere needs candidates with extremely strong software engineering skills, statistical expertise in evaluating experiments, and experience designing data collection with human annotators. Candidates also need hands-on LLM training on distributed infrastructure, proficiency in Python and ML frameworks, and strong communication skills. One or more papers at top-tier venues are listed as a qualification.

  Python, PyTorch, TensorFlow, JAX

  1mo

  Member of Technical Staff, Post-Training

  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 like JAX and PyTorch, and hands-on experience training large models with distributed infrastructure. Post-training experience with a focus on performance optimisation is essential.

  Kubernetes, Slurm, Ray, JAX, PyTorch, XLA

  1mo

  Member of Technical Staff, Training Infra Engineer

  Toronto, ON +1

  $295k–$535k/yrRemoteFull Time

  

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

  Candidates need strong software engineering skills, proficiency in Python and ML frameworks such as JAX, PyTorch and XLA/MLIR, and experience with distributed training on Kubernetes, Slurm and Ray. Hands-on experience training large models at scale and contributing to training infrastructure is required.

  Python, JAX, PyTorch, XLA, MLIR, Kubernetes

  1mo

  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

  1mo

  Senior Member of Technical Staff, Multimodal AI

  Toronto, ON +1

  $295k–$535k/yrRemoteFull Time

  

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

  Candidates need exceptional software engineering skills, strong Python, and experience with deep learning frameworks such as JAX, PyTorch and TensorFlow. Experience with distributed training, autoregressive models and evaluating large multimodal models is also expected.

  Python, JAX, PyTorch, TensorFlow, CUDA, HuggingFace

8 companies