ML Kernel Performance Engineer, Edge AI and Science (Winnipeg)
This job is in your area. Enjoy a short commute and work close to home.
Job Description
Amazon Devices is an inventive research and development company that designs and engineers high‑profile consumer products such as the Kindle family, Fire Tablets, Fire TV, Health & Wellness devices, Amazon Echo, and Astro. We are building the next generation of edge AI capabilities through our advanced compression platform and custom neural accelerator silicon.
Within Edge AI & Science, the AI Platform team builds a compression platform—the first of its kind—enabling 20‑100x neural network compression for edge and cloud deployment. As model sizes grow from billions to hundreds of billions of parameters, compute efficiency becomes the single largest return on engineering investment during training. The gap between eager‑mode Python and optimized GPU execution is where months of training time are won or lost.
About The Team
The AI Platform team builds Amazon’s neural network compression platform. We compress models using knowledge distillation, network restru...