Senior Deep Learning Compiler Engineer - PyTorch

Responsibilities

• As a key member of our team, you will be contributing directly to the future of accelerated AI. Your role will be dynamic and deeply technical, placing you at the center of compiler innovation. You will lead the design, implementation, optimization, and maintenance of the core compiler technologies that accelerate…
• What we need to see
• We are looking for engineers who are excited about building powerful, user-centric tools and are comfortable working in a fast-paced, collaborative environment. Here are some of the expertise we would like to see
• A Bachelor's, Master's, or Ph.D. in Computer Science or a related technical field (or equivalent experience).
• 8+ years of relevant work experience
• A strong command of Python and experience building complex, well-tested software systems.
• Hands-on experience with deep learning frameworks like PyTorch or JAX. You understand how models are built and where the performance challenges lie.
• A solid foundation in compiler concepts such as abstract syntax trees (ASTs), intermediate representations (e.g., SSA form), program analysis, and code generation.
• Excellent communication and collaboration skills, essential for working effectively in a distributed, open-source environment.
• Ways to stand out from the crowd

Languages