ML Compiler Engineer — PyTorch & Hardware Accelerators

HBPad; Town of Texas (WI); On-site; USD 177,000 - 251,000; Full time

Benefits offered by this job

Bonus

Equity

Benefits

Job summary

Meta Reality Labs is seeking a Software Engineer to support the compiler tool-chain for AI hardware accelerators in AR/VR systems. You will design and implement a clean-slate compiler, optimize code generation for ML accelerators, and work with ML algorithms, software, firmware, and ASIC teams.

The role involves full-stack work from PyTorch models to binaries and mentoring others. Applicants should have a CS degree and 3+ years in compilers or similar optimization, with Python/C++ experience and

Qualifications

• Bachelor’s degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience.
• 3+ years experience writing compilers or similar code optimization software.
• Experience in Python and/or C/C++ for development, debugging, testing and performance analysis.
• Experience crossing multi-disciplinary boundaries to drive optimal system solutions.
• Experience in AI framework development or accelerating deep learning models on hardware architectures.

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
• Contribute to the development of machine-learning libraries, intermediate representations, export formats, and analysis tools
• Conduct design and code reviews. Evaluate code performance, debug, diagnose and drive resolution of compiler and cross-disciplinary system issues
• Analyze and improve the efficiency, scalability, and stability of our toolchains
• Interface with other compiler-focused teams to evaluate and incorporate their innovations and vice versa
• Mentor other engineers on improving the quality of engineering work in the broader team

Skills

Python

C/C++

Compiler design

Cross-disciplinary

ML framework

Education

Bachelor’s degree in Computer Science/Engineering or related field

Tools

LLVM

MLIR

GCC

MSVC

Glow

Job description

Meta Reality Labs is seeking a Software Engineer to support the compiler tool-chain for AI hardware accelerators in AR/VR systems. You will design and implement a clean-slate compiler, optimize code generation for ML accelerators, and work with ML algorithms, software, firmware, and ASIC teams.

The role involves full-stack work from PyTorch models to binaries and mentoring others. Applicants should have a CS degree and 3+ years in compilers or similar optimization, with Python/C++ experience and