Join Our PyTorch Compiler Team at Meta

Overview:

The PyTorch Compiler team at Meta is on a mission to enhance PyTorch's performance, focusing on efficiency and speed without compromising its renowned flexibility. Our groundbreaking PT2 technology has set a new benchmark in the ML framework landscape since its inception in March 2023, propelling PyTorch towards becoming the industry-leading ML compiler. We're looking for a Software Engineer to contribute to the PT2 compiler and push the boundaries of machine learning technology.

Key Responsibilities:

• Develop and enhance the PT2 compiler components such as TorchDynamo, TorchInductor, and PyTorch Core.
• Drive improvements in PyTorch performance, providing systematic solutions that benefit the global PyTorch community.
• Research and optimize generative AI models, focusing on pre-training, fine-tuning, and inference stages.
• Explore synergies between the PyTorch compiler and distributed PyTorch systems.
• Collaborate actively with PyTorch users to foster adoption and innovative uses of PT2 technologies.

Minimum Qualifications:

• BS/MS in Computer Science or a closely related field.
• Hands-on experience in compilers, ML systems, high-performance computing, and GPU optimization.
• Proficiency in programming with Python or CUDA.
• Pursuing or have obtained a Bachelor’s degree in Computer Science or a related technical field.

Preferred Qualifications:

• Ph.D. in Computer Science, Applied Mathematics, or related fields.
• Experience with PT2 technologies, Triton, MLIR, or direct contributions to PyTorch.
• Deep expertise in GPU performance and development of high-performance CUDA kernels.
• Proven track record of research and software engineering excellence, evidenced by publications, patents, or participation in relevant competitions.

About Meta:

Meta continues to redefine the landscape of connectivity and social interaction through innovative technologies. Since launching Facebook in 2004 and subsequent apps like Instagram and WhatsApp, we're now venturing into augmented and virtual reality to engineer the next era of social technology. At Meta, your work transcends traditional digital boundaries, creating new ways for people to connect, communities to thrive, and businesses to grow.

Compensation and Benefits:

The role offers a competitive salary ranging from $117,000 to $173,000 per year, plus bonuses, equity, and comprehensive benefits. We're committed to inclusive support and accommodations in our hiring process for diverse candidates.

If you are ready to shape the future of technology with us at Meta, advancing one of the most exciting fields in the world, apply to be part of our PyTorch Compiler team today!