SDE, Neuron Explorer: AI Performance Profiling

Amazon; Toronto; On-site; CAD 115,000 - 192,000; Full time

Job summary

Amazon in Toronto is seeking a Software Development Engineer for the Neuron Explorer team to design, build, and operate the full stack behind Neuron Explorer—from data collection to visualizations and CLI workflows.

You will own features end to end, collaborating with compiler, runtime, and chip architecture teams to surface actionable insights for training workloads on Trainium and Inferentia. Strong ML/LLM familiarity is valued.

Qualifications

• Bachelor's degree in Computer Science, Engineering, Mathematics, or a related field.
• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
• Experience programming with at least one software programming language.

Responsibilities

• Design and build performance data collection pipelines and analysis engines that transform raw hardware execution data into actionable insights for Neuron Explorer users.
• Develop interactive visualizations and developer workflows that make complex profiling data understandable, helping customers pinpoint bottlenecks across single-node and distributed training workloads.
• Own features end to end across the full software development lifecycle — from scoping and design through implementation, testing, deployment, and operations — ensuring reliability and maintainability.
• Collaborate with compiler, runtime, and chip architecture teams to surface the right performance information so internal and external customers can optimize AI workloads on Trainium and Inferentia hardware.
• Evaluate and integrate generative AI capabilities into Neuron Explorer's recommendation engine, building features that automatically identify optimization opportunities and guide customers toward higher performance.

Skills

Professional software development

System design

Programming experience

Education

Bachelor's degree in Computer Science, Engineering, Mathematics

Tools

PyTorch

JAX

Job description

Amazon in Toronto is seeking a Software Development Engineer for the Neuron Explorer team to design, build, and operate the full stack behind Neuron Explorer—from data collection to visualizations and CLI workflows.

You will own features end to end, collaborating with compiler, runtime, and chip architecture teams to surface actionable insights for training workloads on Trainium and Inferentia. Strong ML/LLM familiarity is valued.