The PyTorch TPU team focuses on enabling efficient and accurate inference of 1P models directly on TPU. The work spans performance modeling, sharding and parallelism design, kernel authoring and automating with agents. Our internships are twelve (12) to twenty four (24) weeks long, and we have various start dates throughout the year.
Job details are sourced from the employer's original posting.
Open job postingAbout the company
Meta builds technologies that help people connect, find communities, and grow businesses.