Amazon Devices is an inventive research and development company that designs and engineer high-profile devices like the Kindle family of products, Fire Tablets, Fire TV, Health Wellness, Amazon Echo & Astro products. This is an exciting opportunity to join Amazon in developing state-of-the-art techniques that bring Gen AI on edge for our consumer products. We are looking for exceptional scientists to join our Applied Science team and help develop the next generation of edge models, and optimize them while doing co-designed with custom ML HW based on a revolutionary architecture. Work hard. Have Fun. Make History.


Key job responsibilities
What will you do?
* Developing End to End pipelines from model training to model running on device with exception automation.
* Building tools and dashboards to auto test these pipelines and provide health metrics
* Scaling to more H100 clusters, Trainiums, and test frameworks like Pytorch ligtining
* Simplifying and reinventing systems, processes, and tools to make ML pipelines better for our Scientists, System Integration teams, Device OS teams
* Investigating technical issues scientifically and thoroughly, and assist in fixing them so they don't come back
* Providing technical solutions to real business problems in a global organization