Sr. Software Engineer, AI Compiler

Tenstorrent$100K — $500K *
Enterprise Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5-7 years of software engineering experience, focusing on compilers or similar technologies.
  • Proficient in C++ and Python, with a track record of developing intricate systems.
  • Familiarity with machine learning frameworks such as PyTorch, JAX, or TensorFlow.
  • Experience in compiler optimization and constructing computational graphs.
  • A collaborative mindset, thriving in cross-functional teams particularly in open-source environments.

Responsibilities

  • Drive the development of innovative AI compilation techniques for optimal performance.
  • Create and enhance custom compiler dialects and transformation passes.
  • Collaborate with hardware engineers to ensure seamless integration of software and hardware.
  • Contribute to the open-source compiler framework supporting various AI models and workloads.
  • Optimize computational graphs for efficiency and effectiveness in AI processes.

Benefits

  • Hybrid work model with flexibility for on-site work in Toronto.
  • Opportunity to work on groundbreaking AI and compiler technologies.
  • Collaborative and inclusive team culture with a focus on problem-solving.
  • Access to advanced tools and expertise to support personal and professional growth.
Full Job Description
Join the team revolutionizing AI computing at Tenstorrent. You'll work on TT-Forge, our MLIR-based compiler that enables developers to run AI on all configurations of Tenstorrent hardware using an open-source, performant, and general-purpose compiler. You will be at the forefront of the AI hardware revolution, building compiler technologies that redefine what's possible. This role is hybrid and based out of Toronto, ON. We welcome candidates at various experience levels for this role. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ from the one in this posting. **Who You Are** - A passionate software engineer eager to work on compiler technologies and the challenges of AI hardware, whether from compilers, systems, or broader software backgrounds. - Fluent in C++ and Python, with experience building complex systems that bridge high-level frameworks to low-level execution. - Excited by compiler optimization and machine learning, with experience in PyTorch, JAX, TensorFlow, or deep systems programming. - A collaborative problem-solver who thrives in open-source and enjoys working closely with hardware and software engineers. **What We Need** - A drive to solve novel challenges in AI compilation, from optimizing computational graphs to creating custom dialects and transformation passes. - Experience or strong interest in MLIR and how modular compiler frameworks connect AI models to advanced hardware. - Motivation to build technology that impacts the future of AI, knowing your work will enable the next wave of breakthroughs. **What You Will Learn** - How to build open-source compiler frameworks supporting diverse AI models and workloads, including training and multi-chip scaling. - Deep expertise in compiler technologies including custom MLIR dialects (TTIR, TTNN, TTKernel) and transformation passes. - New methods for human-in-the-loop compiler optimization using TT-Explorer, making advanced tuning tools usable by all developers. - How compiler technology powers Tenstorrent's mission to deliver affordable, open-source AI computing in a highly competitive space. Compensation for all engineers at Tenstorrent ranges from $100k - $500k including base and variable compensation targets. Experience, skills, education, background and location all impact the actual offer made. Tenstorrent offers a highly competitive compensation package and benefits.

About Tenstorrent

Tenstorrent is a semiconductor company that designs and develops computer processors for artificial intelligence and machine learning applications. The company's processors are designed to be energy-efficient and scalable, and are used by a range of businesses, from small startups to large enterprises. Tenstorrent was founded in 2016 and is headquartered in Toronto, Ontario.
Learn more about Tenstorrent
Size
51 employees
Industry
Founded
2016

Similar Jobs

More Jobs at Tenstorrent

More Enterprise Technology Jobs

Find similar Sr. Software Engineer, AI Compiler jobs: