Research Scientist Intern, PyTorch On-Device

PyTorch is Meta’s deep learning framework for fast, flexible AI/ML experimentation used across industry and backing all of Meta’s ML workloads. Our team brings PyTorch to edge devices through the use of compilers, an optimized runtime, and leveraging unique mobile hardware (CPU, GPU, NPU, DSP) for inference and training. Team scope: - Develop ExecuTorch as the PyTorch solution for on-device AI - Partner with Reality Labs to run AI on AR/VR hardware - Partner with Meta family of apps for running AI on iOS and Android - Partner with hardware vendors on high-performance AI kernels - Provide on-device AI inference, feature stores, benchmarking, and model delivery. Our internships are twelve (12) to twenty-four (24) weeks long and we have various start dates throughout the year.