Sr. Engineer, Software - AI Compiler (AI)

TL;DR

Sr. Engineer, Software - AI Compiler (AI/MLIR): Developing the TT-Forge MLIR-based compiler to enable AI workloads on hirify.global hardware with an accent on compiler optimization and the bridge between high-level frameworks and low-level execution. Focus on creating custom dialects, transformation passes, and optimizing computational graphs for high-performance AI computing.

Location: Hybrid based in Belgrade, Serbia

Company

hirify.global is an AI technology company developing high-performance RISC-V CPUs and open-source AI computing platforms to redefine the computing paradigm.

What you will do

• Develop TT-Forge, an MLIR-based compiler enabling AI execution on various hirify.global hardware configurations.
• Build compiler technologies that bridge high-level frameworks to low-level hardware execution.
• Create and implement custom MLIR dialects (TTIR, TTNN, TTKernel) and transformation passes.
• Optimize computational graphs to enhance the performance of diverse AI models and workloads.
• Collaborate with hardware and software engineers in an open-source development environment.
• Utilize TT-Explorer for human-in-the-loop compiler optimization and tuning.

Requirements

• Fluency in C++ and Python.
• Experience building complex systems that bridge high-level frameworks to low-level execution.
• Experience with PyTorch, JAX, TensorFlow, or deep systems programming.
• Strong interest or experience in MLIR and modular compiler frameworks.
• Must be based in Belgrade, Serbia (Hybrid)
• Must be eligible to access U.S. export-controlled technology (compliance with EAR regulations)

Culture & Benefits

• Opportunity to work at the forefront of the AI hardware revolution.
• Collaborative environment that values curiosity and commitment to solving hard problems.
• Focus on open-source technology and transparent development.
• Competitive compensation package and benefits.
• Equal opportunity employer.