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