Member of Technical Staff (GPU Performance Engineer)
Core
Design and implement improvements to training infrastructure, optimize model performance, and work on post-training processes including reinforcement learning and fine-tuning.
Role type
Senior IC GPU Performance Engineer
Builds
Training infrastructure, model serving infrastructure, and optimized large-scale deep learning models
Domain
Artificial Intelligence / Deep Learning / GPU Computing
Deliverable
production ML models
Required skills
Python, PyTorch, CUDA, C++, large-scale model training, GPU workload profiling, cluster scaling (Slurm/Kubernetes), performance tuning
Preferred skills
Experience with reinforcement learning, fine-tuning, low-level GPU code optimization
Technologies
Python, PyTorch, CUDA, C++, Slurm, Kubernetes
Responsibilities
Design and implement improvements to training infrastructure; optimize performance of GPU-accelerated workloads; work on post-training processes including reinforcement learning and fine-tuning; contribute to improving efficiency and scalability of model serving infrastructure
Seniority
Senior, hands-on IC
