AI & Compute Foundation Intern: Compiler & Kernel

Meta; Menlo Park (CA); On-site; USD 33,000 - 50,000; Full time

Job summary

Meta’s MTIA Software team in Menlo Park, CA offers 12–24 week internships focused on AI software and accelerator hardware. Interns contribute to AI framework cores, compilers, and high-performance kernel development for AI workloads.

Ideal candidates are pursuing a PhD, with strong experience in PyTorch, AI frameworks, and performance optimization. Start dates occur throughout the year, with a focus on datacenter-scale AI acceleration.

Qualifications

• Pursuing a PhD in a related field.
• Strong knowledge of AI frameworks and compilers.
• Experience with high-performance kernel development and acceleration.

Responsibilities

• Develop software stack for AI frameworks, compiler stacks, or kernel development.
• Contribute to PyTorch core compilers to support new AI hardware accelerators.
• Analyze networks and implement compiler optimization algorithms.

Skills

AI frameworks

Compiler optimization

High-performance kernel

Education

PhD in Computer Science/Engineering

Tools

PyTorch

Job description

Meta’s MTIA Software team in Menlo Park, CA offers 12–24 week internships focused on AI software and accelerator hardware. Interns contribute to AI framework cores, compilers, and high-performance kernel development for AI workloads.

Ideal candidates are pursuing a PhD, with strong experience in PyTorch, AI frameworks, and performance optimization. Start dates occur throughout the year, with a focus on datacenter-scale AI acceleration.