Software Engineer, Systems ML - Compilers

About the position

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.

Requirements

• Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience
• 2+ years experience developing compilers, runtime, 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 in AI framework development or accelerating models on hardware architectures
• 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
• Experience with machine-code generation or compiler back-ends
• Experience working and communicating cross functionally in a team environment
• Experience developing in a mainstream machine-learning framework, e.g. PyTorch, MLIR, Tensorflow or Caffe