From 2eb172dab1eb8ccc9c5914b870780f2ee0ac507f Mon Sep 17 00:00:00 2001 From: Kartik Pandit <94665227+kartikpandit712@users.noreply.github.com> Date: Mon, 2 Jun 2025 13:50:39 -0400 Subject: [PATCH] Update README.md --- README.md | 23 +++++++++++++++++++++-- 1 file changed, 21 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 3bfd7a6..bbf5221 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,21 @@ -# TPU_Gen -This repo contains the framework to generate the TPU and their approximate versions, leveraging the Large Language Models. +# TPU-Gen: LLM-Driven Custom Tensor Processing Unit Generator + +📌 **Accepted at ICLAD 2025** + +This repository hosts the official implementation, dataset, and experimental results for our paper: + +**"TPU-Gen: LLM-Driven Custom Tensor Processing Unit Generator"** +Authored by Kartik Pandit, Deepak Vungarala, Mohammed Elbtity, Shaahin Angizi, Ramtin Zand, and collaborators. + +--- + +## 🧠 TPU-Gen Framework Overview + +TPU-Gen is the first end-to-end framework that uses fine-tuned LLMs with RAG (Retrieval-Augmented Generation) to generate both **exact and approximate TPU architectures**. + +It consists of: + +- ✅ Prompt generation from user input +- ✅ LLMs enhanced with multi-shot learning +- ✅ RAG module for dependency fetching +- ✅ Automated design validation and PPA reporting