Software Engineer: ML Optimization
Generalist8 months ago
Base Salary
$200k - $350k/yr
Responsibilities
- Own optimizations across both the training and on-robot inference stacks.
- Implement ML, hardware, and software changes that produce substantial performance gains.
- Optimize GPUs, inference systems, and training systems.
Requirements
- Proficiency with current machine-learning techniques for training and inference optimization in transformer and diffusion-based architectures.
- Ability to pursue optimization opportunities across CUDA kernels, ML architecture, frontend and backend network bottlenecks, CPU bottlenecks, NVLink communications, Torch, NumPy, and Python.
Tech Stack
Categories
About Generalist
Generalist builds general‑purpose robots powered by large‑scale AI, focusing on vision‑language‑action and embodied multimodal models for real‑world tasks in industry and homes. The company develops robotics platforms and software and is privately held. It was founded in 2024, is headquartered in San Mateo, California, and its founding team includes alumni of OpenAI, Google DeepMind, and Boston Dynamics.