INTERNSHIP DETAILS

Machine Learning Intern, AI Compiler - Model Training (Serbia)

CompanyTenstorrent University Jobs
LocationBelgrade
Work ModeOn Site
PostedMarch 4, 2026
Internship Information
Core Responsibilities
Interns will participate in the implementation and validation of model training pipelines and support benchmarking and performance optimization of various model architectures. They will also contribute to scaling out distributed model training and assist in integrating Tenstorrent software with widely used tools in the machine learning community.
Internship Type
full time
Company Size
1109
Visa Sponsorship
No
Language
English
Working Hours
40 hours
Apply Now →

You'll be redirected to
the company's application page

About The Company
Tenstorrent is a next-generation computing company that builds computers for AI. Headquartered in the U.S. with offices in Austin, Texas, and Silicon Valley, and global offices in Toronto, Belgrade, Seoul, Tokyo, and Bangalore, Tenstorrent brings together experts in the field of computer architecture, ASIC design, RISC-V technology, advanced systems, and neural network compilers. Tenstorrent is backed by Eclipse Ventures and Real Ventures, Archerman Capital, Samsung Catalyst Fund, and Hyundai Motor Group among others. Join us: www.tenstorrent.com/careers.
About the Role

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.

As an intern at Tenstorrent, you will experience how we do engineering for 3+ months, working on a carefully defined and real engineering project, guided by a dedicated mentor and a manager, and as a part of a team.

This role is on-site based in Belgrade, Serbia.

 

Responsibilities:

  • Participate in the implementation and validation of model training pipelines on Tenstorrent compiler stack, specifically using PyTorch and JAX. This involves the process of getting neural network models to train correctly and efficiently on our hardware.
  • Support benchmarking, analysis, and performance optimization of various model architectures.
  • Contribute to efforts in scaling out distributed model training under guidance
  • Participate in discussions related to the co-design of Tenstorrent's hardware and software stack.
  • Assist in the integration of Tenstorrent software with widely used tools in the machine learning community.

Experience & Qualifications:

  • Possessing or currently pursuing a Bachelor's or Master's degree in Computer Science, Electrical Engineering, or a related field.
  • Strong understanding of neural network fundamentals.
  • Experience with model training - experience with distributed training is considered a plus.
  • Familiarity with modern neural network architectures. In-depth understanding of one architecture.
  • Proficiency in Python. Experience with C++ is a plus.
  • Strong problem-solving and analytical skills, with a demonstrated ability or potential to debug and understand model training processes in PyTorch and/or JAX.
  • Ability to learn and work effectively within a team environment.

 

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.

Key Skills
Machine LearningModel TrainingPyTorchJAXNeural NetworksPerformance OptimizationPythonC++Problem-SolvingAnalytical SkillsDistributed TrainingTeamworkSoftware IntegrationBenchmarkingArchitecture Understanding
Categories
TechnologyEngineeringData & AnalyticsSoftware
Benefits
Highly Competitive Compensation Package