Senior Machine Learning Engineer - Chemistry and Materials Science

This role is based in Santa Clara, CA or remote.

Compensation:

$152,000 - $287,500/year

Responsibilities:

• Develop machine learning frameworks including NVIDIA PhysicsNeMo and ALCHEMI.
• Validate applications built using the framework across NVIDIA's products.
• Stay updated with the latest research in deep learning techniques.

Requirements:

• BS or MS degree (PhD preferred) in relevant technical fields.
• 5+ years of relevant experience.
• Strong Python programming skills.
• Familiarity with AI frameworks like PyTorch and Jax.
• Experience applying AI/ML techniques in chemistry and materials science.