Neural Network Optimization Engineer

Are you passionate about cutting-edge AI and eager to work on the worlds highest-performance AI accelerator chips? Do you thrive in a hands-on, collaborative environment where you can make a real impact? If so, we want to hear from you!

About the Role

As a Neural Network Optimization Engineer, you will play a crucial role in designing, developing, and optimizing state-of-the-art neural networks for customers. You'll work with a dynamic cross-functional team of engineers, acting as a bridge between customers and our software and hardware teams. Your mission? To ensure customers make the most effective use of technology by developing innovative solutions that push the boundaries of AI performance.

What You'll Do

• Optimize & Develop: Use and refine our software tools to optimize customer neural networks for peak performance.
• Analyze & Diagnose: Evaluate neural network performance at the chip, system, and application levels, identifying areas for enhancement.
• Engage & Support: Work closely with customers post-sale to ensure they maximize the value of our products, providing hands-on support and troubleshooting.
• Collaborate & Innovate: Partner with our software, hardware, and product teams to translate customer needs into product improvements and future technology innovations.
• Deliver High-Performance Models: Manage and execute projects related to Neural Network Model Garden Models, including the design and implementation of low-level C++ kernels for various neural net operations.
• Drive Optimization: Develop complex kernels and compiler strategies to meet demanding neural network requirements and communicate performance ideas to internal teams.

What Were Looking For

• Passion for AI and deep learning technologies.
• Degree in Computer Science, Engineering, or a related field.
• 5+ years of software development experience, specifically in AI.
• Strong knowledge of neural network operator algorithms (Convolutions, Transformers, RNNs).
• Experience in end-to-end software development, including data structures, algorithms, and software architecture patterns.
• Proven ability to work in cross-functional teams with a strong bias for teamwork and problem-solving.
• Hands-on experience in tuning and optimizing code and neural networks for high performance.
• Excellent communication skills and experience in program/project management.