diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix
index 3373c3d3d3b2..9f701493f07a 100644
--- a/pkgs/development/python-modules/torch/default.nix
+++ b/pkgs/development/python-modules/torch/default.nix
@@ -12,7 +12,10 @@
   Accelerate, CoreServices, libobjc,
 
   # Propagated build inputs
+  filelock,
   sympy,
+  networkx,
+  jinja2,
   numpy, pyyaml, cffi, click, typing-extensions,
 
   # Unit tests
@@ -285,8 +288,14 @@ in buildPythonPackage rec {
     click
     numpy
     pyyaml
+
+    # From install_requires:
+    filelock
     typing-extensions
     sympy
+    networkx
+    jinja2
+
     # the following are required for tensorboard support
     pillow six future tensorboard protobuf
   ] ++ lib.optionals MPISupport [ mpi ]