TensorFlow Jobs in Belgrade.
All active TensorFlow roles based in Belgrade.
Pick a job to read the details
Tap any role on the left — its description and apply link will open here.
Sr. Engineer, Software - AI Compiler
Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. With AI redefining the computing paradigm, solutions must evolve to unify innovations in software models, compilers, platforms, networking, and semiconductors. Our diverse team of technologists have developed a high performance RISC-V CPU from scratch, and share a passion for AI and a deep desire to build the best AI platform possible. We value collaboration, curiosity, and a commitment to solving hard problems. We are growing our team and looking for contributors of all seniorities.
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 based in Belgrade, Serbia.
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.
Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.
This offer of employment is contingent upon the applicant being eligible to access U.S. export-controlled technology. Due to U.S. export laws, including those codified in the U.S. Export Administration Regulations (EAR), the Company is required to ensure compliance with these laws when transferring technology to nationals of certain countries (such as EAR Country Groups D:1, E1, and E2). These requirements apply to persons located in the U.S. and all countries outside the U.S. As the position offered will have direct and/or indirect access to information, systems, or technologies subject to these laws, the offer may be contingent upon your citizenship/permanent residency status or ability to obtain prior license approval from the U.S. Commerce Department or applicable federal agency. If employment is not possible due to U.S. export laws, any offer of employment will be rescinded.
Ready to apply?
Apply to Tenstorrent