Senior ML Training Architect — Distributed, Flexible & Remote

Waabi; United States; Hybrid; USD 141,000 - 249,000; Full time

Benefits offered by this job

Competitive compensation and equity awards

Health and Wellness benefits (Medical, Dental, Vision)

Unlimited Vacation

Flexible hours and Work from Home support

Daily drinks, snacks and catered meals

Team building activities and social events

World-class facility with gym and collaborative spaces

Job summary

Waabi is looking for a skilled professional to develop distributed training frameworks that enhance the stability and efficiency of our systems. This role requires at least 4 years of relevant experience and proficiency in Python, C++, or Rust. You will work closely with researchers to advance our self-driving technologies and create innovative solutions.

We offer a competitive salary range of $141,000 to $249,000, health benefits, unlimited vacation, and flexible work hours. Join us and be part of a diverse and inclusive team that is redefining the future!

Qualifications

• 4+ years of industry experience in relevant fields.
• Solid coding proficiency in Python, C++, or Rust.
• Experience in deep learning frameworks such as PyTorch or Jax.
• Skilled in profiling CPU and GPU code using tools like PyTorch Profiler.

Responsibilities

• Build standardized distributed training frameworks.
• Profile model runtime and memory to find performance bottlenecks.
• Work with researchers and ML engineers on best practices.

Skills

Python

C++

Rust

Deep learning frameworks

Profiling tools

Education

MS/PhD or Bachelor’s in Computer Science or related

Tools

PyTorch

Jax

NVIDIA Nsight

Job description

Waabi is looking for a skilled professional to develop distributed training frameworks that enhance the stability and efficiency of our systems. This role requires at least 4 years of relevant experience and proficiency in Python, C++, or Rust. You will work closely with researchers to advance our self-driving technologies and create innovative solutions.

We offer a competitive salary range of $141,000 to $249,000, health benefits, unlimited vacation, and flexible work hours. Join us and be part of a diverse and inclusive team that is redefining the future!