Tenstorrent is seeking a Software Engineer to join their Product Software Engineering (PSE) team. In this role, the individual will implement customer and state-of-the-art ML models on Tenstorrent hardware, ensuring functionality, accuracy, and performance. They will also document and package the implementations for customer use. The Software Engineer will collaborate with other teams, including software engineers, system engineers, and product managers.
- Implementing ML models using Python and ML frameworks like PyTorch, JAX, ONNX, and TensorFlow.
- Benchmarking, analyzing, and optimizing model performance using C++ and Python.
- Validating the accuracy of implemented models.
- Documenting and packaging models for customer use.
- Collaborating with product managers on requirements.
Requirements:
- BSc, MSc, PhD, or final year of education in a related field.
- Experience implementing ML models on GPUs is advantageous.
- Experience with C++ and Python software development is a plus.
- Familiarity with machine learning, acceleration kernels, ML compilers, or high-performance systems is a plus.
Tenstorrent offers:
- Competitive compensation and benefits.
Apply
Tenstorrent
Tenstorrent is an AI technology company focused on revolutionizing performance, usability, and cost efficiency. They are developing a comprehensive AI platform, unifying advancements in software models, compilers, platforms, networking, and semiconductors. The company has developed a high-performance RISC-V CPU and is committed to building innovative AI solutions. Tenstorrent values collaboration, curiosity, and solving complex challenges within the high-growth, international AI landscape. They manage significant cash portfolios, navigate international banking, and prepare for potential IPO readiness, demonstrating maturity as a company.