Senior Software Engineer - Digital Biology

NVIDIA AI Software Engineer

NVIDIA has been redefining computer graphics, PC gaming, and accelerated computing for more than 30 years. Today, we're tapping the unlimited potential of AI to define the next era of computing. As part of the digital biology team, you'll build and scale the AI software that powers breakthroughs in drug discovery and biological sciences. If you love moving between science, code, and the infrastructure that makes it reliable, fast, and cost-efficient, you'll feel at home here.

What You Will Be Doing:

• Own testing-at-scale and reliability pipelines: Build hermetic, reproducible test matrices across GPUs SKUs, multi-node scale, and scientific parameters relevant to the biology space. Create integration and performance test harnesses for large models.
• Productize AI algorithms: Ship LLMs and geometric deep learning models into production-quality services and SDKs; ensure observability, reproducibility, and model/package versioning.
• Develop and deploy distributed learning systems and tools to synchronize and debug workloads on thousands of GPUs.
• Collaborate across teams: Partner with applied research, AI infrastructure, and full-stack teams; contribute to and upstream improvements across the open-source ecosystem.
• Be hands-on: Dive into whatever is needed—infra, glue code, tests, or docs—to unblock the team and ship.

What We Need To See:

• 3+ years of relevant experience.
• BS/MS in CS, EE, Math, Physics, or equivalent experience.
• Python, pytorch expertise
• CI/CD and automation experience: GitHub Actions, YAML workflows, runners, authentication, caching, artifact stores, and release pipelines.
• Distributed training fundamentals: DDP/FSDP, NCCL, mixed precision, data/pipe/tensor parallelism.
• MLOps for AI: Linux, bash, containers (Docker/NGC), SLURM and/or Kubernetes.
• Recognized for ownership and technical leadership, with excellent communication and a bias for action.

Ways To Stand Out From The Crowd:

• Worked in mixed applied science and engineering teams, familiar with the production-grade vs agility balance required to make forward progress.
• Systems intuition for compute efficiency: kernel optimization, IO/data pipelines, and performance tradeoffs.
• A natural interest in biological and physical sciences and desire to continuously learn-as-you-go.

With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us and, due to unprecedented growth, our exclusive engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 148,000 USD - 235,750 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4. You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until September 29, 2025. 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.