Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta builds technologies that help people connect, find communities, and grow businesses through social platforms like Facebook, Instagram, WhatsApp, and immersive AR/VR experiences.
$85,100 - $251,000
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
8+ years of experience
AI

Description For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Meta is seeking a Software Engineer for their MTIA (Meta Training & Inference Accelerator) Software team, which is part of the industry-leading PyTorch AI framework organization. This role offers an exciting opportunity to work on cutting-edge AI technology at one of the world's leading tech companies.

The position involves developing crucial components of Meta's AI infrastructure, including PyTorch framework components, AI compiler and runtime systems, and high-performance kernels. You'll be working directly with AI researchers to analyze and optimize deep learning models for MTIA hardware platforms, while also collaborating with hardware design teams to develop efficient compiler optimizations.

This is an ideal role for an experienced software engineer with a strong background in AI frameworks and systems optimization. The position offers the chance to impact Meta's AI infrastructure directly, working on technology that powers products used by billions of people globally. You'll be at the forefront of AI hardware-software co-design, developing solutions for next-generation AI challenges.

The role requires expertise in C/C++ programming and AI framework development, with preferred experience in GPU/CPU architectures and frameworks like PyTorch, TensorFlow, or ONNX. You'll be working in Bellevue, WA, with competitive compensation ranging from $85,100 to $251,000 annually, plus bonus and equity opportunities.

Meta offers a comprehensive benefits package and a collaborative environment where you'll work with leading experts in AI and machine learning. This position provides an excellent opportunity to shape the future of AI infrastructure while working on challenging technical problems at scale. The role combines deep technical work with the opportunity to influence the direction of Meta's AI capabilities, making it an exciting opportunity for someone passionate about both AI and systems engineering.

Last updated 18 days ago

Responsibilities For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

  • Development of SW stack with focus on AI frameworks, compiler stack, high performance kernel development
  • Contribute to the development of PyTorch AI framework core compilers
  • Analyze deep learning networks, develop & implement compiler optimization algorithms
  • Collaborate with AI research scientists to accelerate deep learning models
  • Performance tuning and optimizations of deep learning framework & software components

Requirements For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Python
  • Proven C/C++ programming skills
  • Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
  • Experience in AI framework development or accelerating deep learning models on hardware architectures

Benefits For Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Medical Insurance
Dental Insurance
Vision Insurance
  • Bonus
  • Equity
  • Benefits Package

Interested in this job?

Jobs Related To Meta Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Software Engineer, Systems ML - GenAI Evals Platform

Senior Software Engineer role at Meta focusing on building and optimizing GenAI evaluation platforms and infrastructure for large language models.

Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Senior Software Engineering role at Meta focusing on AI compiler development and optimization for machine learning frameworks and hardware acceleration.

Software Engineer, Computer Vision - XR World AI

Senior Software Engineer role at Meta focusing on computer vision and AI for XR/AR applications, developing 3D mapping solutions.

Research Engineer - FAIR, Agents

Research Engineer position at Meta's FAIR team focusing on LLM Agents development, combining AI research with practical engineering implementation.

Software Engineer, Systems ML - Frameworks / Compilers / Kernels

Senior Software Engineering role at Meta focusing on AI compiler development and optimization for machine learning infrastructure, requiring expertise in C++, AI frameworks, and hardware acceleration.