Fully Sharded Data Parallel Jobs

1 job found

1mo ago

Senior Machine Learning Engineer

General Motors Sunnyvale, CA 🇺🇸 Remote

Sunnyvale, CA

1mo ago

$170k - $240k/yr Full time Senior AWS Azure

Apply Now

Job Description

General Motors invites you to join as a Senior Machine Learning Engineer. This role focuses on designing and building scalable ML training infrastructure that underpins AI research and model development, enabling intelligent driving technologies across GM vehicles. You will collaborate with ML engineers and researchers to deliver a scalable, reliable platform with strong performance, observability, and cross-functional collaboration. The team emphasizes rigorous engineering, a collaborative culture, and opportunities to impact the future of mobility. A comprehensive rewards package supports health, growth, and work-life balance.

Responsibilities

Design and develop a scalable, reliable, high-performance ML framework to support model training at scale.
Analyze and optimize model training performance to scale distributed workflows and maximize resource utilization across heterogeneous hardware, while controlling costs.
Raise the bar on system observability, debuggability, operational excellence, and user experience.
Collaborate with cross-functional teams to integrate new features and technologies into the platform.

Requirements

Bachelors degree or higher in Computer Science or an equivalent major, or relevant experience.
3+ years of professional software engineering experience.
2+ years of specialized experience in AI/ML infrastructure, enabling distributed training for scaling large ML models.
Strong Python programming skills, with proficiency in PyTorch (preferred), TensorFlow, or similar frameworks.
Experience with distributed computing, GPU computing, and cloud environments (AWS, GCP, Azure).
Willingness to travel to Sunnyvale, CA as needed.
Comfortable working in highly ambiguous and dynamic environments.

Technologies

Python
PyTorch
TensorFlow
AWS
GCP
Azure
PyTorch 2.x+
FSDP
Pipeline Parallelism

Compensation

Salary range: $170,000 to $240,000 per year. The actual base salary offered within this range will vary based on factors relevant to the position.
Bonus Potential: An incentive pay program pays out based on company performance, role level, and individual performance.

Location and Relocation

Sunnyvale, CA (remote)

This role may be eligible for relocation benefits.

Benefits

Medical
Dental
Vision
Health Savings Account
Flexible Spending Accounts
Retirement savings plan
Sickness and accident benefits
Life insurance
Paid vacation & holidays
Tuition assistance programs
Employee assistance program
GM vehicle discounts

View full job page →

Get Job Alerts

New jobs delivered to your inbox.