README update

smsharma · Mar 16, 2024 · 8217dc7 · 8217dc7
1 parent 8b23e4f
commit 8217dc7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -34,6 +34,7 @@ mamba env create --file environment.yaml
 
 ## Code overview
 
+The code primarily uses Jax. The main components are:
 
 - The script for downloading the data is [download_data.py](download_data.py), the summarization script is [summarize.py](summarize.py), and training script is [train.py](train.py).
 - [notebooks/01_create_dataset.ipynb](notebooks/01_create_dataset.ipynb) is used to create the `tfrecords` data used for training.