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.
Interested in learning more about this job Scroll down and find out what skills, experience and educational qualifications are needed.
Key job responsibilities
In this customer-facing role, you will be responsible for helping our most strategic customers port their models to the AWS Trainium & Inferentia platforms by delivering high-quality code and customizations to make the models functional and performant.
You will use and provide feedback to the various Neuron SDK libraries and help prototype and develop new features based on the latest research findings and customer requests.
A day in the life
You will be required to assist our most strategic customers in porting their models to AWS Trainium and Inferentia.
You will work directly with customer data scientists and ML engineering teams and write code to have the models be performant on AWS purpose-built silicon solutions.
It may require low-level coding in C++ and writing custom kernels to get the best performance possible.
You will also be responsible for porting the latest open-source models to AWS Trainium / Inferentia. You will also contribute to open-source projects to help add support for AWS Trainium / Inferentia in popular projects.
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.
BASIC QUALIFICATIONS
- 5+ years of non-internship professional software development experience
- 5+ years of programming using a modern programming language such as Java, C++, or C#, including object-oriented design experience
- 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Experience as a mentor, tech lead or leading an engineering team
- 3+ years of experience writing code to train and / or deploy deep learning models in PyTorch
PREFERRED QUALIFICATIONS
- Bachelor's degree in computer science or equivalent
- Experience deploying Generative AI applications with large language or vision models into production.
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 $151,300 / year in our lowest geographic market up to $261,500 / 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.
Posted : August 24, 2024 (Updated about 22 hours ago)
Posted : August 24, 2024 (Updated about 22 hours ago)
Posted : August 22, 2024 (Updated about 23 hours ago)
Posted : August 23, 2024 (Updated 1 day ago)
Important FAQs for current Government employees
Before proceeding, please review the following FAQs
https : / / www.amazon.jobs / en / faqs#faqs-for-us-government-employees
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.
J-18808-Ljbffr
AIML - Staff Machine Learning Infrastructure Engineer, ML Platform & Technology
Here, you'll do more than join something - you'll add something! As a Staff Engineer on ML Compute Team, you'll join a phenomenal team of hardworking engineers and will be entrusted with a range of responsibilities. We are looking for an experienced software engineer who is excited by the opportunit...
Sr Director, Fraud Risk Machine Learning
As Head of Fraud Risk ML, you will manage the entire portfolio of Fraud Risk solutions, and be entrusted with leading our globally distributed top-tier ML scientists to bring cutting-edge AI/ML risk solutions to fruition. Effectively engage and collaborate with cross-functional teams, including stra...
Senior Machine Learning Engineer - SIML, ISE
We are looking for an expert Machine Learning engineer, or Data Scientist who can help create and improve the datasets used in Generative AI through proven understanding and usage of ML and stats. In our team, you are encouraged to collaborate with other AIML product stakeholders and partners to und...
Senior Software Engineer, Perception, Machine Learning/Computer Vision
Our team is growing and we are looking for experienced machine learning researchers and engineers to join us. In this role, you will apply your expertise in machine learning to solve an array of challenges in Nuro's autonomy stack — detection with sensor fusion, tracking, fine grained classification...
Machine Learning Engineer, AI Solutions
Machine Learning Engineer, AI Solutions. Would you like to be part of a world class team enabling Machine Learning applications for world class datacenters and the mightiest supercomputers? AMD is searching for talented and highly motivated Software Engineers to join our team of developers pushing t...
Software Engineer Intern (Applied Machine Learning) - 2025 Summer/Fall (BS/MS)
The Applied Machine Learning Enterprise team combines system engineering and machine learning to develop and operate massively distributed machine learning training, inference systems and services to serve both the big model vendors and users around the world. In the engineering team, you'll have th...
Senior Machine Learning Engineer
Senior Machine Learning Engineer - Robotics. We’re building a world-class AI robotics team and are seeking a Machine Learning Engineer to join their ranks. In this role, you’ll focus on designing, developing, testing, and deploying machine learning algorithms that enable robotic manipulators to exec...
Machine Learning Engineer Graduate (Location Based Service) - 2025 Start (PhD)
Experienced in one or more of the following topics: Tensorflow, Caffe, MxNet, PyTorch or other machine learning frameworks-Have good team communication and collaboration skills. Responsibilities-Work on ML models and algorithms to drive the creation of cutting-edge technical solutions to bridge the ...
Machine Learning Ops Engineer, Applied Machine Learning
Join Apple's Applied Machine Learning Team, as a Senior Software Engineer, to enable GenAI across our Applications & Platforms. Apple’s Applied Machine Learning team has built systems for a number of large-scale data science applications. We do all this with an exceptional group of software engineer...
AIML - Senior Machine Learning Engineer, Computer Vision, Siri and Information Intelligence
We are seeking a highly skilled Senior Machine Learning Engineer specializing in Computer Vision to join our dynamic team. As a Machine Learning Engineer within the Siri team, you will define new approaches for evaluating innovative computer vision algorithms and models to enhance product capabiliti...