Browse All Jobs

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

Machine Learning Intern for AI Compiler Model Training in Serbia.
Tenstorrent University Jobs
On-Site
Serbia
Undisclosed salary
Job Description
Tenstorrent is seeking a Machine Learning Intern to join their AI Compiler team in Belgrade, Serbia. This internship offers the opportunity to work on a real engineering project for 3+ months, with guidance from a dedicated mentor and manager. The intern will contribute to the implementation and validation of model training pipelines on Tenstorrent's compiler stack, using PyTorch and JAX. They will also support benchmarking, analysis, and performance optimization of various model architectures, as well as contribute to efforts in scaling out distributed model training. The intern will participate in discussions related to the co-design of Tenstorrent's hardware and software stack and assist in the integration of Tenstorrent software with widely used tools in the machine learning community.

Role Involves:
  • Participating in the implementation and validation of model training pipelines on Tenstorrent compiler stack, specifically using PyTorch and JAX.
  • Supporting benchmarking, analysis, and performance optimization of various model architectures.
  • Contributing to efforts in scaling out distributed model training under guidance.
  • Participating in discussions related to the co-design of Tenstorrent's hardware and software stack.
  • Assisting in the integration of Tenstorrent software with widely used tools in the machine learning community.

Requirements:
  • 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.
Apply Manually

Tenstorrent University Jobs

Tenstorrent University Jobs focuses on advancing AI technology through the co-design of hardware and software solutions. Their team develops high-performance RISC-V CPUs and optimizes machine learning applications. They foster collaboration and innovation to address complex challenges in software models, compilers, platforms, networking, and semiconductors. With focus on machine learning, compilers, parallel programming, and computer architecture, Tenstorrent is committed to building advanced AI platforms. The company is dedicated to pushing the boundaries of AI and high-performance computing through continuous innovation and development.

All Jobs at Tenstorrent University Jobs (10)