• Job Description

  Req#: JR1975719

  NVIDIA is using the power of high performance computing and AI to accelerate digital chemistry, biology and atomistic modeling! We are seeking passionate and hardworking individuals to help us realize our mission. As an Applied Geometric DL Scientist Intern, you will join a research and development team passionate about infrastructure development and collaborations with industry and academic partners. This position provides the opportunity to research, implement, productize, and deliver deep learning algorithms for applications to atomistic modeling. The team engages in applied research and then helps productize this work.

  What you will be doing:

  • Prototype and build deep learning algorithms for graph and geometric deep learning in atomistic modeling

  • Design metrics for and assist with the evaluation of model predictions and results

  • Know the latest research and identify ways to capitalize on new advancements, either as applied research projects or by directly integrating into product development

  • Collaborate with multiple AI infrastructure and research teams

  • Find opportunities to incorporate advances in the field and other NVIDIA products into our infrastructure

  What we need to see:

  • Pursuing a PhD Degree in a quantitative field such as Statistics, Physics, Computational Biology, Quantum Chemistry, Computer Science, Mathematics (or a related field), or meaningful experience.

  • Knowledge in deep learning and machine learning in atomistic modeling or related applications with geometric learning tasks

  • Experience with Python and deep learning frameworks (PyTorch, TensorFlow, Jax, Warp) and relevant deep learning libraries (e.g. PyG, DGL, e3nn)

  • Experience with modeling biological, chemical or material science systems, such as small molecules, proteins, solids, surfaces and experience with related modeling and simulation tools

  • Recognition for technical leadership contributions, understanding of the fundamentals, capable of self-direction, and willingness to learn from others.

  Ways to stand out from the crowd:

  • Familiarity with latest advances in geometric and/or generative atomistic deep learning models in biological sciences, such as DiffDock, or similar equivariant models from other application areas

  • Experience with protein or small molecule simulation tools that use atomistic or coarse grained interaction models such as OpenMM, GROMACS, LAMMPS

  • Background with ML model design for atomistic systems and with infrastructure building for model design, deployment or data generation stages (e.g. active learning frameworks)

  • Experience with C/C++, CUDA, docker

  • Experience with open source development

  • Relevant publication history and/or conference attendance

  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 in the world working for us. If you are creative and autonomous and you love the challenge of constantly innovating in a growing and quickly expanding field, we want to hear from you.

  The hourly rate for our interns is 19 USD - 93 USD. Our internship hourly rates are a standard pay determined based on the position and your location, year in school, degree, and experience.

  You will also be eligible for Intern 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.

• About the company

  9637389 Nvidia Corporation is an American multinational technology company incorporated in Delaware and based in Santa Clara, California.

Notice