Research Scientist: Pretraining

Generalist AI; Boston (MA); Sur place; USD 180 000 - 280 000; Plein temps

Résumé du poste

Generalist is building the base intelligence layer for robotics by pretraining large-scale robot foundation models from multimodal data. You will design and run core training efforts to enable new general capabilities across embodiments and environments.

If you have worked on pretraining large-scale ML systems or generative foundation models (multimodal, language, audio), this role may be a fit. You’ll collaborate with ML Infra and Systems to push performance.

Qualifications

• Experience with pretraining large-scale ML systems or generative foundation models (multimodal, language, audio) is a plus.
• Strong background in transformer or diffusion models used at scale for generative tasks.
• Experience leading multi-node, multi-GPU distributed training efforts.
• Knowledge of scaling laws and data quality effects on model performance.
• Solid PyTorch fundamentals and debugging across stack layers.

Responsabilités

• Designing and executing large-scale pretraining runs for robot foundation models (transformer- and diffusion-based).
• Defining model architectures, objectives, and training curricula across multimodal robotic data.
• Developing scalable data mixtures and sampling strategies over petabyte-scale datasets.
• Guiding data collection directions and sourcing new datasets for training.
• Running ablations to understand scaling laws and architecture tradeoffs.
• Collaborating with ML Infra and Systems to improve cluster utilization and reliability.
• Turning raw robotic interaction data into generalizable model capabilities.

Connaissances

Large-scale ML systems

Distributed training

PyTorch

Multimodal models

Description du poste

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

You will build the base intelligence layer for robotics. We train large-scale robot foundation models from massive multimodal datasets spanning video, proprioception, action traces, language, and more. You will design and run the core large-scale training efforts that give our models fundamentally new general capabilities across embodiments, tasks, and environments. You will “live and breathe” all forms of robot data.

If you have worked on pretraining large-scale ML systems or generative foundation models e.g. multimodal, language, audio, etc. (not only robotics), then this may be the role for you.

You’ll be responsible for:

• Designing and executing large-scale pretraining runs for robot foundation models (transformer- and diffusion-based architectures)
• Defining model architectures, objectives, and training curricula across multimodal robotic data (vision, action, state, language)
• Developing scalable data mixtures and sampling strategies across petabyte-scale datasets
• Guiding data collection operations towards new directions, as well as sourcing new datasets
• Running ablations to understand scaling laws, data quality effects, and architecture tradeoffs
• Collaborating closely with ML Infra and Systems to push cluster utilization, throughput, and reliability
• Turning raw robotic interaction data into generalizable model capabilities

You might thrive in this role if you:

• Have worked on large-scale ML systems with senior/staff (L6+) experience.
• Have deep experience training large transformer or diffusion models at scale (for generative models e.g. including language models, audio models, or video models)
• Have led or significantly contributed to multi-node, multi-GPU distributed training efforts
• Have worked on scaling laws, optimization dynamics, and large-model failure modes
• Have strong PyTorch fundamentals and comfort debugging at every layer of the stack
• Care about both empirical rigor and raw iteration speed
• Are excited about building general-purpose robot intelligence from first principles