Join NVIDIA: Software Engineering Intern for Generalist Embodied Agents Research

NVIDIA is on the lookout for an exceptional Software Engineering Intern to contribute to building humanoid robot foundation models and systems within our Generalist Embodied Agent Research (GEAR) group. Our mission is to create general-purpose embodied agents proficient in exploring and mastering complex skills in both virtual and physical worlds. At NVIDIA, we believe everything that moves will eventually be autonomous.

About the Role

As an intern, you will join an innovative and collaborative research team known for producing impactful works on multimodal foundation models, large-scale robot learning, game AI, and physical simulation. Some of our noteworthy projects include Eureka, VIMA, Voyager, MineDojo, MimicPlay, Prismer, and the recent Project GR00T, a foundation model for humanoid robots. Your contributions will be integral to our groundbreaking research projects and product development roadmaps.

Key Responsibilities

• Collaborate with world-class researchers at NVIDIA to develop robust AI solutions for general-purpose humanoid robots and embodied agents.
• Create large-scale AI training and inference infrastructure for foundation models.
• Develop robust and scalable data pipelines.
• Systematically evaluate and analyze AI models in physical simulations and on robot hardware.
• Collaborate across NVIDIA's research and engineering teams to ensure seamless software integration.

Requirements

We’re looking for candidates who are pursuing a Bachelor’s, Master’s, or PhD degree in Computer Science/Engineering, Electrical Engineering, or related fields. Ideal candidates will have:

• Outstanding engineering skills for rapid prototyping and product development, with proficiency in Python. C++ and CUDA skills are a plus.
• Knowledge in machine learning frameworks like PyTorch, Jax, TensorFlow, or robotics frameworks like ROS.
• Experience in one or more of the following areas:
  • Large-scale AI Systems:
    • Experience with large-scale machine learning/AI systems and compute infrastructure.
    • Hands-on training experience with LLM or vision-language models.
    • Expertise in building foundation model training data pipelines.
  • Practical Robotics Systems:
    • Understanding of robot kinematics, dynamics, and sensors.
    • Proficiency in safely operating robot hardware, lab equipment, and tools.
    • Experience with rapid prototyping technologies, such as CAD design and 3D fabrication.
    • Knowledge of control methods, such as PID, MPC, whole body control, etc.
    • Familiarity with physics simulation frameworks like Mujoco and Isaac suite.

Why Work at NVIDIA?

NVIDIA is widely regarded as one of the technology world’s most desirable employers. You will be joining some of the most forward-thinking and innovative minds globally, contributing to the future of general-purpose robots and embodied agents.

The hourly rate for our interns ranges from $16 to $88, determined based on the position, location, education, and experience. Interns are also eligible for NVIDIA benefits. We accept applications on an ongoing basis and encourage you to apply when you are ready.

NVIDIA is committed to fostering a diverse and inclusive workplace. We are proud to be an equal-opportunity employer, valuing diversity in all forms, and do not discriminate 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.

Company Name:

NVIDIA

Job Title:

Software Engineering Intern, Generalist Embodied Agents Research - Fall 2024