You will join a dynamic team working in the GenaAI revolution by applying AI to AI. You will work building agents, tools, and models to simplify and accelerate customer adoption of Neuron, the software stack supporting Amazon's Machine Learning silicon: Trainium. You will work with external and internal customers to identify the main obstacles and the opportunities to accelerate their adoption of the Neuron technology. You will be a key member of the team driving our vision and strategy in this space critical to AWS's Generative AI business. Will also lead a team building AI agents and tools that simplify AWS Neuron adoption for Machine Learning developers working with Trainium chips, partnering with external and internal customers to identify obstacles and accelerate their migration to AWS's ML silicon.
Key job responsibilities
This is a highly visible role that requires partnering with other Neuron Software teams, Applied Science, AWS AI Services, external partners and customers with a potential high impact on AWS's top and bottom line. As a member of the team applying Generative AI to accelerate Neuron adoption, you will play a key role in shaping this space with the following technical and leadership responsibilities:
* Collaborate with scientists, engineers, product managers and executive leadership to create the technical vision and roadmap for agents and tools used by internal and external Machine Learning developers.
* Serve as a technical lead on demanding, cross-functional projects critical to AWS's future success in the AI space..
* Deliver on ambitious goals to improve the time and effort it takes to port and optimize Machine Learning workloads on Neuron.
* Contribute intellectual property through patents.
* Hire and assist in the career development of your team members, actively mentoring individuals on advanced technical issues and their career growth.
* Contribute to the definition of best practices for the software development lifecycle used by junior engineers, including its design, implementation, testing, and operational characteristics.
About the team
The Neuroboros team was recently created to pursue the ambitious goal of leveraging and expanding Generative AI technologies to help customers benefit from the scale and price/performance equation offered by Amazon Machine Learning hardware. The creation of the team in NYC is key to Annapurna Labs location strategy, with the goal of creating an additional hub attracting top talent with varied backgrounds to work on challenging problems, using and building state-of-the-art tooling.
BASIC QUALIFICATIONS
- 10+ years of engineering experience
- 5+ years of engineering team management experience
- 10+ years of planning, designing, developing and delivering consumer software experience
- Experience partnering with product or program management teams
- Experience managing multiple concurrent programs, projects and development teams in an Agile environment
- Experience in one or more of the following areas ML compilers, production coding agents, GenAI model architecture, model training, neural network optimization. or alternatively applied math
PREFERRED QUALIFICATIONS
- Proven track record building AI agents that automate ML workload optimization
- ML compiler tuning
- Distributed inference and training
- ML kernel authoring and optimization
The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.
USA, NY, New York - 242,100.00 - 327,500.00 USD annually
