Think Beyond The Label Jobs

Mobile Think Beyond The Label Logo

Job Information

Nvidia Research Engineer, Generalist Embodied Agent Research - New College Grad 2024 in Santa Clara, California

NVIDIA is searching for an outstanding engineer to build humanoid robot foundation models and systems in the Generalist Embodied Agent Research (GEAR) group. Everything that moves will eventually be autonomous. Our mission is to build general-purpose embodied agents that learn to explore and master complex skills across the virtual and the physical world.

You will work with an amazing and collaborative research team that consistently produces influential works on multimodal foundation models, large-scale robot learning, game AI, and physical simulation. Our past projects include Eureka (https://voyager.minedojo.org/) , VIMA (https://vimalabs.github.io/) , Voyager (https://mimic-play.github.io/) , MineDojo (http://minedojo.org/) , MimicPlay (https://eureka-research.github.io/) , Prismer (https://github.com/NVlabs/prismer) , and more. One of our team’s most recent milestones includes Project GR00T (https://developer.nvidia.com/project-gr00t) , a foundation model for humanoid robots. Your contributions will have a significant impact on our moonshot research projects and product roadmaps.

What you will be doing:

  • Work with world-class researchers at NVIDIA to develop robust AI solutions for general-purpose humanoid robots and embodied agents;

  • Develop large-scale AI training and inference infrastructure for foundation models;

  • Develop robust and scalable data pipelines;

  • Systematically evaluate and analyze AI models in physical simulation and on robot hardware;

  • Collaborate with research and engineering teams across all of NVIDIA to ensure your software integrates seamlessly up and down the stacks.

What we need to see

  • Bachelors or Masters in Computer Science/Engineering, Electrical Engineering, etc., or equivalent research experience

  • Significant work experience in one or both of the following domains:

Large-Scale AI Systems:

  • Outstanding engineering skills for rapid prototyping and product development. Python is required; C++ and CUDA proficiencies are a plus;

  • Excellent knowledge and hands-on experience in working with large-scale machine learning/AI systems and compute infrastructure;

  • Hands on training experience with LLMs or vision-language models;

  • Expertise in building LLM and multimodal data pipelines.

Practical Robotics Systems:

  • Deep understanding of robot kinematics, dynamics, and sensors;

  • Ability to safely operate robot hardware, lab equipment, and tools;

  • Experience with rapid prototyping technologies, such as CAD design and 3D fabrication;

  • Strong programming skills in Python, C++, ROS, and machine learning frameworks like PyTorch;

  • Knowledge of control methods, such as PID, MPC, and whole body control;

  • Familiarity with simulation frameworks such as MuJoCo and Isaac Sim.

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and productive people in the world. Please join us and be part of the forefront of developing general-purpose robots and embodied agents!

The base salary range is 104,000 USD - 189,750 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits (https://www.nvidia.com/en-us/benefits/) . NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

DirectEmployers