Software Engineer: ML Optimization
Generalist3 months ago
Somerville, MA, USA or San Mateo, CA, USAMid Level
H1B Sponsor
Base Salary
$200k - $350k/yr
Responsibilities
- Own optimizations on the training and on-robot inference stacks.
- Implement ML, hardware, and software changes for performance gains.
- Optimize both inference and training stacks.
- Pursue ML optimizations across various system components.
Requirements
- Proficient in the latest ML techniques for training and inference optimizations.
- Experience with transformer and diffusion-based architectures.
- Ability to identify optimizations in CUDA kernels, ML architecture, and network bottlenecks.
- Familiarity with tools like torch, numpy, and Python.