Update README.md

martius-lab · Jul 18, 2024 · 6ce4b56 · 6ce4b56
1 parent aac932e
commit 6ce4b56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -59,7 +59,7 @@ pip3 install torch torchvision torchaudio
   <img src="/assets/docs/gso_vs_svd3.gif" width="400" />
 </p>
 
-All experiments are implemented in PyTorch. 
+All experiments are implemented in PyTorch. A few scripts used to create figures use JAX.
 Most experiments simply use [train.py](scripts/train.py).
 Depending on which command line arguments are passed (see below), 
 [train.py](scripts/train.py) runs different neural network regression tasks via [hydra](https://hydra.cc/).