NVIDIA is seeking a Senior Software Engineer to build and scale AI software that enables breakthroughs in biology. You will be instrumental in developing tools for challenging problems like drug discovery and virtual cell simulation in a fast-paced, collaborative environment.

What You'll Do

Build reproducible test matrices across GPU SKUs, multi-node scale, and scientific parameters relevant to biology.
Create integration and performance test harnesses for large models.
Productionize AI algorithms, delivering production-quality models, libraries, and SDKs with a focus on observability, reproducibility, and versioning.
Develop and deploy distributed learning systems and tools to synchronize and debug workloads across multiple GPUs.
Collaborate with research and engineering teams across NVIDIA to transfer research to products and services, and upstream to the open-source ecosystem.
Dive into whatever is needed—infrastructure, glue code, tests, or documentation—to unblock the team.
Mentor emerging engineers.

What We're Looking For

3+ years of relevant professional experience.
BS/MS in Computer Science, Electrical Engineering, Math, Physics, or equivalent experience.
Expertise in Python and PyTorch.
Fundamentals in distributed training: DDP/FSDP, NCCL, mixed precision, data/pipe/tensor parallelism.
MLOps experience with Linux, bash, containers (Docker/NGC), and SLURM and/or Kubernetes.
CI/CD and automation experience with GitHub Actions, YAML workflows, runners, authentication, caching, artifact stores, and release pipelines.
Recognized ownership and technical leadership, excellent communication skills, and a bias for action.
Ability to work collaboratively in a tight-knit team environment.

Nice to Have

Experience in mixed applied-science and engineering teams, balancing engineering maturity with the agility to ship research releases quickly.
Systems intuition for compute efficiency: kernel optimization, IO/data pipelines, and performance tradeoffs.
Interest in biological and physical sciences and a desire to continuously learn.
Contributions to open source software projects.

Technical Stack

Programming: Python, PyTorch
Systems: Linux, bash
Containers & Orchestration: Docker, NGC, SLURM, Kubernetes
CI/CD: GitHub Actions
Communication: NCCL

Team & Environment

You'll be part of NVIDIA's digital biology team, working alongside researchers, developers, and domain experts in a fast-paced, collaborative environment.

Benefits & Compensation

Highly competitive salaries: $136,000 - $212,750 USD for Level 3, and $168,000 - $264,500 USD for Level 4.
Eligible for equity.
Comprehensive benefits package.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. We do not discriminate 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.