Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Staff ML Engineer

Home > Python programming jobs

Staff ML Engineer in Canada new

  • Robots and Pencils
  • Full time
  • Email
  • Remote

Responsibilities

  • Train and operate models on Amazon SageMaker using AWS Trainium compute.
  • Write and optimize PyTorch training code for Trainium compilation and execution.
  • Diagnose hardware-, compiler-, device-, data-, and code-related training issues.
  • Build cost-aware, end-to-end Trainium training pipelines.
  • Tune distributed training runs for throughput and cost on SageMaker infrastructure.
  • Work directly with client and internal engineering teams to deliver production training workloads.

Requirements

  • Strong hands-on PyTorch experience, ideally with distributed or multi-device training.
  • Production experience using Amazon SageMaker for training and/or inference.
  • Understanding of device-specific compilation and the ability to debug accelerator-related issues.
  • AWS Trainium or Inferentia and Neuron SDK experience is preferred.
  • Deep PyTorch experience and the ability to learn new hardware targets quickly may substitute for direct Trainium experience.
  • Solid Python fundamentals and comfort working in a client-facing production engineering environment.

Robots and Pencils

Robots & Pencils is an applied AI engineering firm that designs, builds, and runs production AI systems and agentic co-workers for enterprise operations, with a focus on AWS. The privately held company, founded in 2009 and headquartered in Chagrin Falls, Ohio, provides consulting and implementati...

Similar positions

Senior Data Engineer - Finance

  • Circle
  • Full time
  • USA
  • 11/04/2026
  • Salary: $152k-$205k
  • Remote

Staff Software Engineer - Developer Experience

  • Addepar
  • Full time
  • USA
  • 11/04/2026
  • Salary: $158k - $248k/yr
  • Remote

Senior Software Engineer, GRC

  • Reddit
  • Full time
  • USA
  • 11/04/2026
  • Salary: $191k - $267k/yr
  • Remote

Senior Software Engineer

  • Turing
  • Full time
  • Others
  • 11/04/2026
  • São Paulo, Brazil

Senior Machine Learning Engineer

  • Adyen
  • Full time
  • USA
  • 11/04/2026
  • Salary: $228k - $317k/yr
  • San Francisco, CA

Listings

by


Job Type

Language

Country

CodingJobBoard.com © 2026