Machine Learning Engineer, AI Compiler - Tenstorrent

Tenstorrent is seeking a talented and motivated Machine Learning Engineer to join our team in Belgrade, Serbia. As a Machine Learning Engineer, you will play a crucial role in developing and optimizing our model training pipeline within our machine learning graph compiler.

This role is hybrid, based out of Belgrade, Serbia.

Responsibilities:

• Develop and optimize the model training pipeline in our machine learning graph compiler.
• Benchmark, analyze, and optimize the performance of training multiple state-of-the-art model architectures across Tenstorrent's hardware and software stack.
• Scale out to various modes of distributed model training.
• Participate in the co-design of Tenstorrent's hardware and software stack.
• Develop performance analysis and estimation infrastructure that feeds into Tenstorrent compiler.
• Integrate Tenstorrent software into leading machine learning frameworks.

Experience & Qualifications:

• 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 programming languages such as Python and C++.
• Excellent problem-solving and analytical skills.
• Ability to work effectively in a collaborative team environment.

Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.

Due to U.S. Export Control laws and regulations, Tenstorrent is required to ensure compliance with licensing regulations when transferring technology to nationals of certain countries that have been licensing conditions set  by the U.S. government.

As this position will have direct and/or indirect access to information, systems, or technologies that are subject to U.S. Export Control laws and regulations, please note that citizenship/permanent residency, asylee and refugee information and supporting documentation will be required and considered as a condition of employment.

If a U.S. export license is required, employment will not begin until a license with acceptable conditions is granted by the U.S. government.  If a U.S. export license with acceptable conditions is not granted by the U.S. government, then the offer of employment will be rescinded.