Software Engineer, ML Frameworks

Tenstorrent

Santa Clara, CA

Full Time

Mid Level

27 days ago

Job Description

About the Role

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. The role involves designing, developing, and optimizing compilers for semiconductor products, working closely with hardware and software engineers, and staying current with research in ML models and compiler technology.

Key Responsibilities

Participate in the co-design of Tenstorrent's hardware and software stack
Benchmark, analyze, and optimize performance of key machine learning applications across Tenstorrent's hardware and software stack
Collaborate with cross-functional teams to develop and implement compiler features, including optimization algorithms, code generation, and language front-ends
Work with customers and stakeholders to understand their needs and requirements and deliver high-quality solutions that meet those needs
Stay up-to-date with the latest research and trends in ML models and compiler technology, and incorporate these insights into our products

Requirements

BSc, MSc or PhD in Electrical/Computer Engineering or Computer Science
Good experience with algorithms, data structures, and software development in C/C++
Python expertise is welcome
Experience with machine learning frameworks such as TensorFlow and PyTorch
Familiarity with and passion for machine learning, compilers, parallel programming, high-performance and massively parallel systems, processor and computer architecture is a plus

Nice to Have

Experience with machine learning frameworks such as TensorFlow and PyTorch
Familiarity with machine learning, compilers, parallel programming, high-performance and massively parallel systems, processor and computer architecture