📝

Signed-off-by: peter szemraj <peterszemraj@gmail.com>

Author: pszemraj

README.md

@@ -4,6 +4,28 @@
 
 This aims to be a simpler implementation of the [original repo](https://github.com/microsoft/Samba).
 
+## repo structure
+
+```text
+samba-pytorch/
+├── pyproject.toml
+├── README.md
+├── samba_pytorch/
+│   ├── __init__.py
+│   ├── config.py
+│   ├── modules/
+│   │   ├── __init__.py
+│   │   ├── fused_rotary_embedding.py
+│   │   ├── gla.py
+│   │   ├── mamba_simple.py
+│   │   ├── multiscale_retention.py
+│   │   ├── rmsnorm.py
+│   │   └── rotary.py
+│   ├── samba.py
+│   ├── tokenizer.py
+│   └── utils.py
+```
+
 ## Citations
 
 ```bibtex