Robotics Machine Learning Research Scientist - Large Behavior Models

Toyota Research Institute • Los Altos, CA

18d • $176,000 - $264,000

About The Position

At Toyota Research Institute (TRI), we’re on a mission to improve the quality of human life. We’re developing new tools and capabilities to amplify the human experience. To lead this ground-breaking shift in mobility, we’ve built an extraordinary team in Automated Driving, Energy & Materials, Human-Centered AI, Human-Interactive Driving, and Robotics. The Mission We are working to create general-purpose robots capable of accomplishing a wide variety of dexterous tasks. To do this, we're building general-purpose machine learning foundation models for dexterous robot manipulation. These models, which we call Large Behavior Models (LBMs), use generative AI techniques to produce robot action from sensor data and human request. To accomplish this, we are creating a large curriculum of embodied robot demonstration data and combining that data with a rich corpus of internet-scale text, image, and video data. We are also utilizing high-quality simulation to augment real-world robot data with procedurally generated synthetic demonstrations. The Team The Robotics Machine Learning Team’s charter is to push the frontiers of research in robotics and machine learning to develop the future capabilities required for general-purpose robots able to operate in unstructured environments such as homes or factories. The Job We have several research thrusts under our broad mission, and we are looking for a research scientist in any of these areas: -Data-efficient and general algorithms for learning robust policies using multiple sensing modalities: proprioception, images, 3D representations, force, and dense tactile sensing. -Scaling learning approaches to large-scale models trained on diverse sources of data, including web-scale text, images, and video. -Leveraging test time computation for embodied applications. -Quick and efficient improvement of learned policies. -Continual Learning and Adaption -Multi-Modal Reasoning Models. -Structured hierarchical reasoning using learned models. -Reinforcement Learning with Language Action Models -Leveraging history and memory for learning policies for long context tasks. -Improving robustness and few-shot generalization by using sub-optimal and self-play data. -Interactive agents that can reduce the embodied and instructional ambiguity and can seek help and clarification. The researcher who joins will be encouraged to collaborate in our code infrastructure, work together with team members, run experiments with both simulated and real (physical) robots, and participate in publishing work to peer-reviewed venues and open-sourcing code. We’re looking for a research scientist who is comfortable working with both existing large static datasets as well as a growing dynamic corpus of robot data.

Requirements

• Hands-on experience with using machine learning for learned control, including RL, offline RL or behavior cloning, for manipulation.
• Or experience with machine learning and familiarity with large multi-modal datasets and models.
• Strong software development skills in Python.
• A “make it happen” demeanour and comfort with fast prototyping.
• A passion for robotics and doing research grounded in important fundamental problems.
• A track record of relevant publications in top international conferences (RSS, NeuRIPS, ICML, ICLR, CoRL, ICRA, IROS, …)

Nice To Haves

• Hardware experience.
• A track record of relevant open-source software contributions
• Experience working with large-scale datasets and multi-node training

Benefits

• TRI offers a generous benefits package including medical, dental, and vision insurance, 401(k) eligibility, paid time off benefits (including vacation, sick time, and parental leave), and an annual cash bonus structure.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed
• Number of Employees: 101-250 employees