Research Scientist Intern, MSL Infra Kernels & Optimizations (PhD)

Join Meta SuperIntelligence Labs (MSL) Infra Kernels & Optimizations team as a PhD Research Scientist Intern to explore, develop, and productionize high-performance model, software & hardware technologies for MSL's mission to build superintelligence. Responsibilities include exploring ML kernels, optimizing parallelisms, co-designing novel model architectures, and collaborating with partner teams to bring learnings into production.

Requirements

• Currently has, or is in the process of obtaining, a PhD degree in Computer Science, Computer Vision, Generative AI, NLP, or equivalent practical experience
• Specialized experience in one or more of the following areas: Accelerators/GPU architectures, High Performance Computing (HPC), Machine Learning Compilers, Training/Inference ML Systems, Model Compression, Communication Collectives, ML Kernels/Operator optimizations, Machine learning frameworks (e.g. PyTorch) and SW/HW co-design
• Experience developing AI-System infrastructure or AI algorithms in C/C++ or Python

Benefits

• $7,650/month to $12,134/month