← Back to Jobs
Generalist logo
Todayashby

Software Engineer: ML Infra

Generalist / Robotics Engineering
Somerville, Massachusetts, United StatesUSD260,000 - USD350,000 / YEAROn-site
Apply on company site
AI Summary

You will be responsible for managing and optimizing large-scale GPU compute fleets and data loading infrastructure for robot foundation models. Additionally, you will oversee the orchestration of robot inference fleets to ensure high performance in compute-constrained environments.

The ideal candidate has deep experience in managing large-scale distributed training runs and ML workload orchestration using Slurm or Kubernetes. You must possess a comprehensive understanding of ML hardware, storage, networking stacks, and the Nvidia GPU ecosystem.

Job Details
Job Type: FULL TIME
Visa Sponsorship: No
Salary: USD260,000 - USD350,000 / YEAR
Work Arrangement: On-site
Experience Level: 5 - 10 years
Hours: 40 hrs/week
Language: English
Benefits
Equity
Key Skills
GPU compute managementSlurmKubernetesML workload orchestrationData loading systemsDistributed systemsInference infrastructureNvidia GPU ecosystemStorage infrastructureNetworking stacksHardware optimizationRobot foundation models
Insider connections @Generalist
Members only

Find people at Generalist who may share hiring insights or referrals for this role.

Meet the people behind the opportunity

Create a free account to search for hiring managers and potential referral contacts.

No contact data is shown on this public page.


About Generalist

At Generalist, we are on a mission to build general intelligence for the physical world and make it useful to everyone. We believe the industries and homes of the future will depend on humans and machines working together in new ways. Robots can help us build more and get more done.

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. Before Generalist, we pioneered large embodied multimodal models and vision-language-action models (PaLM-E, RT-2, Gemini Robotics), launched and scaled ChatGPT and GPT-4 to hundreds of millions of users, engineered the foundations of autonomous driving, built next-generation robots (Atlas, Spot, Stretch) and pushed the limits of what they can do (from parkour to manipulation, and testing robustness).

We are an equal opportunity employer, and we do not discriminate on the basis of race, religion, color, national origin, sex, sexual orientation, age, veteran status, disability, genetic information, or other applicable legally protected characteristic.

About the Role

Generalist trains very large robot foundation models. This requires utilizing very large numbers of the latest generation GPU hardware and infrastructure (currently Nvidia) to run distributed training jobs and researcher experiments. We have extreme requirements on storage and data loading infrastructure that requires maximizing cloud infrastructure and custom solutions.

You will also own inference infrastructure. For our robots this is a fleet of on-prem GPUs attached to robots that have extreme real-time and latency budgets in compute constrained environments.

You’ll be responsible for:

  • Owning our GPU compute fleets

  • Ensure our GPUs are easy for researchers to use and maximally utilized

  • Optimizing and improving ML data loading transport and storage in highly distributed fully utilized environments.

  • Orchestration of robot inference fleets

You might thrive in this role if you:

  • Have managed large fleets of GPUs doing large-scale, long-term, highly distributed training runs or inference

  • Deep experience in Slurm or Kubernetes for ML workload orchestration

  • Have build high-scale ML data loaders and preparation systems

  • Deeply understand every layer of the ML hardware, storage, and networking stacks

  • Have experience in the NVidia GPU ecosystem

About Generalist

Generalist is an AI robotics company on a mission to make general-purpose robots a reality. 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.

Robotics Engineering
106 employees
San Mateo, California
$753M raised
Categories
Artificial Intelligence (AI)Generative AIRobotics
Specialties