Software Engineer: ML Optimization
Generalist6 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 is an AI robotics company building general intelligence for the physical world and making it useful to everyone. The founding team includes engineers from OpenAI, Google DeepMind, and Boston Dynamics. The company embraces both large-scale AI and robotics as core to its DNA.