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 biology. If you love moving between science, code, and the infrastructure that makes it reliable, fast, and cost‑efficient, you’ll feel at home here.
We're looking for a Senior Software Engineer who shares our vision for transforming healthcare. This is a unique opportunity to apply your expertise in a fast-paced, collaborative environment where you'll be instrumental in developing the tools that will redefine what is possible in drug discovery, virtual cell, and beyond. As part of our team, you'll work alongside other researchers, developers, and experts in the field to tackle some of the most challenging and impactful problems of our time.
What you'll be doing:
Building reproducible test matrices across GPU SKUs, multi-node scale, and scientific parameters relevant to the biology space.
Creating integration and performance test harnesses for large models.
Productionizing AI algorithms, delivering production-quality models, libraries, and SDKs, while ensuring observability, reproducibility, and versioning.
Developing and deploying distributed learning systems and tools to synchronize and debug workloads across multiple GPUs.
Collaborating with research and engineering teams across all of NVIDIA to transfer research to products and services, and upstreaming to the open-source ecosystem
Diving into whatever is needed—infra, glue code, tests, or docs—to unblock the team
Mentoring emerging engineers
What we need to see:
3+ years of relevant experience
BS/MS in CS, EE, Math, Physics, or equivalent experience.
Python, pytorch expertise
Distributed training fundamentals: DDP/FSDP, NCCL, mixed precision, data/pipe/tensor parallelism.
MLOps for AI: Linux, bash, containers (Docker/NGC), SLURM and/or Kubernetes.
CI/CD and automation experience: GitHub Actions, YAML workflows, runners, authentication, caching, artifact stores, and release pipelines.
Recognized for ownership and technical leadership, with excellent communication and a bias for action.
Ability to work together in a tight-knit team environment.
Ways to stand out from the crowd:
Experience working in mixed applied-science and engineering teams, understanding the balance between engineering maturity and agility to ship research releases quickly into the ecosystem
Systems intuition for compute efficiency: kernel optimization, IO/data pipelines, and performance tradeoffs.
Interest in biological and physical sciences and desire to continuously learn-as-you-go.
Contribution to open source software projects
With highly competitive salaries and a comprehensive benefits package, NVIDIA is widely considered to be one of the technology industry's most desirable employers. We have some of the most forward-thinking, resourceful and talented people in the world working with us and our engineering teams are growing fast in some of the most progressive fields: Digital Biology, Artificial Intelligence, and Autonomous Vehicles. Are you a creative and autonomous engineer with a real passion for machine learning, computational chemistry, data science & parallel computing? If so, 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 136,000 USD - 212,750 USD for Level 3, and 168,000 USD - 264,500 USD for Level 4.You will also be eligible for equity and benefits.