Senior Drug Discovery AI Engineer

Nvidia - Santa Clara, CA

Full-time - Senior

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

NVIDIA has been at the forefront of transforming computer graphics, PC gaming, and accelerated computing for over 25 years. As a Senior Deep Learning Engineer focused on Drug Discovery AI, you will play a pivotal role in maintaining and improving NVIDIA's BioNemo and related software. This position involves scaling up the testing of framework software for deep learning model development and taking responsibility for major software features. You will also be tasked with productizing a wide range of AI algorithm implementations, including large language models and geometric deep learning algorithms. Collaboration is key in this role, as you will work closely with applied research, AI infrastructure, and full-stack software teams to drive innovation and excellence in our projects. In this dynamic environment, you will be immersed in a culture that values diversity and encourages every team member to perform at their best. NVIDIA is committed to pushing the boundaries of what is possible with AI, and your contributions will be essential in defining the next era of computing. This is an opportunity to make a lasting impact on the world through cutting-edge technology and innovative solutions.

Responsibilities

• Maintain and improve NVIDIA's BioNemo and related software.
• Scale up testing of framework software for deep learning model development.
• Take responsibility for major software features.
• Productize a wide range of AI algorithm implementations, including large language models and geometric deep learning algorithms.
• Collaborate with applied research, AI infrastructure, and full-stack software teams.

Requirements

• 8+ years of relevant experience.
• BS or MS in a quantitative field such as Statistics, Physics, Computer Science, Mathematics, or equivalent experience.
• Strong experience with Python.
• Excellent PyTorch skills.
• Good understanding of deep learning concepts.
• Recognized for technical leadership contributions, capable of self-direction, and willingness to learn from and guide others.

Nice-to-haves

• Experience implementing large language models and geometric deep learning models.
• Knowledge of implementing model parallelism techniques for large scale models.
• Outstanding communication skills.

Benefits

• Highly competitive salaries.
• Comprehensive benefits package including equity options.