NVIDIA is seeking a Senior Software Engineer for the cuEquivariance team, which focuses on accelerating geometric neural networks on NVIDIA GPUs. This role involves building the computational foundation for biological discovery and supporting researchers in various scientific fields. The engineer will work on CUDA kernel engineering and Python library development, collaborating with research teams and external developers. This position offers the opportunity to make a significant impact in the scientific community through GPU computing and graph-based deep learning.

• 6+ years of software engineering experience with a strong background in CUDA and GPU programming.
• Deep proficiency in C++ and Python, with experience building and shipping production libraries used by external developers.
• Good foundation in GPU computing, including memory hierarchy, warp-level execution, occupancy, and performance profiling methodology.
• Experience building or contributing to production scientific software libraries, ML frameworks, or developer-facing GPU APIs.
• Familiarity with concepts in geometric machine learning, such as equivariance and tensor products.
• BS/MS in Computer Science, Physics, Applied Mathematics, or a related field, or equivalent experience.

• Build, implement, and optimize CUDA kernels for equivariant neural network primitives targeting peak performance across NVIDIA GPU generations.
• Be responsible for the end-to-end delivery of GPU-accelerated geometric ML primitives from implementation to validated, production-quality software.
• Build and maintain the interfaces for PyTorch and JAX that expose cuEquivariance primitives to application developers and researchers.
• Drive CI/CD infrastructure for multi-GPU kernel builds, automated correctness testing, and performance regression tracking.
• Collaborate with Applied Science and research teams to evaluate new equivariant architectures and translate prototypes into production kernels.
• Engage directly with third-party framework developers and partners to align on interfaces and ensure delivered software integrates cleanly into production pipelines.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.