Location: United States
Company: d-Matrix
- 12 weeks: June 1st - August 21st or June 22nd - September 11th - 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 - Your responsibilities will span developing reference implementations for accuracy verification, defining unit tests for implemented operators, performance tuning, scalability analysis across varied problem sizes, and packaging/shipping the final implementations - You will also collect performance metrics and identify bottlenecks to improve core functionality - 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 ### Benefits - Equity - Health care - Flexible time-off - Paid paternity leave - 401k retirement plan - Remote & hybrid working model - Work with world-leading engineers and researchers - Team-building activities - Peer coaching & development - Free food and snacks - Happy hours - On-site fitness center- General understanding of computer architecture (CPU, GPU, custom ASICs, etc.) - Ability to implement high performance kernels in low-level languages; Assembly/ISA coding experience is advantageous - Currently enrolled in a graduate program (Master’s or Ph.D) in a relevant discipline - Experience with ML frameworks such as PyTorch and ML packages like Numpy - Solid background in Machine Learning model architecture (e.g., LLMs, CNNs) - Proficiency in Python and/or C++ - Interest in hardware-software co-design - Familiarity with additional ML frameworks (TensorFlow, etc.) - Previous internship or project experience related to high performance computing or ML kernel development
This is a visa-sponsored position. Apply at avisajob.com/jobs/1779845098367