ML Infra SDE II: Optimize ML Accelerator Fleet

Amazon

Seattle (WA)

On-site

USD 144,000 - 194,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Health insurance
401(k) matching
Paid time off
Parental leave

Job summary

Amazon is hiring a Software Engineer to lead the development of ML infrastructure tools for running, optimizing, and analyzing ML workloads. You will guide ML tool projects from architecture through delivery, with deep profiling, resource management, scheduling, and code generation.

The role involves building scalable infra for ML accelerators and coordinating with ML scientists, engineers, and hardware teams. Hybrid onsite with remote options when needed.

Job description

Amazon is hiring a Software Engineer to lead the development of ML infrastructure tools for running, optimizing, and analyzing ML workloads. You will guide ML tool projects from architecture through delivery, with deep profiling, resource management, scheduling, and code generation.

The role involves building scalable infra for ML accelerators and coordinating with ML scientists, engineers, and hardware teams. Hybrid onsite with remote options when needed.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

SDE II: Infra for ML Workloads & Cloud Services
SDE II: Infra for ML Workloads & Cloud Services

Amazon • Cupertino (CA)

On-site
USD 165,000 - 224,000
Health insurance
401(k) matching
Paid time off
+1
SDE II: ML Infra – Capacity, Scheduling & Fleet Orchestration
SDE II: ML Infra – Capacity, Scheduling & Fleet Orchestration

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 150,000 - 210,000
Sign-on payments
Restricted stock units (RSUs)
Health insurance
+14
ML Engineer, Infrastructure at Scale
ML Engineer, Infrastructure at Scale

Socket.dev • Austin (TX)

On-site
USD 143,000 - 195,000
ML Infra Engineer: Build Scalable ML Services & MLOps
ML Infra Engineer: Build Scalable ML Services & MLOps

Stripe • United States

Hybrid
CAD 172,000 - 258,000
Equity
Retirement plans
Health benefits
+1
SDE: EC2 ML Provisioning & Scalable Infra
SDE: EC2 ML Provisioning & Scalable Infra

Amazon • Seattle (WA)

On-site
USD 144,000 - 194,000
RSU eligibility
Health insurance
401(k) match
+1
SDE II, ML Infra Services, Annapurna Labs
SDE II, ML Infra Services, Annapurna Labs

Amazon Web Services (AWS) • Seattle (WA)

On-site
USD 150,000 - 210,000
Sign-on payments
Restricted stock units (RSUs)
Health insurance
+14
SDE II, ML Infra Services, Annapurna Labs (AWS)
SDE II, ML Infra Services, Annapurna Labs (AWS)

Amazon • Seattle (WA)

On-site
USD 144,000 - 194,000
Health insurance
401(k) matching
Paid time off
+1
Software Engineer II — ML Data Infra for Search
Software Engineer II — ML Data Infra for Search

Amazon • Palo Alto (CA)

On-site
USD 165,200 - 223,600
Sign-on bonuses
Restricted stock units (RSUs)
Health benefits
Machine Learning Engineer, Infra AI Solutions
Machine Learning Engineer, Infra AI Solutions

Amazon • Austin (TX)

On-site
USD 144,000 - 194,000
Medical, Dental, and Vision Coverage
Paid Time Off (PTO)
401(k) Plan
Senior ML Infrastructure Engineer | Remote Options
Senior ML Infrastructure Engineer | Remote Options

Matchgroup • West Hollywood (CA)

On-site
USD 190,000 - 246,000