
Research Scientist Intern, TPU(PhD)
Meta · Menlo Park, CA
Applications are made on Meta's own site. We never sit in the middle.
Job description
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.
Other jobs you might want
Meta
Producer - Creative X
Los Angeles, CA
Meta
Global Supply Chain Manager, ODM Network Switches Infrastructure Supply Chain
Remote, US
Meta
Network Implementation Engineer - Fiber Delivery
Menlo Park, CA
Meta
HR Business Partner, Meta Superintelligence Labs (MSL)
Menlo Park, CA
Meta
AI Specialist - VR
Redmond, WA
Meta
UX Research Scientist, User Safety
Burlingame, CA