Taro Logo

Software Engineer, ML Compiler

Meta builds technologies that help people connect, find communities, and grow businesses through social technology and immersive experiences like AR/VR.
$70,670 - $208,000
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
2+ years of experience
AI · AR/VR
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Software Engineer, ML Compiler

Meta is seeking a Software Engineer specializing in ML Compiler development to join their innovative AR/VR systems team. This role focuses on developing compiler tool-chains for cutting-edge deep learning hardware components. As part of a diverse team including compiler, machine learning algorithms, software, firmware, and ASIC experts, you'll work on architecting and implementing a clean slate compiler.

The position involves compiling PyTorch models to binaries for custom hardware accelerators, requiring expertise in both machine learning and compiler development. You'll collaborate with algorithm research teams to optimize ML implementations, conduct performance analysis, and improve toolchain efficiency.

This is an exciting opportunity to work at Meta, a company at the forefront of social technology and immersive experiences. You'll be part of the team pushing boundaries in AR/VR technology, working beyond traditional 2D screens to shape the future of digital connection.

The role offers competitive compensation ranging from $70,670 to $208,000 annually, plus bonus and equity opportunities. Meta provides a comprehensive benefits package and maintains a strong commitment to equal employment opportunity and workplace accommodations.

The ideal candidate will have experience with deep learning frameworks, compiler toolchains, and strong programming skills in Python and C++. This position offers the chance to work on groundbreaking technology while contributing to Meta's mission of connecting people through innovative social platforms and immersive experiences.

Last updated 4 months ago

Responsibilities For Software Engineer, ML Compiler

  • Analyze and design effective compiler passes and optimizations
  • Implement and enhance code generation targeting machine learning accelerators
  • Work with algorithm research teams to map ML graphs to hardware implementations
  • Contribute to machine-learning libraries, intermediate representations, export formats, and analysis tools
  • Conduct design and code reviews
  • Analyze and improve efficiency, scalability, and stability of toolchains
  • Interface with other compiler-focused teams

Requirements For Software Engineer, ML Compiler

Python
  • Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
  • 2+ years experience writing compilers, runtime, or similar code optimization software
  • Experience in software design and programming experience in Python and/or C/C++
  • Experience in AI framework development or accelerating deep learning models on hardware architectures

Benefits For Software Engineer, ML Compiler

Equity
  • bonus
  • equity
  • benefits

Interested in this job?