Software Engineer, Systems ML - Compilers / Backend

About the Role

The Software Engineer, Systems ML - Compilers / Backend role at Meta involves developing a compiler tool-chain for deep learning hardware components optimized for AR/VR systems. You will work closely with algorithm research teams and hardware architects to design and implement efficient compiler passes and optimizations, contributing to the full stack development of machine learning models.

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.
• Analyze and improve the efficiency, scalability, and stability of our toolchains.
• Conduct design and code reviews.

Minimum Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience.
• 2+ years experience developing compilers, toolchains, runtime, or similar code optimization software.
• Experience in software design and programming experience in Python and/or C/C++.
• Experience in AI framework development or accelerating models on hardware architectures.

Preferred Qualifications

• Experience of developing in a mainstream machine-learning framework, e.g. PyTorch, MLIR, Tensorflow or Caffe.
• Experience with machine-code generation or compiler back-ends for on-device inference workloads.
• Demonstrated ongoing AI skill development.
• Experience in deep learning algorithms and techniques.