NVIDIA is looking to build general-purpose embodied agents that learn to explore and master complex skills across the virtual and physical world.
Requirements
- Pursuing a PhD degree in Computer Science/Engineering, Electrical Engineering, etc.
- Outstanding engineering skills in rapid prototyping and model training frameworks (PyTorch, Jax, Tensorflow, etc.)
- Excellent skills in working with large-scale machine learning/AI systems and computer infrastructure
- Experience with at least one of the following areas: multimodal foundation model or robotics
- Python is required; C++ and CUDA proficiencies are a plus
- Hands-on training experience and publications in at least one of the following models: LLMs, vision-language models, video generative models, diffusion models, and action-based transformers
Responsibilities
- Design and implement novel AI algorithms and models for general-purpose humanoid robots and embodied agents
- Develop large-scale AI training and inference methods for foundation models
- Optimize and deploy AI models in physical simulation and on robot hardware
- Collaborate with research and engineering teams across all of NVIDIA to transfer research to products and services
Other
- Pursuing a PhD degree in Computer Science/Engineering, Electrical Engineering, etc.