ML Optimization Engineer: Training & Inference

Generalist; Somerville, San Mateo (MA, CA); On-site; USD 100,000 - 150,000; Full time

Job summary

Generalist in Somerville, Massachusetts is seeking a motivated individual dedicated to making general-purpose robots a reality. In this role, you will take charge of optimizations for both training and on-robot inference stacks, leading significant advancements in AI techniques.

The ideal candidate thrives in a dynamic, innovative environment and is proficient in the latest machine learning methodologies. We aim for groundbreaking changes that transform how robots can intelligently interact with the physical world.

Qualifications

• Proficient in the latest ML techniques for optimizing robots.
• Chase optimizations across CUDA kernels and hardware bottlenecks.
• Stay current with advancements in AI implementations.

Responsibilities

• Own optimizations on both training and on-robot inference stacks.
• Implement ML and software changes for significant improvements.
• Optimize both the inference and training stacks.

Skills

Proficiency in ML techniques for training and inference optimizations

Experience with CUDA, torch, numpy, and Python

Understanding of transformer and diffusion-based architectures

Job description

Generalist in Somerville, Massachusetts is seeking a motivated individual dedicated to making general-purpose robots a reality. In this role, you will take charge of optimizations for both training and on-robot inference stacks, leading significant advancements in AI techniques.

The ideal candidate thrives in a dynamic, innovative environment and is proficient in the latest machine learning methodologies. We aim for groundbreaking changes that transform how robots can intelligently interact with the physical world.