Machine Learning Engineer AI Models

Tenstorrent is an AI-computing company that sells AI hardware and licenses AI and RISC-V intellectual property.

Toronto, Canada
About Tenstorrent

Tenstorrent builds computers for AI, including AI processors, scalable server systems, and open-source software and compiler tooling. It also licenses AI and RISC-V IP for customers building customized silicon.

View jobs by Tenstorrent

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will port, tune, validate, and debug LLM and vision workloads on a custom AI accelerator. You will analyze model behavior and bottlenecks across the stack, improve execution efficiency and correctness, and translate advanced model architectures into reliable production deployments.

Requirements

  • Experience building and working with machine learning models in PyTorch or TensorFlow
  • Understanding of modern machine learning model architectures such as transformers
  • Software engineering fundamentals
  • Debugging skills
  • Problem-solving skills

Responsibilities

  • Bring up, run, and debug modern machine learning models using PyTorch or TensorFlow
  • Analyze model behavior and performance to identify bottlenecks
  • Improve the efficiency, correctness, and scalability of model execution
  • Collaborate with compiler, kernel, and hardware specialists on system-level improvements
  • Translate advanced model architectures into high-performance production deployments
Machine Learning Engineer AI Models at Tenstorrent | JobStash