Machine Learning Engineer, Amazon General Intelligence (AGI)

Amazon; Seattle (WA); On-site; USD 129,000 - 224,000; Full time

Job summary

Amazon is seeking a Machine Learning Engineer to join its Artificial General Intelligence (AGI) team. This role focuses on developing innovative large language models, leveraging Amazon's extensive data and computing resources to create cutting-edge AI solutions. Candidates should possess strong software development experience and a passion for advancing Generative AI technologies.

Benefits

Comprehensive health benefits

Equity and sign-on bonuses

Career development programs

Qualifications

• 3+ years of professional software development experience.
• Experience optimizing GPU workloads.
• Proficient in at least one programming language.

Responsibilities

• Develop and maintain platforms for LLM applications.
• Optimize model training and performance using GPUs.
• Collaborate with applied scientists on data processing and scaling.

Skills

Generative AI

Machine Learning

GPU Optimization

Data Strategy

Education

Master's degree in Computer Science or equivalent

Job description

Machine Learning Engineer, Amazon General Intelligence (AGI)

Join to apply for the Machine Learning Engineer, Amazon General Intelligence (AGI) role at Amazon

Machine Learning Engineer, Amazon General Intelligence (AGI)

5 days ago Be among the first 25 applicants

Join to apply for the Machine Learning Engineer, Amazon General Intelligence (AGI) role at Amazon

Description

The Artificial General Intelligence (AGI) team is looking for a passionate, talented, and inventive Machine Learning Engineer(MLE) to play pivotal role in the development of industry-leading multi-modal and multi-lingual Large Language Models (LLM). As our SDE/MLE superstar, you'll lead the charge in developing training algorithms and modeling techniques that will push the boundaries of large model training using GPUs and AWS Trainium. Your work will directly impact our customers' lives through game-changing products and services powered by your Generative AI breakthroughs!

Description

The Artificial General Intelligence (AGI) team is looking for a passionate, talented, and inventive Machine Learning Engineer(MLE) to play pivotal role in the development of industry-leading multi-modal and multi-lingual Large Language Models (LLM). As our SDE/MLE superstar, you'll lead the charge in developing training algorithms and modeling techniques that will push the boundaries of large model training using GPUs and AWS Trainium. Your work will directly impact our customers' lives through game-changing products and services powered by your Generative AI breakthroughs!

Get ready to dive into Amazon's vast and diverse data sources and harness the immense power of our large-scale computing resources to turbocharge the development of multi-modal Large Language Models (LLMs) and other awe-inspiring Generative Artificial Intelligence (Gen AI) applications. Your expertise and insights will be invaluable in defining data strategies, model optimizations, and evaluation methods that will set new standards in the industry!

So, if you're passionate about pushing the limits of AI, thrive in a fast-paced and innovative environment, and are ready to make a lasting impact on the world, this is your chance! Join us on this exhilarating adventure and let's revolutionize AGI together! #MLE

Key job responsibilities

• Ability to quickly learn new technologies and algorithms in the field of Generative AI to participate in our journey to build the best LLMs.
• Responsible for the development and maintenance of key platforms needed for developing, evaluating and deploying LLM for real-world applications.
• Work with other team members to investigate design approaches, prototype new technology and evaluate technical feasibility.
• Work closely with Applied scientists to process massive data, scale machine learning models while optimizing.
• Proven track record of optimizing GPU workloads at the kernel level.
  
  

A day in the life

As a SDE/MLE with the AGI team, you will be responsible for leading the development of modeling techniques and optimizing performance of the state of the art of large model training using hardware like NVDIA GPUs. You will leverage Amazon’s heterogeneous data sources and large-scale computing resources to accelerate development with multi-modal Large Language Models (LLMs) and other Generative Artificial Intelligence (Gen AI) applications. As a key player in our team, you'll have a significant influence on our overall strategy, shaping the future direction of AGI at Amazon. You'll be driving system architecture and champion best practices that will ensure an unparalleled infrastructure of the highest quality. Work in an Agile/Scrum environment to move fast and deliver high quality software.

About The Team

Join our AGI team and work at the forefront of AI. Collaborate with top minds pushing boundaries in deep learning, reinforcement learning, and more. Gain valuable experience and accelerate your career growth. This is a unique opportunity to create history and shape the future of artificial intelligence.

Mission of the team: We leverage our hyper-scalable, general-purpose large model training and inference systems to develop and deploy cutting-edge sensory AI foundational models that revolutionize machine perception, interpretation and interaction, with humans and with the physical world.

Basic Qualifications

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
• Experience programming with at least one software programming language
  
  

Preferred Qualifications

• 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
• Master's degree in computer science or equivalent
• Experience in techniques like kernel fusion and custom kernels to improve GPU utilization, mixed precision training using lower precision and dynamic loss scaling while leveraging hardware specific mixed precision capabilities and/or demonstrated ability to implement efficient memory management like gradient (activation) checkpointing, gradient accumulation, offloading optimizer states, and smart prefetching.
  
  

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

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.


Company - Amazon.com Services LLC

Job ID: A3003288

Seniority level

• Seniority level

  Not Applicable

Employment type

• Employment type

  Full-time

Job function

• Job function

  Information Technology, Consulting, and Engineering

• Industries

  Software Development

Referrals increase your chances of interviewing at Amazon by 2x

Sign in to set job alerts for “Machine Learning Engineer” roles.

Seattle, WA $99,500.00-$187,000.00 5 days ago

Software Engineer - AI/ML, Multiple Locations

Redmond, WA $81,900.00-$174,600.00 1 week ago

Seattle, WA $100,000.00-$220,000.00 3 months ago

Seattle, WA $127,000.00-$240,000.00 2 weeks ago

Seattle, WA $118,657.00-$168,150.00 1 week ago

Machine Learning Scientist I - Experienced Graduates

Senior ML Engineer, Applied Machine Learning

Seattle, WA $56.25-$173,000.00 2 weeks ago

Machine Learning Engineer, Monetization Engineering

Machine Learning Engineer, Identity Product

Seattle, WA $212,000.00-$318,000.00 1 week ago

Machine Learning Engineer, Stripe Capital

Bellevue, WA $56.25-$173,000.00 2 weeks ago

Redmond, WA $56.25-$173,000.00 2 weeks ago

Machine Learning Engineer Graduate (Search E-Commerce - Seattle) - 2025 Start (BS/MS)

Seattle, WA $118,657.00-$168,150.00 6 days ago

Machine Learning Engineer, Computer Vision

Machine Learning Engineer, Core Engineering

Machine Learning Engineer, Forecast Platform

Bellevue, WA $145,000.00-$204,000.00 2 weeks ago

Machine Learning Engineer Graduate (TikTok E-Commerce - Conversational AI) -2025 Start (BS/MS)

Seattle, WA $104,500.00-$139,650.00 1 week ago

Machine Learning Engineer Graduate Data-Search - TikTok.US - San Jose) - 2025 Start (BS/MS)

Seattle, WA $106,400.00-$168,150.00 1 week ago

Seattle, WA $100,000.00-$720,000.00 2 weeks ago

Bellevue, WA $125,000.00-$168,000.00 1 hour ago

Greater Seattle Area $155,000.00-$175,000.00 3 days ago

Machine Learning Engineer Intern (E-commerce-Recommendation) - 2025 Summer/Fall (BS/MS)

Kirkland, WA $141,000.00-$202,000.00 1 day ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.