Search jobs > Cupertino, CA > Internship > Sr machine learning

Sr. Machine Learning Engineer, Annapurna ML

Amazon
Cupertino, CA, United States
$151.3K a year
Full-time

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

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

14 days ago
Related jobs
Promoted
VirtualVocations
Santa Clara, California

...

Promoted
Apple
Cupertino, California

This involves developing sophisticated machine learning models, using word embeddings and deep learning to understand the quality of matches, online learning to react quickly to change, natural language processing to understand queries, taking advantage of petabytes of data and signals from millions...

Promoted
OPPO US Research Center
Palo Alto, California

As a technology innovation leadership team in the fast-growing OPPO Research Center, We are looking for talented Machine Learning Engineers to join our Research and Development team. If you thrive on advancing machine learning technology, leveraging algorithms to address significant business challen...

Promoted
Apple
Sunnyvale, California

We're looking for an expert hard-working Senior Computer Vision/Machine Learning Software Engineer to join our team of applied researchers and software engineers who develop CV/ML technologies for new Apple products. We're looking for a Senior Computer Vision/Machine Learning Software Engineer to jo...

Promoted
Apple, Inc.
Cupertino, California

Practical experience and knowledge of generative AI and modern machine learning methods, particularly deep neural networks and reinforcement learning - Solid math background and understanding of algorithms and data structures - Experience with current deep learning frameworks, such as PyTorch, Tenso...

Promoted
HiTalent
Sunnyvale, California

Signal processing and/or machine learning. ...

Ford Motor Company
Palo Alto, California

Machine Learning Engineer role resides within the Ford’s Electric Vehicle organization. Data engineering, data product development and software product launches At least three of the following languages: Java, Python, Spark, Scala, SQL Developing Machine Learning algorithms working with Data Scienti...

Qualified Health
Palo Alto, California

We are seeking a highly skilled and experienced Senior Machine Learning Engineer to lead our technical team. Drive the adoption of best practices in machine learning operations (MLOps) to automate model training, deployment, and monitoring. Master's or PhD in Computer Science, Engineering, Machi...

Apple
Cupertino, California

As a Senior CV & ML Engineer, you will be working alongside our world-class creatives, designers, and engineers to help innovate in the creative space in ways that only Apple can. Extensive knowledge in theory and practice of computer vision, machine learning and deep learning techniques. Experience...

BHO Tech
Palo Alto, California

Senior Software Engineer - Machine Learning. We are seeking world-class machine learning experts to join our engineering team. You will be working with a team of talented engineers and researchers to develop core machine learning algorithms for improving the unique user experience available only wit...