Research Engineer, Systems ML - Compilers - Reality Labs Silicon AI Team

Job description

We are seeking a software engineer to support the development of the compiler tool-chain for state-of-the-art deep learning hardware components optimized for AR/VR systems. You will be part of our efforts to architect, design and implement a clean slate compiler for this activity and will be part of a team that includes compiler, machine learning algorithms and software, firmware and ASIC experts. You will contribute to a full stack development effort compiling PyTorch models down to binaries for custom hardware accelerator blocks. This position is focused on the graph-level optimizations, including graph partitioning and memory planning, that are needed to efficiently deploy models on a variety of edge devices. Responsibilities: Contribute to the development of machine-learning libraries, intermediate representations, export formats, and analysis tools - Design and implement effective compiler passes and optimizations in PyTorch's intermediate representations - Analyze and improve the efficiency, scalability, and stability of our toolchains, and make sure they can be extended to new use cases - Generalize contributions to be applicable to as many devices as possible in the Reality...