About this role
Join Generalist as a Software Engineer focused on ML Optimization to significantly enhance our embodied AI systems. You will drive step-function improvements in both training and on-robot inference stacks, pushing the boundaries of what robots can achieve. This role is for those who are passionate about optimizing every layer of the ML stack, from hardware to software.
What we are looking for 6
- Optimize ML training and inference stacks for embodied AI
- Implement hardware and software changes for significant performance gains
- Deep expertise in transformer and diffusion model optimizations
- Identify and resolve bottlenecks across CUDA, ML architecture, and network layers
- Proficiency in optimizing CPU, NVLink, and communication protocols
- Experience with PyTorch, NumPy, and Python performance tuning