Senior Performance Engineer - Deep Learning
Core
Build and optimize libraries and tools (Transformer Engine, PyTorch, JAX) to accelerate Deep Learning model training and deployment on NVIDIA hardware.
Role type
Senior IC performance engineer (Deep Learning)
Builds
Optimized Deep Learning frameworks and libraries for LLMs and AI applications
Domain
AI / Deep Learning / High-Performance Computing
Deliverable
production ML models
Required skills
C++, Python, parallel systems programming, GPU programming, code optimization, computer architecture, software project development
Preferred skills
PyTorch, JAX, LLM architectures, performance profiling, CUDA, OpenAI Triton, cuBLAS, cuDNN, cuSOLVER, open-source contributions
Technologies
PyTorch, JAX, Transformer Engine, CUDA, OpenAI Triton, CuTeDSL, Pallas, cuBLAS, cuDNN, cuSOLVER
Responsibilities
Build and support Transformer Engine; Collaborate on systems research for model performance; Implement and optimize new Deep Learning models; Contribute to community benchmarks; Engage with open-source community and enterprise customers; Influence hardware and platform software design
Seniority
Senior, hands-on IC