Software Engineer Systems Machine Learning - Frameworks

Meta is seeking a Software Engineer Systems Machine Learning specialist to join their Inference Enablement Team within the PyTorch AI framework organization. This role combines software systems expertise with machine learning, focusing on building foundational technology and frameworks that enable and optimize state-of-the-art model architectures on various hardware platforms.

The position involves working with cutting-edge model architectures across Meta's key products including Ads, Reels, Feed, Marketplace, Instagram, and GenAI. You'll be responsible for supporting model deployment on diverse hardware including CPU, GPU, and custom Silicon, while building scalable model publishing frameworks and platforms.

As a team member, you'll play a crucial role in helping Meta deploy important, state-of-the-art models for inference, contributing to significant improvements in Ads Revenue and Reels Watch Time, while also focusing on power efficiency optimization. The role offers an opportunity to work with PyTorch at a deep technical level and make meaningful contributions to Meta's AI infrastructure.

The position requires strong programming skills in Python and C/C++, along with experience in AI framework development or model acceleration on hardware architectures. Knowledge of GPU, CPU, or AI hardware accelerator architectures is highly valued, as is experience with kernel frameworks like Triton or CUDA.

Meta offers a competitive compensation package ranging from $56,250 to $173,000 annually, plus bonus and equity opportunities. The company provides comprehensive benefits and maintains a strong commitment to diversity and inclusion in its hiring practices.