About the Role

This role involves developing compiler tool-chains for deep learning hardware in AR/VR systems. The engineer will architect and implement compilers to optimize PyTorch models for hardware accelerators, working with cross-disciplinary teams.

Key Responsibilities

• Analyze and design compiler passes and optimizations for ML accelerators
• Map ML graphs to hardware implementations and model data-flows
• Develop ML libraries, intermediate representations, and analysis tools
• Conduct design/code reviews and debug system issues
• Improve efficiency and scalability of toolchains

Required Skills & Qualifications

Must Have:

• Bachelor's degree in Computer Science or equivalent experience
• 3+ years experience writing compilers or code optimization software
• Programming experience in Python and/or C/C++
• Experience accelerating deep learning models on hardware architectures

Nice to Have:

• Experience with machine-code generation or compiler back-ends
• Experience in mainstream ML frameworks like PyTorch or TensorFlow
• Experience with deep learning algorithms like CNNs or RNNs
• Experience contributing to compiler toolchains like LLVM or MLIR

Benefits & Perks

• bonus
• equity
• benefits