Senior Software Engineer - Python

November 7

Apply Now
Logo of NVIDIA

NVIDIA

GPU-accelerated computing β€’ artificial intelligence β€’ deep learning β€’ virtual reality β€’ gaming

10,000+

Description

β€’ Contribute to design and development of accelerated and distributed implementations of Python APIs for numerical computing β€’ Help develop and optimize GPU-accelerated and distributed implementations of Python numerical libraries β€’ Work closely with product management and partners to understand use cases and requirements β€’ Architect, prioritize, and develop accelerated and distributed implementations of numerical algorithms β€’ Design future-proof Python APIs for accelerated numerical/scientific computing libraries β€’ Analyze and improve performance of developed APIs β€’ Write effective, maintainable, and well-tested code for production use β€’ Contribute to the development of runtime systems that underlay the foundation of multi-GPU computing at NVIDIA

Requirements

β€’ BS, MS or PhD in Computer Science, Applied Math, Electrical Engineering or related field (or equivalent experience) β€’ 5+ years of relevant industry experience or equivalent academic experience after BS β€’ Excellent Python, C++ and CUDA programming skills β€’ Strong understanding of fundamental numerical methods, dense and sparse array computing β€’ Deep familiarity with Python numerical computing libraries (e.g. NumPy, SciPy), including accelerated implementations (e.g. CuPy, Jax.NumPy, NumS, cuNumeric) β€’ Experience developing and publishing Python libraries, following standard methodologies for pythonic API design β€’ Strong background with parallel programming and performance analysis

Benefits

β€’ Equity and benefits

Apply Now

Similar Jobs

Built byΒ Lior Neu-ner. I'd love to hear your feedback β€” Get in touch via DM or lior@remoterocketship.com