Senior Software Engineer, PyTorch - Deep Learning
Core
Design and build PyTorch components and production AI solutions that run efficiently on supercomputers with thousands to hundreds of thousands of GPUs.
Role type
Senior IC deep learning software engineer (PyTorch)
Builds
PyTorch framework components and scalable AI solutions for enterprise customers
Domain
Deep Learning / High-Performance Computing / GPU Systems
Deliverable
production ML models | product features
Required skills
C++ programming, systems software design, distributed parallel programming, large-scale software project development
Preferred skills
CUDA programming, deep learning compilers, open source contributions, deep learning modeling trends
Technologies
PyTorch, CUDA, C++, Python
Responsibilities
Design and build PyTorch components for supercomputers, collaborate with hardware teams to improve GPU performance, design and support production AI solutions, work with applied researchers to improve AI tools
Seniority
Senior, hands-on IC