Sr. Machine Learning Engineer, Amazon General Intelligence (AGI)

Amazon; Seattle (WA); On-site; USD 151,000 - 262,000; Full time

Job summary

A leading tech company is seeking a Senior Machine Learning Engineer for the development of next-generation ML training infrastructure in Seattle. This high-impact role involves leading teams and driving innovation in AI solutions, necessitating extensive experience in software development, distributed systems, and machine learning technologies.

Qualifications

• 8+ years of professional software development experience.
• Experience with AI frameworks like PyTorch and TensorFlow.
• 5+ years of leading design or architecture of new systems.

Responsibilities

• Lead the definition and implementation of advanced ML infrastructure.
• Mentor and guide engineering teams in best practices.
• Influence technical decisions and align teams to a coherent vision.

Skills

Distributed systems

Machine Learning infrastructure

Mentorship

Programming (Python, C++, Rust)

Parallel computing

Education

Bachelor's degree in computer science or equivalent

Tools

CUDA

OpenMP

Docker

Kubernetes

AWS

Job description

Sr. Machine Learning Engineer, Amazon General Intelligence (AGI)

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

Sr. Machine Learning Engineer, Amazon General Intelligence (AGI)

5 days ago Be among the first 25 applicants

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

Get AI-powered advice on this job and more exclusive features.

Description

Our Machine Learning training infrastructure (ML Infra) team is responsible for designing, implementing, and optimizing large-scale computing infrastructure that powers our cutting-edge AI and machine learning initiatives. We leverage advanced hardware, innovative software architectures, and distributed computing techniques to enable breakthrough research and product development across the company.

Description

Our Machine Learning training infrastructure (ML Infra) team is responsible for designing, implementing, and optimizing large-scale computing infrastructure that powers our cutting-edge AI and machine learning initiatives. We leverage advanced hardware, innovative software architectures, and distributed computing techniques to enable breakthrough research and product development across the company.

We are seeking a Senior Machine Learning Engineer to join our team and lead the development of our next-generation ML training infrastructure. This is a high impact, high visibility role that will shape the future of our machine learning capabilities and contribute to the advancement of AI technology across the industry.

Key job 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.
  
  

A day in the life

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
  
  

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

• 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 Qualifications

• 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
  
  

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 $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.


Company - Amazon.com Services LLC - A57

Job ID: A3003295

Seniority level

• Seniority level

  Mid-Senior level

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

Get notified about new Machine Learning Engineer jobs in Seattle, WA.

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

Software Engineer, Open Source, AI/ML, Developer Infrastructure

Software Engineer - AI/ML, Multiple Locations

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

Bellevue, WA $125,000.00-$168,000.00 14 hours ago

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

Machine Learning Engineer, Identity Product

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

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

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

Redmond, WA $100,600.00-$215,400.00 1 hour ago

Seattle, WA $7,302.00-$11,866.00 6 days ago

Senior ML Engineer, Applied Machine Learning

Seattle, WA $120,000.00-$150,000.00 1 week ago

Software Engineer III, Google Cloud Business Platforms

Seattle, WA $141,000.00-$202,000.00 4 days ago

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

Greater Seattle Area $94,000.00-$120,000.00 1 day ago

Seattle, WA $167,000.00-$185,500.00 6 hours ago

Machine Learning Engineer, Monetization Engineering

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

Machine Learning Engineer, Stripe Capital

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