Software Engineer, ML Compiler jobs in United States
cer-icon
Apply on Employer Site
company-logo

Meta · 1 day ago

Software Engineer, ML Compiler

Meta Reality Labs is the world leader in the design of virtual and augmented reality systems. They are seeking a software engineer to support the development of the compiler tool-chain for deep learning hardware components optimized for AR/VR systems, contributing to a full stack development effort compiling PyTorch models down to binaries for hardware accelerator blocks.

Computer Software
check
Comp. & Benefits

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

Qualification

Compiler developmentPythonC/C++Machine learning frameworksAI framework developmentDeep learning algorithmsMachine-code generationCross-functional communicationMentoring

Required

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 software design and programming 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

Preferred

Experience with machine-code generation or compiler back-ends
Experience working and communicating cross functionally in a team environment
Experience driving large efforts across multiple teams
Experience of developing in a mainstream machine-learning framework, e.g. PyTorch, Tensorflow or Caffe
Experience in deep learning algorithms and techniques, e.g., convolutional neural networks, recurrent networks, etc
Experience working on and contributing to an active compiler toolchain codebase, such as LLVM, MLIR, GCC, MSVC, Glow

Benefits

Bonus
Equity
Benefits

Company

Meta's mission is to build the future of human connection and the technology that makes it possible.

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Kathryn Glickman
Director, CEO Communications
linkedin
leader-logo
Christine Lu
CTO Business Engineering NA
linkedin
Company data provided by crunchbase