3 days ago
Toronto, CanadaSenior
Responsibilities
- Design and build performance data collection pipelines and analysis engines that transform raw hardware execution data into actionable insights.
- Develop interactive visualizations and developer workflows for analyzing profiling data across single-node and distributed training workloads.
- Own features across scoping, design, implementation, testing, deployment, and operations.
- Build across the backend processing pipeline, web UI, and CLI for Neuron Explorer.
- Collaborate with compiler, runtime, and chip architecture teams to support optimization of AI workloads on Trainium and Inferentia.
- Evaluate and integrate generative-AI capabilities into the recommendation engine to identify optimization opportunities and guide customers.
- Partner with technical leaders and management to define product direction and deliver features to customers.
Requirements
- 5+ years of non-internship professional software development experience.
- Experience as a mentor, tech lead, or engineering team leader.
- Bachelor's degree in computer science or equivalent.
- Preferred experience with machine learning and large language model fundamentals, including model architecture, training and inference lifecycles, and execution optimization.
- Preferred experience developing and deploying LLMs in production on GPUs, Neuron, TPU, or other AI acceleration hardware.
- Preferred experience with PyTorch, JIT compilation, and AOT tracing.
- Preferred 6+ years of full software development lifecycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
Benefits
- Health insurance including medical, dental, vision, prescription, basic life, and AD&D insurance.
- Registered Retirement Savings Plan (RRSP) and Deferred Profit Sharing Plan (DPSP).
- Paid time off and other health and well-being resources.
- Flexible working hours with an emphasis on work-life balance.
- Location: Toronto, Ontario, Canada.
Categories
About Amazon
Amazon builds and operates a global e-commerce marketplace, logistics network, and consumer devices, and provides cloud computing via AWS for businesses and developers. The company earns revenue from online retail, third‑party seller services, subscriptions like Prime, advertising, and AWS usage. Founded in 1994 and headquartered in Seattle, it is publicly traded on NASDAQ (AMZN) and serves customers in dozens of countries.