Machine Learning Engineer Intern - PEY
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 are seeking a passionate and driven Machine Learning Engineer Intern to assist in the development and optimization of machine learning algorithms and models on Tenstorrent devices. This internship will provide you with hands-on experience working with industry-leading technologies and will involve collaboration with cross-functional teams.
This role is on-site, based out of Toronto.
Responsibilities:
- Understand how models map onto the Tenstorrent devices through compilation steps and kernels, investigate any gaps in functionality and performance, propose innovative solutions to mitigate any issues.
- Implement or improve kernels for computation and data movement.
- Conduct experiments to evaluate the performance of different approaches on various hardware platforms.
- Increase model and software stack robustness by devising benchmarking experiments and CI/CD pipelines.
- Stay current with the latest research and developments in the field of machine learning and AI, applying relevant advancements to our projects.
Experience & Qualifications:
- Currently pursuing a degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
- Strong understanding of machine learning concepts (especially LLMs and CNNs) and frameworks (e.g., TensorFlow, PyTorch).
- Experience with programming languages such as Python, C++, and CUDA.
- Familiarity with data processing and analysis tools (e.g., NumPy, Pandas).
- Knowledge of hardware architectures and performance optimization is a plus.
- Strong analytical and problem-solving skills, with attention to detail.
- Excellent communication skills and ability to work collaboratively in a 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.
Apply for this job
*
indicates a required field