ML Optimization Engineer: Robotics Training & Inference

Generalist AI, Inc.; San Francisco (CA); In loco; USD 180.000 - 240.000; Tempo pieno

Descrizione del lavoro

Generalist AI, Inc. in San Francisco is seeking an ML optimization engineer to own optimizations on training and on-robot inference stacks. The role requires pushing step-function gains across hardware, software, and ML components.

You will explore CUDA kernels, model architectures, and bottlenecks from CPU to NVLink, delivering practical improvements in transformer and diffusion-based systems. Join a team with deep roots in AI and robotics.

Competenze

• Own optimizations on both training and on-robot inference stacks.
• Implement ML, hardware, and software changes that yield step-function gains.
• Proficient with latest ML techniques for training and inference in transformer and diffusion architectures.

Mansioni

• Implementing ML, hardware, and software changes that lead to step-function gains.
• Optimizing both the inference and training stacks.
• Investigating CUDA kernels, architectures, and bottlenecks from CPU to interconnects.

Descrizione del lavoro

Generalist AI, Inc. in San Francisco is seeking an ML optimization engineer to own optimizations on training and on-robot inference stacks. The role requires pushing step-function gains across hardware, software, and ML components.

You will explore CUDA kernels, model architectures, and bottlenecks from CPU to NVLink, delivering practical improvements in transformer and diffusion-based systems. Join a team with deep roots in AI and robotics.