← All roles

Software Engineer: ML Optimization

Generalist AI· Technical Staff
Apply on company site
San Francisco Bay Area (San Mateo) or Boston (Somerville)$260k - $350kFull-timePosted 3mo ago

About the role

Software Engineer focused on ML optimization for training and on-robot inference stacks at Generalist, enabling step-function improvements in GPU performance and model efficiency.

Responsibilities

  • Making GPUs go brrrrr
  • Implementing ML, hardware, and software changes that lead to step-function gains
  • Optimizing both the inference and training stacks

Qualifications

  • Proficient and current with the latest ML techniques for training and inference optimizations in transformer and diffusion-based architectures
  • Ability to optimize across CUDA kernels, ML architecture, frontend/backend bottlenecks, CPU bottlenecks, NVLink and communications, and software libraries (torch, numpy, Python)

Skills

Required:ML optimization for training and inferenceML frameworks and libraries
Nice to have:CUDA kernels

Benefits

  • Offers Equity
Last updated Sep 1, 2026. Interested? Apply on the company site.
Generalist AI logo

About Generalist AI

Generalist is a frontier AI research and product driven company building general intelligence for the physical world.

Generalist AI: a frontier AI research and product driven company building general intelligence for the physical world.

AIGEN-1PRODUCTSVLAsembodied AIembodied intelligencegeneral-purposegeneral-purpose AI
Company profileWebsite

More open roles at Generalist AI