10 days ago
Remote, Worldwide +2 moreMid Level
Base Salary
$165k - $259k/yr
Responsibilities
- Profile and optimize existing PyTorch and JAX models without compromising accuracy.
- Develop performance-critical code sections using frameworks like CUDA, Triton, and Warp.
- Liaise with NVIDIA to implement their tooling in our environment.
- Collaborate with scientists to identify impactful areas for optimization.
Requirements
- At least two years of professional experience in GPU optimization.
- Deep understanding of GPU programming fundamentals.
- Proven track record of optimization work demonstrated through observable artifacts like GitHub.
- Experience collaborating on software projects within multi-person teams.
Benefits
- Flexible work arrangements with the option for full-time remote work.
- Travel opportunities to San Francisco and New York offices for collaboration and conferences.
Tech Stack
PyTorch
Categories
AI & MLData Science