ML Optimization Engineer - Training & On-Robot Inference

Generalist AI, Inc.; Boston (MA); In loco; USD 140.000 - 210.000; Tempo pieno

Descrizione del lavoro

Generalist AI, Inc. in Boston is seeking an ML optimization engineer to own improvements across training and on-robot inference systems. This role focuses on advancing transformer and diffusion-based architectures, CUDA kernels, and Python performance.

You will contribute to scalable embodied foundation models and work within a leading robotics and AI team. Join a company with a track record of shipping AI breakthroughs, collaborating with researchers and engineers to push the limits of what

Mansioni

• You will own optimizations on both the training and on-robot inference stacks.
• Implementing ML, hardware, and software changes that lead to step-function gains
• Optimizing both the inference and training stacks

Conoscenze

ML optimization

Transformer models

Diffusion models

CUDA kernels

Python performance

PyTorch

NumPy

Strumenti

CUDA

PyTorch

NumPy

Descrizione del lavoro

Generalist AI, Inc. in Boston is seeking an ML optimization engineer to own improvements across training and on-robot inference systems. This role focuses on advancing transformer and diffusion-based architectures, CUDA kernels, and Python performance.

You will contribute to scalable embodied foundation models and work within a leading robotics and AI team. Join a company with a track record of shipping AI breakthroughs, collaborating with researchers and engineers to push the limits of what