Research Scientist Intern, TPU(PhD)

Research Scientist Intern, TPU(PhD)

Menlo Park, CA

Artificial Intelligence

• 1 more

Apply now

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.

Research Scientist Intern, TPU(PhD) 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.

Minimum Qualifications

• Currently has, or is in the process of obtaining, a PhD in computer science, computer engineering, electrical engineering, or a related field
• Must obtain work authorization in the country of employment at the time of hire and maintain ongoing work authorization during employment
• Proficient in Python
• Experience with system and kernel optimization

Preferred Qualifications

• Experience with GPU/TPU/ASICs
• Experience with Triton, Cuda, Cutedsl, Pallas
• Familiarity with performance modeling

About Meta

Meta builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps like Messenger, Instagram and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology. People who choose to build their careers by building with us at Meta help shape a future that will take us beyond what digital connection makes possible today—beyond the constraints of screens, the limits of distance, and even the rules of physics.

For those who live in or expect to work from California if hired for this position, please click here for additional information.

United States of America: $7,650/month to $12,667/month + benefits Individual compensation is determined by skills, qualifications, experience, and location. Compensation details listed in this posting reflect the base hourly rate, monthly rate, or annual salary only, and do not include bonus, equity or sales incentives, if applicable. In addition to base compensation, Meta offers benefits. Learn more about benefits at Meta.

Equal Employment Opportunity

Meta is proud to be an Equal Employment Opportunity employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or other applicable legally protected characteristics. You may view our Equal Employment Opportunity notice here.

Meta is committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, fill out the Accommodations request form.

Apply for this job Take the first step toward a rewarding career at Meta.

Apply now

APPLY NOW

Find your role

Explore jobs that match your skills and experience. Search by technology, team or location to find an opening that’s right for you.

View jobs

Meta AI

Recruiters can view your conversations with AI. Using AI is optional and does not impact the outcome of your application process. Learn more about AI usage and settings.

Sign up

to ask follow-up questions