Software Engineer, Systems ML - Compilers
Meta PlatformsJob Title
Software Engineer, Systems ML - Compilers
Role Summary
The Reality Labs Silicon compiler team develops compiler toolchains that compile PyTorch models to optimized binaries for custom AI accelerators in AR/VR and wearable devices. This role architects and implements compiler infrastructure, optimization passes, and code generation for on-device ML targets.
Experience Level
Mid-level - requires approximately 3+ years of relevant experience.
Responsibilities
Key responsibilities include designing, implementing, and owning compiler infrastructure and coordinating with hardware and ML teams.
- Architect and implement ML compiler infrastructure: IRs, optimization passes, and code generation for custom AI accelerators.
- Develop LLVM/MLIR-based toolchains to compile PyTorch models into optimized binaries.
- Design compiler transformations constrained by hardware architecture to maximize performance and power efficiency.
- Collaborate with hardware architects to co-design compiler features for edge devices.
- Analyze and improve compiler toolchain efficiency, scalability, and stability across targets.
- Lead technical roadmaps, coordinate cross-team execution, and mentor engineers on compiler patterns.
- Conduct design and code reviews, evaluate performance, and resolve cross-disciplinary system issues.
- Interface with other compiler-focused teams to evaluate and integrate innovations.
Requirements
Must-have technical skills and experience; preferred items listed separately.
- 3+ years developing compilers, toolchains, or code-optimization software.
- Proven experience with LLVM, MLIR, or similar compiler frameworks.
- Experience designing intermediate representations and implementing optimization passes.
- Familiarity with hardware architectures such as GPUs, TPUs, or custom AI accelerators.
- Proficiency in C++ for compiler and systems-level development.
- Experience leading end-to-end technical design and delivery of compiler infrastructure across teams.
Education Requirements
Bachelor's degree in Computer Science, Computer Engineering, or a related technical field, or equivalent practical experience.
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.
