The opportunity
Pioneering the next generation of AI requires breakthrough innovations in GPU performance and systems engineering. As a GPU Performance Engineer, you'll architect and implement the foundational systems that power Claude and push the frontiers of what's possible with large language models.
What you'll do
Have deep experience with GPU programming and optimization at scale
Are impact-driven, passionate about delivering measurable performance breakthroughs
Can navigate complex systems from hardware interfaces to high-level ML frameworks
Enjoy collaborative problem-solving and pair programming
Want to work on state-of-the-art language models with real-world impact
Care about the societal impacts of your work
What they're looking for
- Thrive in ambiguous environments where you define the path forward
- GPU Kernel Development: CUDA, Triton, CUTLASS, Flash Attention, tensor core optimization
- ML Compilers & Frameworks: PyTorch/JAX internals, torch.compile, XLA, custom operators
- Performance Engineering: Kernel fusion, memory bandwidth optimization, profiling with Nsight
