ML Compiler Intern (AI2424)

Roles and Responsibilities:  

 As a member of the MLA Software team, you will have the opportunity to make significant contributions to SiMa.ai’s state-of-the-art Machine Learning Accelerator (MLA) compiler.  This compiler framework supports hundreds of Convolutional Neural Networks (CNNs), as well as a wide variety of Large Language Models (LLMs) and Large Multimodal Models (LMMs).  Projects that you may work on include, but are not limited to, the following:

• Implement compiler support for new CNNs, LLMs, and LMMs.
• Implement algorithms within the compiler framework that improve accuracy and run-time performance for a wide variety of ML models.
• Identify and implement optimizations to speed up compilation time.

Requirements (Experience and Expertise Areas):  

• Candidates are expected to have recently graduated, or nearing graduation, with an M.S. or Ph.D. in Computer Science or a related field.
• Strong programming skills in Python, C, or C++.
• Strong background in algorithms and data structures.
• Familiarity with PyTorch and ONNX deep learning frameworks.
• Solid understanding of machine learning algorithms (e.g., matrix multiplication and convolution).
• Solid understanding of Generative AI, including LLM architecture.
• Solid understanding of processor architecture.
• Can-do attitude.
• Curious, creative, and good at solving problems.
• Execution and results oriented.
• Self-driven.
• Good communication skills.

EEO Employer: SiMa is an equal opportunity employer; all qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or any other protected classification.