Machine Learning Engineer, Amazon General Intelligence (AGI) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Amazon · 9 hours ago

Machine Learning Engineer, Amazon General Intelligence (AGI)

positionSeattle, WA
timeFull-time
remoteOnsite
seniorityMid, Senior Level
money$129K/yr - $224K/yr
date3+ years exp

Amazon is seeking a passionate and talented Machine Learning Engineer to join their Artificial General Intelligence (AGI) team. The role involves developing training algorithms and modeling techniques for large language models, utilizing Amazon's vast data and computing resources to create impactful AI applications.

Artificial Intelligence (AI)DeliveryE-CommerceFoundational AIRetail
check
H1B Sponsor Likelynote

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

Qualification

Large Language ModelsGPU optimizationGenerative AISoftware developmentMachine learningAgile/ScrumCollaborationProblem-solving

Required

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

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

Benefits

Equity
Sign-on payments
Full range of medical, financial, and/or other benefits

Company

Amazon

twittertwittertwitter
Glassdoor3.7
company-logo
Amazon is a tech firm with a focus on e-commerce, cloud computing, digital streaming, and artificial intelligence.
dateFounded in 1994
location
Seattle, Washington, USA
people-size10001+ employees
web-link
https://amazon.com

H1B Sponsorship

Amazon has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22803)
2024 (21175)
2023 (19057)
2022 (24088)
2021 (12233)
2020 (14881)

Funding

Current Stage
Public Company
Total Funding
$8.11B
Key Investors
AmazonKleiner Perkins
2023-01-03Post Ipo Debt· $8B
2001-07-24Post Ipo Equity· $100M
1997-05-15IPO

Leadership Team

leader-logo
Douglas J. Herrington
CEO, Worldwide Amazon Stores
linkedin
leader-logo
Werner Vogels
VP & CTO
linkedin

Recent News

The Verge
Amazon has started automatically upgrading Prime members to Alexa Plus
2026-01-13
Investing.com
Italy antitrust cuts Amazon record fine to $878.2 million
2026-01-13
Business Standard India
Responsible growth: India must ensure data centres are sustainable
2026-01-13
Company data provided by crunchbase