Machine Learning Applications & Benchmarking Intern

Описание вакансии

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.

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.

This role is on-site based in Santa Clara, CA.

Who You Are:

• Background in computer science, ML, or a related field.
• Proficient in Python with experience using PyTorch or TensorFlow.
• Comfortable writing clean, reproducible code and debugging performance issues.
• Curious, collaborative, and always looking to learn something new.

What We Need:

• Benchmark ML models across hardware setups and analyze performance.
• Identify slowdowns and optimize for speed and efficiency.
• Build demos and apps that highlight real ML use cases on our hardware.
• Contribute to software testing and release validation.

What You Will Learn:

• Real-world ML benchmarking workflows from model to hardware.
• Performance tuning and optimization techniques.
• How ML applications are tested, validated, and productized.
• Collaboration across software, hardware, and research teams.

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.