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.
$56,250 - $173,000
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
1+ year 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 AR/VR systems team. This role focuses on developing compiler tool-chains for cutting-edge deep learning hardware components. You'll be part of a diverse team including compiler experts, ML algorithm specialists, and hardware engineers, working on compiling PyTorch models for hardware accelerators.

The position offers an exciting opportunity to work at the intersection of machine learning and compiler optimization, where you'll be responsible for designing and implementing sophisticated compiler passes, optimizing ML graph mappings to hardware, and contributing to the development of machine learning libraries and tools. You'll be working with state-of-the-art technologies in both AI and compiler development.

This role is perfect for someone who combines strong programming skills in Python and C++ with knowledge of AI frameworks and compiler development. You'll be part of Meta's mission to push the boundaries of AR/VR technology while working on challenging technical problems that impact billions of users. The position offers competitive compensation including base salary, bonuses, equity, and comprehensive benefits.

The ideal candidate should have experience with deep learning algorithms, compiler toolchains like LLVM or MLIR, and machine learning frameworks such as PyTorch or TensorFlow. You'll be working in a collaborative environment where cross-functional communication and technical expertise are equally valued.

Last updated 5 months ago

Responsibilities For Software Engineer, ML Compiler

  • Analyze and design effective compiler passes and optimizations
  • Research, implement, and enhance code generation targeting machine learning accelerators
  • Investigate new techniques 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
  • Experience in software design and programming in Python and/or C/C++
  • Experience in AI framework development or accelerating deep learning models on hardware architectures
  • Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
  • Must obtain work authorization in country of employment

Benefits For Software Engineer, ML Compiler

Equity
  • bonus
  • equity
  • benefits