Deep Learning Compiler Engineer - Remote

Overview

The role involves developing and optimizing deep learning compilers to enhance machine intelligence capabilities, focusing on reproducibility and performance.

In Short

• Lower deep learning graphs from frameworks to an intermediate representation.
• Write novel algorithms for transforming compute graphs.
• Own areas in front-end, middle-end, or back-end compiler development.
• Contribute to the integration of deep learning kernels.
• Debug compilation passes and transformations.
• Work in an applied research environment with high autonomy.
• Collaborate with a small team focused on high performance.
• Engage in open-source contributions to compilers/frameworks.
• Participate in company retreats and receive relocation assistance.
• Enjoy competitive salary and benefits including health insurance.

Requirements

• Understanding of traditional compilers and graph traversals.
• Solid software engineering skills with production code experience.
• Knowledge of parallel programming, especially for GPUs.
• Experience with high-level and low-level IR optimizations.
• Excellent communication skills and self-motivation.
• Preferred experience with computer architecture and deep learning frameworks.
• Exposure to deep learning compiler frameworks.
• Kernel optimization experience for GPUs.

Benefits

• Competitive salary and equity options.
• Fully remote work with flexible time zones.
• Relocation assistance available.
• All expenses paid company retreats.
• Provision of necessary equipment.
• Paid sick leave.
• Comprehensive health, vision, and dental insurance.