Top Engineering Software Machine Learning Jobs

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're looking for ML Researchers/Engineers to help us make LLMs go fast on Tenstorrent’s leading edge AI platform. If you’re enthusiastic about large scale deep learning models, are a competent software/ML engineer or researcher, and enjoy working on challenging problems, this is your opportunity to be at the bleeding edge of AI processing.

Responsibilities

• Develop novel LLMs and primitives that take advantage of Tenstorrent’s breakthrough accelerator architecture to deliver orders of magnitude performance & efficiency improvements.
• Model benchmarking, Performance Analysis, Characterization on novel hardware accelerator.
• Fine tune/Retrain emerging LLM models after architectural modification to improve performance.
• Implement ideas from latest research papers and validate new optimizations on novel hardware, suggest improvements on the existing ideas and validate new ideas through experiments.
• Implement ML models on Tenstorrent’s PyBUDA Compiler framework and analyze results, suggest novel architectural changes to make them more efficient.

• Deep understanding of Transformer based model architecture (e.g., BERT, GPT, LLaMA, FALCON) - should be able to analyse performance bottlenecks in training or inference.
• Expertise (through evidence in prior projects or research) in large-scale foundation model (CNN/ Transformer/Generative) training and optimisations including quantisation, pruning, low-rank approximation (e.g., LoRA, SVD), sparsity aware training, early exit models, conditional computing, sparse attentions, NAS, fine-tuning (e.g., RLHF, PEFT)
• Excellent working knowledge of Python, PyTorch or TensorFlow, (optional C/C++ if interested in compiler and software stack)
• Familiarity with and passion for any of the following -- ML compilers, high-performance and massively parallel systems, distributed model training and inference, GPU or AI accelerator architecture -- is a plus.

Locations:

Boston (US), Toronto (CAN), Cambridge (UK)

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 sanctioned 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 information and/or documentation will be required and considered as Tenstorrent moves through the employment process.

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 looking for an experienced Machine Learning engineer to support our growing customer base as they build Deep Learning models on Tenstorrent hardware.  If you're enthusiastic about Machine Learning, are a competent software engineer, and enjoy working with other people, this is your opportunity to be at the bleeding edge of AI processing.  You'll get exposure to a broad array of problem types from different industries and be at the forefront of our customer engagements.

Responsibilities

• Designing and developing demonstration machine learning and deep learning systems
• Model benchmarking
• Running machine learning tests and experiments on behalf of customers
• Implementing appropriate ML algorithms
• Select appropriate datasets and data representation methods
• Run machine learning tests and experiments
• Perform statistical analysis and fine-tuning using test results
• Train and retrain systems when necessary
• Extend existing ML libraries and frameworks
• Develop novel ML models and primitives that take advantage of Tenstorrent’s breakthrough architecture to deliver orders of magnitude performance & efficiency improvements

• Degree in Electrical/Computer Engineering, Computer Science, Machine Intelligence, Engineering Science, or Math;
• 5 years of experience as a Machine Learning Engineer or similar role (a fleshed out GitHub repo a plus)
• Experience with algorithms, data structures, and software development in Python and C/C++.
• Deep knowledge of math, probability, statistics and algorithms
• Experience in solving business problems with Machine Learning models
• Familiarity with and passion for any of the following -- machine learning, compilers, parallel programming, high-performance and massively parallel systems, processor and computer architecture -- is a plus

Location:

We have presence in Toronto, Boston, Austin, Santa Clara, Portland, and Raleigh. We are open to remote candidates on a case-by-case basis.

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 sanctioned 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 information and/or documentation will be required and considered as Tenstorrent moves through the employment process.