The MTIA (Meta Training & Inference Accelerator) Software team is part of AI Infra PyTorch org. The team’s mission is to explore, develop and help productize high-performance software and hardware technologies for AI at datacenter scale. Team has been developing AI frameworks to accelerate Meta’s DL/ML workloads on the specialized MTIA AI accelerator hardware. As part of the AI acceleration software stack, we develop Pytorch compiler frontend for MTIA, Pytorch runtime for inference & training, high performance kernel libraries exploiting various hardware architectural features and tooling. We are looking for an engineering manager to support MTIA software stack development for inference and training platform.