Software Engineer, Systems ML - PyTorch Compiler, PyTorch Framework, PyTorch Performance

Meta is seeking a Senior Software Engineer to join their PyTorch Compiler team, focusing on developing core PyTorch 2.0 technologies and advancing ML compiler innovation. This role is central to the team that brought compiler technologies to PyTorch's core through PT2, which has gained significant industry recognition since its March 2023 release.

The position involves working on cutting-edge, open-source technology that aims to make PyTorch faster and more resource-efficient while maintaining its flexibility and ease of use. The team is committed to establishing PT2 as the industry's leading ML framework compiler. This role requires deep expertise in Python or C++ programming, with a strong background in compiler development, ML systems, and GPU performance optimization.

Key responsibilities include developing the PT2 compiler components, improving PyTorch performance for the entire community, and optimizing Generative AI models across various stages. The ideal candidate will have 4+ years of relevant experience and a bachelor's degree in Computer Science or a related field. Experience with PyTorch/PT2, Triton, MLIR, or similar technologies is highly valued.

The position offers competitive compensation ranging from $85,100 to $251,000 annually, plus bonus and equity opportunities. As part of Meta's AI Infrastructure team, you'll be working at the forefront of artificial intelligence development, contributing to technology that impacts billions of users worldwide. The role combines technical depth with the opportunity to collaborate with both internal teams and the open-source community, making it ideal for engineers passionate about both systems and machine learning.