diff --git a/requirements.txt b/requirements.txt
index a24e106..1b25727 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -22,7 +22,7 @@ rich            # beautiful text formatting in terminal
 pytest          # tests
 # sh            # for running bash commands in some tests (linux/macos only)
 
-torchdyn==1.0.3
+torchdyn==1.0.6
 zuko
 jetnet
 energyflow