Tenstorrent - Software Engineer - ML Frameworks (5-7 yrs)

Job Description :

Responsibilities :

- Design, develop, and implement high-performance compilers for machine learning frameworks (e., TensorFlow, PyTorch) targeting specific semiconductor architectures.

- Optimize compiler performance by exploring and implementing advanced compiler techniques such as loop transformations, instruction scheduling, and register allocation.

- Develop and maintain compiler infrastructure, including front-ends, intermediate representations, and back-ends.

- Collaborate with hardware engineers to optimize compiler performance for specific hardware features and architectures.

- Analyze and profile application performance to identify and address compiler bottlenecks.

- Conduct research and development on cutting-edge compiler technologies, such as machine learning-driven compilation and domain-specific languages.

- Contribute to the open-source community by contributing to relevant compiler projects.

- Stay abreast of the latest advancements in compiler technology and semiconductor architecture.

Required Skills :

- Strong foundation in compiler design principles : Experience with compiler internals, including lexical analysis, parsing, intermediate representations, code generation, and optimization techniques.

- Proficiency in C/C++ : Excellent programming skills in C/C++ with a strong understanding of data structures and algorithms.

- Experience with at least one machine learning framework : Familiarity with TensorFlow, PyTorch, or other popular ML frameworks.

- Knowledge of computer architecture : Understanding of CPU and GPU architectures, including instruction sets, memory hierarchies, and performance bottlenecks.

- Experience with scripting languages : Proficiency in Python or other scripting languages for automation and analysis.

- Excellent problem-solving and debugging skills.

- Strong communication and collaboration skills.

Nice to Have :

- Experience with LLVM or other compiler infrastructure projects.

- Experience with machine learning techniques for compiler optimization.

- Experience with high-performance computing (HPC) systems.

- Experience with domain-specific languages (DSLs) for hardware description or high-level synthesis.

- Experience with cloud computing platforms (AWS, Azure, GCP)