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
2+ years of experience
AI · AR/VR

Description For Software Engineer, ML Compiler

Meta is seeking a talented Software Engineer specializing in ML Compiler development to join their innovative team. This role focuses on developing compiler tool-chains for cutting-edge deep learning hardware components optimized for AR/VR systems. As part of this position, you'll work alongside compiler experts, machine learning algorithm specialists, and hardware engineers to architect and implement a clean slate compiler.

The role offers an exciting opportunity to work on full stack development, compiling PyTorch models into binaries for custom hardware accelerator blocks. You'll be responsible for analyzing and designing compiler optimizations, working with research teams to map ML graphs to hardware implementations, and contributing to the development of machine-learning libraries and tools.

The ideal candidate should have a strong background in compiler development, with at least 2 years of experience in writing compilers or similar optimization software. Knowledge of Python and C/C++ is essential, as is experience with AI framework development and deep learning models. You'll be working at Meta, a company at the forefront of social technology and immersive experiences, helping to shape the future of digital connection beyond traditional screens.

This position offers competitive compensation ranging from $70,670 to $208,000 annually, plus bonus and equity opportunities. You'll be based in the San Francisco Bay Area, working on technology that pushes the boundaries of what's possible in AR/VR and machine learning. If you're passionate about compiler development, machine learning, and want to work on technology that will define the next evolution of social interaction, this role presents an excellent opportunity to make a significant impact.

Last updated 18 days 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 development
  • Conduct design and code reviews
  • Analyze and improve toolchains efficiency
  • Interface with other compiler-focused teams

Requirements For Software Engineer, ML Compiler

Python
  • Bachelor's degree in Computer Science, Computer Engineering, or relevant technical field
  • 2+ years experience writing compilers, runtime, or similar code optimization software
  • Experience in software design and programming in Python and/or C/C++
  • Experience in AI framework development
  • Experience in deep learning algorithms and techniques
  • Experience with machine-code generation or compiler back-ends
  • Experience working with machine-learning frameworks like PyTorch, MLIR, Tensorflow or Caffe

Benefits For Software Engineer, ML Compiler

Medical Insurance
Equity
  • Bonus
  • Equity
  • Benefits package available

Interested in this job?

Jobs Related To Meta Software Engineer, ML Compiler

Software Engineer (Machine Learning)

Machine Learning Software Engineer role at Meta focusing on developing scalable AI systems and algorithms for social technology platforms.

Research Engineer, Language - OCR

Research Engineer position at Meta focusing on Optical Character Recognition (OCR) technology development and research.

Software Engineer, Machine Learning

Meta is seeking a Machine Learning Software Engineer to develop scalable AI systems and algorithms for their social technology platforms.

Software Engineer – Machine Learning

Machine Learning Software Engineer role at Meta, developing AI systems and scalable solutions for next-generation social technology platforms.

Software Engineer, Machine Learning

Machine Learning Software Engineer role at Meta, developing scalable ML systems and algorithms for social technology applications.