Job Information
Amazon ML Engineer, Annapurna ML in Cupertino, California
Description
Annapurna ML pathfinding team is a new function within the Annapurna ML go-to-market org that help customers accelerate their adoption of Annapurna ML products including AWS Trainium and AWS Inferentia. The team offers hands-on data science and coding services to our most strategic customer opportunities to launch their training and inference workloads on AWS purpose built ML silicon offerings.
Key job responsibilities
You will be responsible for helping customers port their models to the AWS Trainium & Inferentia platforms by delivering code and customizations to make the models functional and performant. You will use and provide feedback to the various Neuron SDK libraries and develop new features based on the latest research findings and customer requests.
A day in the life
You will be required to support customers to develop and port their large scale models to AWS Trainium and Inferentia. You will work directly with customer data scientists and ML engineering teams to port their models to our HW platform and make the models performant by delivering SW features on top of the Neuron SDK. It will require a close collaboration with the Neuron engineering team to help drive the Neuron product roadmap and give feedback on improving product quality.
About the team
Our team's mission is to provide the fastest, cost-effective and user-friendly place to train and deploy Generative AI workloads in the cloud. The team provides white-glove service to our most strategic customers to implement their models for both training and inference using the Neuron SDK associated libraries and APIs.
We are open to hiring candidates to work out of one of the following locations:
Cupertino, CA, USA | Seattle, WA, USA
Basic Qualifications
3+ years of machine learning, statistical modeling, data mining, and analytics techniques experience
Experience programming with at least one software programming language
Preferred Qualifications
- Knowledge of professional software engineering & best practices for full software development life cycle, including coding standards, software architectures, code reviews, source control management, continuous deployments, testing, and operational excellence
Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, protected veteran status, disability, age, or other legally protected status. For individuals with disabilities who would like to request an accommodation, please visit https://www.amazon.jobs/en/disability/us.
Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $129,300/year in our lowest geographic market up to $223,600/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job-related knowledge, skills, and experience. Amazon is a total compensation company. Dependent on the position offered, equity, sign-on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits. For more information, please visit https://www.aboutamazon.com/workplace/employee-benefits. This position will remain posted until filled. Applicants should apply via our internal or external career site.