Sunnyvale, CA
Research Engineer, Reinforcement Learning
Target start date: Immediately. Relocation provided.
Since its founding in 2015, 1X has been at the forefront of developing advanced humanoid robots designed for household use. Our mission is to create an abundant supply of labor via safe, intelligent humanoids. At 1X, you’ll own critical projects, tackle unsolved research problems, deliver great products to customers, and be rewarded based on merit and achievement.
We are looking for a Research Engineer in Simulation and Reinforcement Learning (RL). In this role, you will design simulation environments for NEO and teach NEO to learn new capabilities via RL algorithms. This enables our robots to be safe and robust in a variety of conditions.
Responsibilities
- Full-stack engineering, from data engineering to model architecture design to shipping polished products
- Train NEO to do a diverse set of manipulation and locomotion tasks.
- Close the sim2real gap between policies trained in simulation and real.
- Work with controls, QA, and data collection teams to ship RL policies to the production fleet.
- Deploy skills trained with RL into home environments
REQUIREMENTS
-
Getting general-purpose robots to work in the home is just about the hardest problem one can work on. We are looking for people with the courage to tackle unsolved technical challenges with an intense work ethic.
- 4+ years of Python programming experience.
- Strong empirical research abilities and a keen eye for spotting performance bottlenecks in RL training.
- Experience with tuning reward functions, hyperparameters and exploration strategies to solve complex tasks with deep RL.
- Experience with authoring environments and benchmarks in simulators like Mujoco, Pybullet, or Isaac Sim.
Nice to have
- Advanced degree (MS or PhD) in Computer Science or related field
- Published RL research in top ML conferences (NeurIPS, CoRL, RSS, ICML, etc.)
- Have trained real-world quadruped or biped locomotion with RL
- Robotics and control theory knowledge
Sample Projects
We encourage you to apply even if you do not meet every single qualification. If you have direct experience in solving one of the “sample projects” listed below, please let us know in your cover letter.
- Fuse separate RL policies (walking, running, standing up) into a single hybrid policy that smoothly transitions between modes.
- Speed up the simulator to enable faster training and evaluation.
- Design and implement infra for re-training models deployed in the real world using offline RL methods.
- Reduce the amount of “reward engineering” needed to solve long-horizon tasks by formulating general objectives like energy minimization, self-play, and data-driven reward functions.
Compensation
At 1X your work and results will be rewarded with a total rewards package consisting of a base salary, stock options and benefits. Base salary range is $130,000 to $250,000. Your actual salary will be based on your knowledge, skills and experience.
Location
We believe the best work is done when collaborating and therefore require in-person presence in our office locations.