Performance Engineer - GPU
Anthropic
5 months ago
New York, NY, USA +2 more
Mid Level / Senior
H1B Sponsor
Base Salary
$315k - $560k/yr
Responsibilities
- Architect and implement systems to enhance GPU performance for AI models.
- Maximize GPU utilization and develop optimizations for inference efficiency.
- Implement techniques from custom kernel development to distributed system architectures.
- Optimize end-to-end training and inference pipelines for language models.
- Profile and eliminate performance bottlenecks in production infrastructure.
Requirements
- Deep experience with GPU programming and optimization at scale.
- Impact-driven mindset with a focus on measurable performance breakthroughs.
- Ability to navigate complex systems from hardware interfaces to high-level ML frameworks.
- Experience with GPU Kernel Development, ML Compilers, and Performance Engineering.
- Bachelor's degree in a related field or equivalent experience.
Benefits
- Competitive compensation and benefits.
- Optional equity donation matching.
- Generous vacation and parental leave.
- Flexible working hours.
- Collaborative office space.
Tech Stack
PyTorch
Categories
AI & MLData Engineering