Robotics AI Research Engineer (Equity)

Tau Robotics, Inc.; San Francisco (CA); On-site; USD 150,000 - 210,000; Full time

Job summary

Tau Robotics, Inc. is building a general-purpose AI for robots and is seeking a researcher to develop model-based reinforcement learning algorithms.

You will collaborate with researchers focused on imitation learning and world models, contributing to end-to-end robotic autonomy. Applicants should have a strong background in deep learning and RL/IL, proven debugging of RL methods, and proficiency in Python with JAX or PyTorch, plus a track record of publications or equivalent evidence in top

Qualifications

• Strong background in deep learning and reinforcement or imitation learning.
• Proven ability to implement and debug reinforcement learning algorithms.
• Publications at premier venues (NeurIPS, ICLR, ICML, CVPR, RSS, CoRL, ICRA), or equivalent public evidence.

Responsibilities

• Create and deploy model-based reinforcement learning algorithms.
• Partner with researchers specializing in imitation learning and world models.

Skills

Deep learning

Reinforcement learning

Imitation learning

Python

Tools

JAX

PyTorch

Job description

Tau Robotics, Inc. is building a general-purpose AI for robots and is seeking a researcher to develop model-based reinforcement learning algorithms.

You will collaborate with researchers focused on imitation learning and world models, contributing to end-to-end robotic autonomy. Applicants should have a strong background in deep learning and RL/IL, proven debugging of RL methods, and proficiency in Python with JAX or PyTorch, plus a track record of publications or equivalent evidence in top