Research Intern (Ph.D.): LLMs and Chip Design at ChipStack

About the job:

We are looking for an exceptional Ph.D. Research Interns to join our founding team and contribute to improving our in-house large language models (LLMs). In this role, you will apply cutting-edge techniques to fine-tune LLMs and implement novel ways to improve its performance on our target applications.

This is a unique opportunity to work at the forefront of LLMs while surrounded by exceptional researchers and builders at ChipStack. As a founding member of a rapidly growing startup, you will significantly impact the product and shape its future.

Requirements:

• Pursuing a Ph.D. in CS, with a focus around ML or chip design (computer architecture and related areas)
• Strong programming skills in Python, preferably in industry
• Prior exposure to fine-tuning large language models
• Prior exposure to hardware description languages like Verilog/SystemVerilog (expertise is not necessary)
• Exposure to or active participation in the latest research and advancements in AI

Responsibilities:

• Fine-tune and optimize large language models for our target applications in chip design
• Develop benchmarks and metrics to assess the quality of outputs
• Collaborate and guide data collection
• Ship features that use the task you fine-tuned on

Why ChipStack?

• Exceptional Learning Environment: Our founding team brings decades of deep expertise in AI and chip design, with experience at top tech companies like Nvidia, Google, Meta, Qualcomm, and the Allen Institute for AI. Working alongside these industry leaders offers unparalleled growth and insight.
• Startup Experience: ChipStack is backed by premier investors, including Khosla Ventures, and collaborates with prominent chip designers. Here, you'll enjoy the excitement and dynamic pace of a high-potential startup.
• GenAI Exposure: Be right in the middle of the excitement in the industry around gen AI.
• Meaningful Impact: Chips are at the center of today’s tech-driven world. Chip design as an industry is ripe for disruption with generative AI. Joining ChipStack means contributing to impactful solutions that shape the future.
• Career Advancement: An internship at ChipStack is a valuable opportunity that can lead to a full-time role, providing a clear path to long-term career growth.