🛠 Tech Stack

💼 About This Role

You'll take the lead in bringing up and optimizing cutting-edge AI models on custom AI devices. You'll experiment, optimize, and push boundaries while solving real-world problems. Join a team passionate about AI model performance from application to silicon.

🎯 What You'll Do

• Bring up and optimize state-of-the-art ML models on new hardware platforms.
• Debug performance issues and tune model architectures for accuracy.
• Apply model optimization techniques like quantization and kernel fusing.
• Collaborate with compiler, kernel, and hardware teams to drive improvements.

📋 Requirements

• Python programming proficiency.
• Hands-on experience with PyTorch for deep learning.
• Deep understanding of ML model architectures and optimization.
• Experience leading a small team and working cross-functionally.

✨ Nice to Have

• Experience with model optimization techniques such as quantization and flash attention.
• Familiarity with multi-device parallelization.
• Staying current with ML research.

🎁 Benefits & Perks

• 💰 Competitive compensation package including base and variable targets.
• 🏥 Comprehensive benefits offered.
• 🏢 Hybrid work with remote optional.
• ⚖️ Equal opportunity employer.

📨 Hiring Process

Estimated timeline: 2-4 weeks · AI estimate

1. 1Recruiter Call· 30 min
2. 2Technical Screen· 60 min
3. 3On-site Interviews· 4 hours

🚩 Heads Up

• Salary range very wide ($100k-$500k) indicating lack of clarity.
• U.S. export control restrictions may limit eligibility.