Key facts Location: Toronto, ON, Canada (hybrid) Engagement: Full-time What you'll do Tackle novel problems in AI compilation, from optimizing computational graphs to designing custom dialects and transformation passes Build open-source compiler frameworks that support wide-ranging AI models and workloads, including training and multi-chip scaling Develop expertise in custom MLIR dialects such as TTIR, TTNN, and TTKernel Contribute to TT-Explorer, a human-in-the-loop compiler optimization tool designed to make advanced tuning accessible to all developers Collaborate closely with hardware and software teams in an open-source environment Requirements Proficiency in C++ and Python Experience building complex systems that connect high-level frameworks to low-level execution Familiarity with at least one of PyTorch, JAX, TensorFlow,
or deep systems programming Interest or experience in MLIR and modular compiler frameworks that link AI models to specialized hardware Ability to work collaboratively across hardware and software disciplines Nice to have Background in compilers, systems engineering, or related software domains Prior contributions to open-source projects Skills & tools C++ Python MLIR PyTorch, JAX, or TensorFlow Compiler optimization techniques Computational graph transformations Practical notes Total compensation ranges from $100k to $500k, including base salary and variable targets; actual offers depend on experience, skills, education, background, and location Employment is contingent on eligibility to access U.S. export-controlled technology under U.S. export laws including the Export Administration Regulations (EAR); applicants from EAR Country Groups D:1, E1, and E2 may require prior license approval from the U.S.
Commerce
Department or another federal agency Tenstorrent is an equal opportunity employer