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

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 3 days 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

Interested in this job?

Jobs Related To Meta Software Engineer, ML Compiler

Software Engineer, Machine Learning

Machine Learning Engineer position at Meta focusing on developing scalable ML solutions and recommendation systems.

Research Engineer - Contextual AI Devices

Research Engineer position at Meta's Reality Labs, focusing on developing AI-driven interaction solutions for next-generation devices.

Software Engineer, ML Compiler

ML Compiler Engineer role at Meta focusing on developing compiler toolchains for AR/VR systems and deep learning hardware acceleration.

Software Engineer, Machine Learning

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

Software Engineer, Machine Learning

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