Amazon · 11 hours ago
Sr. Machine Learning Engineer, Amazon General Intelligence (AGI)
Seattle, WA
Full-time
Onsite
Senior Level, Lead/Staff
$151K/yr - $262K/yr
8+ years expAmazon is a leading company in the AI and machine learning sector, seeking a Senior Machine Learning Engineer to join their Machine Learning training infrastructure team. The role involves leading the development of next-generation ML training infrastructure and shaping the company's machine learning capabilities while mentoring other engineers and driving innovation in AI technology.
Artificial Intelligence (AI)DeliveryE-CommerceFoundational AIRetail
Responsibilities
Lead the definition, design, architecture quality, implementation, and delivery of the most advanced, most difficult, most cross-cutting, and/or most ambiguous challenges spanning across our ML infrastructure
Align the teams in ML Infrastructure and related organizations to a coherent technical vision and deliver systems that fit well together
Exert influence over multiple teams, increasing their productivity and effectiveness. You hold peers and teams to a high bar for performance and efficiency, and aid teams through your expert guidance and example
Considered to be an authority on technical issues by both the technical and research community, you are responsible for guiding difficult trade-off decisions and drive awareness about the impact and consequences of technical decisions on AI research and product development
Demonstrate significant innovation, creativity, and judgement when solving challenging AI/ML infrastructure problems. Identify future skills needed across your organization and advocate for the development and/or acquisition of those skills to senior leaders. You scout top talent and recruit them to the company
Actively mentor senior and Principal engineers, scale yourself by developing and institutionalizing best practices in AI/ML infrastructure and distributed computing across the organization
Qualification
Required
8+ years of professional software development experience in distributed systems with emphasis on ML infrastructure
8+ years of current programming experience building ML infrastructure using languages such as Python, C++ or Rust
Hands-on experience with parallel computing platforms such as CUDA, OpenMP, etc
Deep understanding of AI frameworks such as PyTorch, TensorFlow, and JAX, and their demands on underlying compute infrastructure, memory bandwidth, network interconnect, and storage as scale goes up
Knowledge of emerging AI hardware accelerators and architectures
Experience with containerization and orchestration technologies (Docker, Kubernetes)
Experience with cloud computing platforms (AWS, Azure, GCP) and their offerings
5+ years of non-internship professional software development experience
5+ years of programming with at least one software programming language experience
5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
Experience as a mentor, tech lead or leading an engineering team
Preferred
5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
Bachelor's degree in computer science or equivalent
Benefits
Equity
Sign-on payments
Full range of medical, financial, and/or other benefits
Company
Amazon
Amazon is a tech firm with a focus on e-commerce, cloud computing, digital streaming, and artificial intelligence.
10001+ employees
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 CompanyTotal Funding
$8.11BKey Investors
AmazonKleiner Perkins
2023-01-03Post Ipo Debt· $8B
2001-07-24Post Ipo Equity· $100M
1997-05-15IPO
Leadership Team
Douglas J. Herrington
CEO, Worldwide Amazon Stores
Werner Vogels
VP & CTO
Recent News
2026-01-13
2026-01-13
Business Standard India
2026-01-13
Company data provided by crunchbase