Staff Machine Learning Engineer, Model Training

10 hours ago


Belgrade, Central Serbia Tenstorrent Full time €40,000 - €80,000 per year

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.

We're looking for a technical leader to join our team in Belgrade, Serbia. This role is about owning the technical vision for our model training stack, architecting core compiler infrastructure, and leading a team of exceptional engineers by example. This is a hybrid role where you'll have a direct impact on the future of our hardware and software.

This role is hybrid based in Belgrade, Serbia.

We welcome candidates at various experience levels. 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 proven leader with 5+ years of experience driving technical roadmaps and mentoring engineering teams in the ML industry.
  • An expert in neural networks, with extensive, hands-on experience building and scaling distributed training systems.
  • Fluent in Python and C++, with deep knowledge of modern AI frameworks like PyTorch or JAX. Experience with compiler infrastructure or custom hardware is a major plus.

What We Need

  • Technical ownership of the training roadmap for our compiler, driving strategic planning and defining priorities for the team.
  • Scaling out to various modes of distributed model training with architectural oversight and performance optimization.
  • Benchmarking, analysis, and optimization of training performance across Tenstorrent's hardware and software stack.
  • Leadership through mentorship and by driving critical cross-team initiatives to ship reliable, scalable software.

You'll Learn How To

  • Architect and implement compiler optimizations specifically for training workloads on our custom hardware.
  • Identify bottlenecks, debug performance issues, and improve developer workflows across our stack.
  • Adapt and evolve our entire software infrastructure and leverage our hardware capabilities to stay ahead of emerging AI models.

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.



  • Belgrade, Central Serbia EchoTwin AI Full time €40,000 - €80,000 per year

    Company OverviewEchoTwin AI is pioneering AI-driven infrastructure intelligence, redefining how cities are managed. Powered by a proprietary visual intelligence engine with full spatial reasoning, EchoTwin transforms municipal fleets into mobile urban sensors—creating living digital twins that provide real-time insights into infrastructure, compliance, and...


  • Belgrade, Central Serbia AMD Full time €40,000 - €80,000 per year

    WHAT YOU DO AT AMD CHANGES EVERYTHINGAt AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create...


  • Belgrade, Central Serbia Advanced Micro Devices, Inc Full time €80,000 - €120,000 per year

    WHAT YOU DO AT AMD CHANGES EVERYTHING At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create...


  • Belgrade, Central Serbia Everseen Full time €40,000 - €80,000 per year

    Everseen: A leader in vision AI solutions for the world's leading retailers.The RoleWe are seeking a Machine Learning Platform/Backend Engineer to design, build, and maintain scalable infrastructure that empowers our data scientists and machine learning engineers to develop, train, benchmark, and monitor machine learning models efficiently. You will be...


  • Belgrade, Central Serbia Everseen Full time €45,000 - €90,000 per year

    Everseen: A leader in vision AI solutions for the world's leading retailers.The RoleWe are seeking a Machine Learning Platform/Backend Engineer to design, build, and maintain scalable infrastructure that empowers our data scientists and machine learning engineers to develop, train, benchmark, and monitor machine learning models efficiently. You will be...


  • Belgrade, Central Serbia Finom Full time €40,000 - €80,000 per year

    About FinomFinom is a European tech startup headquartered in Amsterdam, and we're on a journey towards revolutionizing the financial landscape for entrepreneurs worldwide. Our mission is to develop an all-in-one financial B2B solution that integrates banking functions, accounting, financial management, and invoicing into a seamless, mobile-first platform.We...


  • Belgrade, Central Serbia Everseen Full time €45,000 - €90,000 per year

    Everseen: A leader in vision AI solutions for the world's leading retailers.The RoleWe are seeking a Full-Stack Engineer to be a key member of the Everseen ML Operations team. As part of that team, you will own the design and implementation of the front-end and back-end components of the Everseen internal ML platform, supporting the AI researchers...


  • Belgrade, Central Serbia INFOMEDIJI d.o.o. Full time €60,000 - €120,000 per year

    Join Our Vision: Establishing Spatial Interactive Media and Leading Global VR Video Streaming Join the team with your passion fuelling innovation and productivity. This isn't just another job.JOB SUMMARY:We are looking for a seasoned ML CV expert to join our team. If you're passionate about computer science and VR video streaming we'd love to hear from...


  • Belgrade, Central Serbia Tenstorrent Full time

    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...

  • Staff Engineer

    10 hours ago


    Belgrade, Central Serbia Sokin Full time €80,000 - €120,000 per year

    Staff EngineerWe are searching for a Staff Engineer that can play a critical role in designing, developing and delivering scalable and efficient software solutions. Serving as a technical leader you'll help mentor your team members, collaborate with cross-functional teams to guide architectural decisions, setting technical standards and ensuring high code...