Software Engineer, Systems ML - Compilers / Backend

Job Title

Software Engineer, Systems ML - Compilers / Backend

Role Summary

Develop the compiler toolchain and backend code generation for custom deep-learning accelerators used in AR/VR systems. Work on a cross-disciplinary team of compiler engineers, ML researchers, firmware and ASIC designers to compile PyTorch models down to hardware binaries and optimize runtime performance.

Experience Level

Mid-level. Typical candidates have 1+ years of experience in compiler development, code optimization, or accelerating ML models on hardware.

Responsibilities

Deliver compiler passes, code generation, and optimizations targeting ML accelerators; collaborate with hardware and algorithm teams.

• Design and implement compiler passes and backend optimizations for ML accelerators.
• Generate and optimize code targeting custom inference hardware.
• Collaborate with algorithm researchers and hardware architects to co-design features and map ML graphs to hardware.
• Model dataflows and perform cost-benefit and power/performance estimations.
• Develop and extend ML libraries, intermediate representations, and export formats.
• Optimize kernels and runtime components for efficiency, scalability, and stability.
• Perform design and code reviews; diagnose and resolve compiler and system-level issues.
• Interface with other compiler teams to evaluate and incorporate innovations.

Requirements

Must-have skills and conditions for consideration.

• Proficient in Python and/or C/C++ for development, debugging, testing, and performance analysis.
• Experience with AI frameworks or accelerating models on hardware architectures (GPU, TPU, or custom AI ASICs).
• Experience in software design, performance analysis, and debugging complex codebases.
• Ability to obtain and maintain work authorization in the country of employment at time of hire.

Nice-to-have

• Experience contributing to compiler toolchains (LLVM, MLIR, Glow, GCC).
• Experience working with PyTorch, TensorFlow, or similar ML frameworks.
• Experience developing high-performance kernels or runtime components for inference accelerators.
• Experience collaborating with hardware, firmware, or algorithm teams.
• Knowledge of deep learning algorithms and responsible AI practices.

Education Requirements

Bachelor's degree in Computer Science, Computer Engineering, or a related technical field (or equivalent practical experience). Degree must be completed prior to joining Meta.

About the Company

Company: Meta Platforms

Headquarters: Menlo Park, California, United States

American technology company that develops social networking products (Facebook, Instagram, WhatsApp) and invests in virtual/augmented reality hardware and software through Reality Labs, focusing on connectivity, advertising, and immersive computing experiences.