AI & Emerging technologies

Senior Machine Learning Software Engineer

Amazon Inc.

Toronto · Ontario · Canada

Share this job: X Facebook WhatsApp

Develop AWS Neuron software for machine learning acceleration in Toronto, combining backend systems, web interfaces, CLI workflows, and performance analysis.

As a Senior Software Development Engineer, you will work on Neuron Explorer, which profiles, debugs, and analyzes workloads on Trainium and Inferentia. The role connects source code with hardware operations across single-node and distributed applications. You will help define product direction while delivering customer-facing capabilities.

Key Responsibilities:

  • Design engines analyzing execution data from AI hardware
  • Build backend, web UI, and CLI features
  • Trace workloads across single-node and distributed applications
  • Partner with compiler, runtime, and chip architecture groups
  • Identify optimization opportunities with recommendation features

Requirements:

  • Five or more years of professional software development experience
  • Experience as mentor, technical lead, or team leader
  • Bachelor's degree in computer science or equivalent
  • Machine learning and LLM architecture fundamentals preferred
  • PyTorch, JIT, and AOT tracing experience preferred

Use machine learning and full-stack development skills to advance Neuron Explorer and performance optimization on AWS accelerators.

#J-18808-Ljbffr

Reference: WJ-4483_1571100 · Posted 11 Nov 2026, 03:33 PM

Find more on GigBlow

This role is listed on GigBlow for discovery and search. Hiring decisions and applications are handled by the employer or their chosen application system.

Machine Learning jobs in other states

Browse jobs