From bf3adcf1b93cf3b5e49c5056f401593ec14c22a6 Mon Sep 17 00:00:00 2001 From: "Yilin (Eleen) Bao" <62961093+yilin-bao@users.noreply.github.com> Date: Sat, 9 Dec 2023 03:08:22 -0600 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 2bcb03e..e8ba25f 100644 --- a/README.md +++ b/README.md @@ -3,6 +3,8 @@ [![Build Status](https://travis-ci.org/user/repo.svg?branch=master)](https://travis-ci.org/user/repo) [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT) +Logo + ## Overview This library is designed to generate visualizations of neural networks by analyzing PyTorch code. Unlike conventional methods, our approach eliminates the need for large parameter files (pth) or specific input formats. With minimal AI knowledge, users can leverage this tool to create visualizations that aid in understanding complex networks, including support for transformers and attention mechanisms.