- Menlo Park, California, United States
- On-site
- Internship · Intern
- 0+ years
The PyTorch TPU team focuses on enabling efficient and accurate inference of 1P models directly on TPU.
Responsibilities
- Develop, evaluate methodologies and systems for running LLM/Multi-modality models cost-effectively on TPU systems.
- Analyze and extract insights from traffic pattern to derive computational intrinsics that can be leveraged into actions in system and kernel optimization.
Requirements
- PhD in Computer Science
- PhD in Computer Engineering
- PhD in Electrical Engineering
- PhD in related field