Senior Software Engineer - Automated Parallel Programming

NVIDIA's PyTorch Team is seeking a Senior Software Engineer to join their Automated Parallel Programming initiative. This role sits at the intersection of AI, parallel computing, and systems optimization, working on NVIDIA/Fuser (nvFuser) and the broader PyTorch ecosystem. The position offers the opportunity to work with cutting-edge technology and influence the future of GPU computing.

The role involves developing sophisticated code generation systems for machine learning frameworks, optimizing GPU performance, and collaborating with both hardware and software teams. You'll be working on tools used by millions of AI practitioners, making AI applications scalable to thousands of GPUs. The position requires expertise in parallel programming, particularly with NVIDIA's CUDA, and strong C++ development skills.

The ideal candidate will have advanced education in Computer Science or related fields, with significant experience in parallel programming and large-scale software development. Knowledge of deep learning compilers, distributed parallelism techniques, and contributions to major frameworks like PyTorch, TensorFlow, or similar systems are highly valued.

This is an opportunity to work with a diverse team at the forefront of AI and parallel computing, with competitive compensation including a base salary range of $184,000 - $287,500 USD, plus equity and benefits. The role offers the chance to influence not just software development but also future hardware design at NVIDIA, making it an exciting opportunity for those passionate about high-performance computing and AI.