ML Frameworks Engineer – Inference Systems & Accelerators

Meta; Sunnyvale (CA); On-site; USD 122,000 - 181,000; Full time

Benefits offered by this job

Bonus

Equity

Benefits

Job summary

Meta is seeking a Software Engineer Systems Machine Learning - Frameworks to advance the PyTorch AI framework core compilers and optimize performance on diverse AI hardware. You will work on compiler optimizations, kernel acceleration, and ML framework stacks supporting state-of-the-art models.

The role emphasizes performance tuning for models across Ads, Reels, Feed, and GenAI, with opportunities to impact energy efficiency and scalability.

Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, or related technical field, or equivalent practical experience.
• Proficiency in Python and C/C++ programming.
• Experience in AI framework development or accelerating deep learning models on hardware.

Responsibilities

• Develop software stack for AI frameworks, compilers, and high-performance kernels.
• Contribute to PyTorch core compilers to support new AI hardware accelerators.
• Analyze networks and implement optimizer algorithms; performance-tune deep learning components.

Skills

Python

C/C++ programming

AI framework development

Education

Bachelor's degree in Computer Science/Engineering or related field

Tools

MLIR

LLVM

IREE

XLA

TVM

Halide

Job description

Meta is seeking a Software Engineer Systems Machine Learning - Frameworks to advance the PyTorch AI framework core compilers and optimize performance on diverse AI hardware. You will work on compiler optimizations, kernel acceleration, and ML framework stacks supporting state-of-the-art models.

The role emphasizes performance tuning for models across Ads, Reels, Feed, and GenAI, with opportunities to impact energy efficiency and scalability.