Intern, AI/ML Compiler Research Engineer

Advancing the World’s Technology Together

Our technology solutions power the tools you use every day--including smartphones, electric vehicles, hyperscale data centers, IoT devices, and so much more. Here, you’ll have an opportunity to be part of a global leader whose innovative designs are pushing the boundaries of what’s possible and powering the future.

We believe innovation and growth are driven by an inclusive culture and a diverse workforce. We’re dedicated to empowering people to be their true selves. Together, we’re building a better tomorrow for our employees, customers, partners, and communities.

What You’ll Learn

• Project Overview: Contribute to the design and implementation of new DSLs, compiler passes and optimizations to improve performance, efficiency, and resource utilization for executing LLM workloads on next-generation accelerators.
• Skills You’ll Learn:
• Co-design of Memory-centric AI accelerator architectures and platforms
• Optimizing the execution of state-of-the-art Generative AI models on specialized hardware
• Hardware-software co-design and co-optimization

What You’ll Do

The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads while minimizing energy consumption and maximizing performance. To achieve this goal, we collaborate closely with both hardware and software engineers to identify and address the unique challenges posed by AI/ML workloads and to explore new computing abstractions that can provide a better balance between the hardware and software components of our systems. Additionally, we continuously conduct research and development in emerging technologies and trends across memory, computing, interconnect, and AI/ML, ensuring that our platforms are always equipped to handle the most demanding workloads of the future.

The ideal candidate for this role would share our passion for creating and increasing the value of new technologies and products, and thrive in a highly dynamic, fast-paced, results-driven environment. We are looking for someone to work with a group of highly talented, passionate, and versatile engineers who share our vision and contribute to the creation of next generation memory-centric AI solutions/products. Join us in our passion to shape the future of computing!

Location: Daily onsite presence at our San Jose, CA headquarters in alignment with our Flexible Work policy

Reports to: Director, SW Dev. Infrastructure Engineering

• Research, develop and evaluate novel compiler-based solutions for efficiently executing emerging Generative AI models on next-generation memory-centric accelerator architectures.
• Profile computational kernels written using different DSLs and analyze/debug functional and performance bottlenecks on target hardware platforms.
• Work with cutting-edge technologies and talented professionals to prepare a novel presentation for our annual Intern Exhibition, draft novel patents and scientific publication to be submitted to top-tier conferences.
• Complete other responsibilities as assigned.

What You Bring

• Pursuing a PhD in Computer Science, Computer Engineering or related field, with focus on AI/ML Compiler technologies and deep learning models.
• Must have at least 1 academic quarter/semester remaining.
• Experience with AI/ML compiler optimizations like tiling, vectorization, parallelization, and quantization.
• Demonstrated knowledge of computer architecture, with a focus on GPUs or AI hardware accelerators and state-of-the-art domain-specific parallel programming languages.
• Prior first author publications in top-tier AI/ML conferences.
• Prior contributions to open-source projects in the AI/ML compiler space such as LLVM, MLIR, Triton, PyTorch Inductor, etc., is a big plus.
• Must be highly motivated with excellent verbal and written communication skills, and ability to thrive in a collaborative, multi-disciplinary environment.
• You’re inclusive, adapting your style to the situation and diverse global norms of our people.
• An avid learner, you approach challenges with curiosity and resilience, seeking data to help build understanding.
• You’re collaborative, building relationships, humbly offering support and openly welcoming approaches.
• Innovative and creative, you proactively explore new ideas and adapt quickly to change.

#LI-AD1

What We Offer

The pay range below is for all roles at this level across all US locations and functions. Individual pay rates depend on a number of factors—including the role’s function and location, as well as the individual’s knowledge, skills, experience, education, and training.

This is in addition to our diverse package of benefits centered around the wellbeing of our employees and their loved ones. In addition to the usual Medical/Dental/Vision, our inclusive rewards plan empowers our people to care for their whole selves. An investment in your future is an investment in ours.

Relocation Support Live 50+ miles away from your new office? No worries! We'll provide a stipend to help support your move.

Housing Stipend Signing a lease on a new place for your summer internship? We'll help you pay your way with a monthly stipend.

Give Back With a charitable giving match and frequent opportunities to get involved, we take an active role in supporting the community.

Prioritize Emotional Wellness With on-demand apps and paid therapy sessions, you’ll have support no matter where you are.

Stay Fit Eating well and being active are important parts of a healthy life. Our onsite Café and gym, plus virtual classes, make it easier.

Embrace Flexibility Benefits are best when you have the space to use them. That’s why we facilitate a flexible environment so you can find the right balance for you.