Senior / Staff Machine Learning Training Optimization Engineer

Job overview

The Senior / Staff Machine Learning Training Optimization Engineer role is hybrid in Pittsburgh, PA. Compensation is $141,000–$249,000 per year, with additional equity incentive awards and an annual performance bonus. The role supports Waabi’s work developing autonomous transportation technology. The engineer builds and improves distributed training frameworks, profiles model runtime and memory, evaluates technologies for training and inference, and develops tools and dashboards. The work helps research and machine learning teams improve training stability, efficiency, and resource use.

What you'll do

• Build standardized distributed training frameworks for research and production
• profile model runtime and memory to identify bottlenecks
• evaluate technologies for training and inference, including CUDA kernels, quantization-aware methods, and compilation and deployment techniques
• collaborate with researchers and machine learning engineers on resource-use practices
• create and improve tools and dashboards.

What we're looking for

Skills & competencies

machine learning engineer

training optimization

distributed training

model profiling

cuda

pytorch

jax

pittsburgh pa

waabi

hybrid

$141,000–$249,000/year

Work arrangement

• Weekend coverage required

Benefits & perks

• Equity awards, medical, dental and vision coverage, unlimited vacation, flexible hours, work-from-home support, office drinks, snacks and catered meals, and team activities.

Why this role

The posting lists equity awards, medical, dental and vision coverage, unlimited vacation, flexible hours, work-from-home support, and office meals and amenities.

About the employer

Waabi is hiring for this role. Industry: Custom Computer Programming Services. Sector: 54.