Software Engineer: ML Optimization
Generalist24 hours ago
Base Salary
$260k - $350k/yr
Responsibilities
- Own optimizations across machine learning training and on-robot inference stacks.
- Implement ML, hardware, and software changes that deliver step-function performance gains.
- Optimize GPU execution, CUDA kernels, ML architectures, network bottlenecks, CPU performance, NVLink communications, and framework-level inefficiencies.
- Investigate and improve performance across PyTorch, NumPy, and Python.
Requirements
- Proficiency with current machine learning techniques for training and inference optimization in transformer- and diffusion-based architectures.
- Willingness to pursue optimization opportunities across kernels, models, networks, CPUs, communications, and software frameworks.
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.