Senior Software Engineer, Robotics and ML



Software Engineering, Data Science
Santa Clara, CA, USA
Posted on Thursday, December 21, 2023

NVIDIA is seeking a Senior Software Engineer - Robotics & ML to integrate, evaluate and productize robotics technologies. For two decades, NVIDIA has pioneered visual computing, the art and science of computer graphics. With our invention of the GPU we also set the stage for advances in the field of medical research, computer vision and artificial intelligence. Today NVIDIA is at the groundbreaking level to innovate in robotics and intelligent autonomous machines. There has never been a more exciting time to join our company and team.

The team around Isaac AMR is building a robotics platform for developing the next generation of intelligent and autonomous robots. Isaac is binding together high-fidelity visual and physical simulation, a high-quality developing platform, hundreds of optimized algorithms to tackle hard problems in computer vision and artificial intelligence, and a small and powerful computational platform to form the brain of intelligent machines. Our team consists of experts from all over the world and dozens of different fields, and together we want to change the way robots are programmed.

What you’ll be doing:

  • Frame the open-ended robotics problems into well-defined ML problems, develop and apply cutting-edge ML approaches (reinforcement learning, imitation learning, etc) to these problems;

  • Work with the ML infra team to scale the ML models to Nvidia-sized data pipelines and streamline them to run in real-time on the robotics.

  • Collaborate with QA, internal and external customers to validate, and extend the capabilities of our ML solution and take it to the next level.

  • Leverage NVIDIA research resources and outcomes like TAO, Cortex, etc to build an ecosystem of solutions that is world-class and scalable to multiple customers and use cases

  • Help identify, propose technology gaps across the product roadmap and vision

  • Mentor and guide the professional and technical development of the team members

What we need to see:

  • BS, MS or higher degree in Computer Science, Machine Learning, Robotics, similar technical field of study, or equivalent experience

  • 5+ years of working experience architecting end-to-end machine learning systems, from data collection and processing to model training and evaluation to deployment and monitoring

  • Proven Python and deep learning programming framework, e.g. TensorRT, Pytorch, TensorFlow, Keras, or similar

  • Experienced in working cross functionally with other teams and groups as part of an E2E autonomous system solution (such as HW, framework, algorithms, sim and more)

  • You have a thirst and ability to learn and adapt to new technologies

Ways to stand out from the crowd:

  • Familiarity working with reinforcement learning and imitation learning methods for robotics

  • Experience with CUDA and deep learning inference acceleration

  • Familiarity with cloud technologies and tooling for robotics applications

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most brilliant and talented people on the planet working for us. If you're creative, autonomous, and passionate about autonomous machines - we would love to hear from you!

The base salary range is 144,000 USD - 270,250 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. 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.#deeplearning