Lead Engineer, ML Models

Key Skills

Aartificial intelligence, Leadership Skill, Machine learning techniques, Optimization, Python Programming, PyTorch

Job Description

We welcome candidates at various experience levels for this role. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ from the one in this posting.

Who You Are

• Confident with Python programming and hands-on experience with PyTorch for developing deep learning models.
• Possess a deep understanding of ML model architectures, with the ability to optimize both individual components and overall model performance.
• Skilled at mentoring engineers, providing technical guidance, and fostering a collaborative team culture.
• Comfortable leading cross-functional discussions, setting priorities, and aligning efforts toward impactful results.

What We Need

• Lead the bring-up of state-of-the-art ML models on new hardware platforms.
• Direct debugging and performance tuning efforts to improve accuracy, efficiency, and robustness.
• Oversee model optimization techniques, like quantization, flash attention, kernel fusing, and multi-device parallelization.
• Champion a curiosity-driven approach by staying ahead of ML research trends and guiding the team in applying relevant advances to solve real-world engineering challenges.

What You Will Learn

• How to lead the deployment of real ML models at scale on a custom AI accelerator.
• Methods to coordinate model optimization from application to silicon level.
• Approaches for bridging research innovations with production-ready engineering.
• How to influence compiler, kernel, and hardware roadmaps through model performance insights.

Tenstorrent offers a highly competitive compensation package and benefits, and we are an equal opportunity employer.