Sr. Machine Learning Engineer, ML Models at Tenstorrent | Ontario, CA | Rezi

Sr. Machine Learning Engineer, ML Models at Tenstorrent

Sr. Machine Learning Engineer, ML Models

Tenstorrent · Ontario, CA

2 months ago

Sr. Machine Learning Engineer, ML Models

Tenstorrent · Ontario, CA

2 months ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

As a Senior Machine Learning Engineer on the ML Models team, you will lead the bring-up and optimization of cutting-edge AI models for custom AI devices. You will experiment, optimize, and push boundaries while solving real-world problems, working on models used in practical applications.

Responsibilities

  • Bring up and optimize cutting edge AI models on custom AI devices.
  • Experiment with and optimize AI models.
  • Push boundaries in AI model development.
  • Solve real-world problems with AI models.
  • Debug performance issues.
  • Tune architectures for improved performance.
  • Boost model accuracy and robustness.
  • Work alongside compiler, kernel, and hardware teams to drive new features, performance optimizations, and fixes.

Requirements

  • Hands-on experience bringing up state-of-the-art ML models on new hardware platforms.
  • Strong debugging instincts.
  • A curiosity-driven mindset that stays current with ML research and brings practical insights to real-world engineering challenges.
  • Eligibility to access U.S. export-controlled technology, potentially contingent on citizenship/permanent residency status or ability to obtain prior license approval from the U.S. Commerce Department or applicable federal agency.

Skills

  • Python programming
  • PyTorch
  • Deep learning models
  • ML model architectures
  • Model optimization techniques (quantization, flash attention, kernel fusing, multi-device parallelization)
  • Leading small groups of engineers
  • Cross-functional team collaboration

Location

  • Toronto, ON
  • Remote (on a candidate-by-candidate basis)

Work Type

  • Hybrid
  • Remote

Experience Level

  • Senior
  • All seniorities

Salary/Compensations

  • $100k - $500k (including base and variable compensation targets)

Benefits

  • Highly competitive compensation package

About the Company

  • Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency.
  • Solutions must evolve to unify innovations in software models, compilers, platforms, networking, and semiconductors.
  • The diverse team has developed a high performance RISC-V CPU from scratch.
  • The team shares a passion for AI and a deep desire to build the best AI platform possible.
  • Values collaboration, curiosity, and a commitment to solving hard problems.

Equal Opportunity

  • Equal opportunity employer.