Software Engineering Intern - Kernels

Develop high performance kernels for Machine Learning models.

• •As a Software Engineering Intern within our Kernels team, you will play a key role in developing high performance kernels essential for accelerating Machine Learning models.
• •Key Responsibilities Implement high performance kernels in low-level languages (Assembly/ISA experience a plus) Develop, test, and tune kernels for machine learning models and performance Create and automate reference implementations and unit tests Analyze scalability and performance, collect metrics, and troubleshoot bottlenecks Package and share implementations with partner teams Requirements Ability to implement high performance kernels in low-level languages; Assembly/ISA coding experience is advantageous Proficiency in Python and/or C++ Solid background in Machine Learning model architecture (e.g., LLMs, CNNs) Experience with ML frameworks such as PyTorch and ML packages like Numpy General understanding of computer architecture (CPU, GPU, custom ASICs, etc.) Currently enrolled in a graduate program (Master's or Ph.D) in a relevant discipline