ML Compiler Intern (AI2425)

Job Title: ML Compiler Intern    Job Location: San Jose, CA    Job ID: AI2425    Internship Term: Aug 2025 through Dec 2025    Internship Stipend: The hourly pay rate range for this position is for MS: $44 - $46 and for PhD: $53 - $57.   ONSITE: This role requires the Intern to work from our San Jose, California office during the duration of the Internship.     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.