Senior Drug Discovery AI Engineer

Nvidia - Santa Clara, CA

Full-time - Senior

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

We are now looking for a Senior Deep Learning Engineer specializing in Drug Discovery AI at NVIDIA. This role is pivotal as NVIDIA has been at the forefront of transforming computer graphics, PC gaming, and accelerated computing for over 25 years. Our legacy of innovation is driven by extraordinary technology and exceptional talent. In this new era of computing, we are harnessing the unlimited potential of AI to redefine the landscape, where our GPUs serve as the brains behind computers, robots, and self-driving cars that can comprehend their surroundings. This is a unique opportunity to be part of a team that is doing what has never been done before, requiring vision, innovation, and the best talent in the industry. As an NVIDIAN, you will be immersed in a diverse and encouraging environment where everyone is inspired to perform at their best. Join us and discover how you can make a lasting impact on the world!

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

• 5+ years of relevant experience in deep learning or related fields.
• BS or MS in a quantitative field such as Statistics, Physics, Computer Science, Mathematics, or equivalent experience.
• Strong experience with Python programming language.
• Excellent skills in PyTorch framework.
• 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

• Equity options as part of compensation package.
• Comprehensive health benefits including medical, dental, and vision insurance.