diff --git a/Project.toml b/Project.toml
index 3c39568..8b1b41a 100644
--- a/Project.toml
+++ b/Project.toml
@@ -18,6 +18,7 @@ TensorOperations = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
+JLD2 = "0.4"
 Zygote = "0.6"
 julia = "1.6.7"