Senior Deep Learning Compiler Engineer - XLA
Core
Develop compiler optimization algorithms and design AI compiler software features for next-generation GPUs to accelerate deep learning workloads.
Role type
Senior IC deep learning compiler engineer
Builds
High-performance, production-grade software for the XLA team and NVIDIA GPU backends
Domain
AI computing, high-performance computing, compiler technology
Deliverable
production ML models
Required skills
C/C++ programming, compiler optimization, performance analysis, distributed programming, GPU architecture knowledge, code generation
Preferred skills
CUDA/OpenCL programming, deep learning framework design, open-source compiler experience, mentoring
Technologies
XLA, JAX, OpenXLA, MLIR, LLVM, OpenAI Triton, CUDA, OpenCL, JAX, PyTorch, TensorFlow
Responsibilities
Develop compiler optimization algorithms for deep learning workloads, optimize inference and training performance for JAX and OpenXLA, design novel graph partitioning and tensor sharding techniques, perform performance tuning and analysis, code-generation for NVIDIA GPU backends, design user-facing features in JAX and related libraries
Seniority
Senior, hands-on IC