- Robot type
- Robot AI and Software
- Location
- USA
- Job type
- Artificial Intelligence
- Posted
- Feb 12, 2026
- Salary
- $260,000–$350,000 a year
Software Engineer: ML Optimization
Job description
We build embodied foundation models, starting with a focus on dexterity. This requires advancing the frontiers of data, models, and hardware, to enable robots to intelligently interact with the physical world.
The company embraces both large-scale AI and robotics as core to its DNA. Our team of researchers, roboticists, and company builders come from OpenAI, Boston Dynamics, Google DeepMind, and other frontier labs—with a track record of shipping AI breakthroughs.
At Generalist, we are on a mission to build general intelligence for the physical world and make it useful to everyone.
Job responsibilities
- Making GPUs go brrrrr
- Implementing ML, hardware, and software changes that lead to step-function gains
- Optimizing both the inference and training stacks
- Are proficient and stay current with the latest ML techniques for training and inference optimizations in transformer and diffusion based architectures
- Will chase ML optimizations anywhere: From the CUDA kernels, to ML architecture, to frontend or backend network bottlenecks, CPU bottlenecks, NVLink and comms, to torch, numpy, and Python inefficiencies.
Similar jobs
Generalist · Artificial Intelligence
Research Scientist: Post-Training
Remote, USA
Generalist · Artificial Intelligence
Software Engineer: ML Robotics Systems
Remote, USA
Generalist · Artificial Intelligence
Software Engineer: ML Infra
Remote, USA
Generalist · Artificial Intelligence
Research Scientist: Pretraining
Remote, USA