Software Engineer, Systems ML - Compilers / Backend

Meta Careers; Sunnyvale, Northern (CA, KY); Hybrid; USD 180,000 - 280,000; Full time

Job summary

Meta Reality Labs in Sunnyvale, CA is hiring a Software Engineer for Systems ML focusing on compilers and backend development for AR/VR hardware accelerators. You will architect, design, and implement a clean slate compiler for deep learning workloads, targeting on-device ML accelerators.

As part of a multip-disciplinary team, you will contribute to building PyTorch model compilation pipelines, with a focus on performance, power efficiency, and programmability for wearable AI devices.

Qualifications

• Experience designing compiler passes and optimizations.
• Ability to map ML graphs to hardware implementations.
• Experience with cost-benefit analysis for silicon power/performance.

Responsibilities

• Analyze and design compiler passes and optimizations.
• Implement or enhance code generation for ML accelerators.
• Collaborate with hardware architects to maximize performance and power efficiency.

Skills

Compiler design

Code generation

ML graph mapping

Tools

PyTorch

Job description

# Software Engineer, Systems ML - Compilers / BackendSunnyvale, CA+3 locationsArtificial Intelligence+ 4 moreApply nowReality Labs (RL) focuses on delivering Meta's vision through Virtual Reality (VR), Augmented Reality (AR) and Wearable AI Devices. The compute performance and power efficiency requirements of our AI devices require custom silicon. Reality Labs Silicon team is driving the state of the art forward with breakthrough work in computer vision, machine learning, mixed reality, graphics, displays, sensors, and new ways to map the human body. Our chips will unlock personalized on-device AI capabilities and blend virtual, physical worlds on wearable devices. We believe the only way to achieve our goals is to look at the entire stack, from transistors, through architecture, firmware, and algorithms.We are seeking a software engineer to support the development of the compiler tool-chain for state-of-the-art deep learning hardware components optimized for AR/VR systems. You will be part of our efforts to architect, design and implement a clean slate compiler for this activity and will be part of a team that includes compiler, machine learning algorithms and software, firmware and ASIC experts. You will contribute to a full stack development effort compiling PyTorch models down to binaries for custom hardware accelerator blocks.---## Software Engineer, Systems ML - Compilers / Backend Responsibilities* Analyze and design effective compiler passes and optimizations. Implement and/or enhance code generation targeting machine learning accelerators* Work with algorithm research teams to map ML graphs to hardware implementations, model data-flows, create cost-benefit analysis and estimate silicon power and performance* Work with hardware architects to co-design hardware features that maximize performance, power efficiency and programmability* Contribute to the development of machine-learning libraries, intermediate representations, export formats, and analysis tools