diff options
Diffstat (limited to 'pkgs/development/python-modules/optax/default.nix')
-rw-r--r-- | pkgs/development/python-modules/optax/default.nix | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/optax/default.nix b/pkgs/development/python-modules/optax/default.nix index bf0383fa15309..6a3b6a9d3e67c 100644 --- a/pkgs/development/python-modules/optax/default.nix +++ b/pkgs/development/python-modules/optax/default.nix @@ -22,10 +22,11 @@ buildPythonPackage rec { sha256 = "1q8cxc42a5xais2ll1l238cnn3l7w28savhgiz0lg01ilz2ysbli"; }; + buildInputs = [ jaxlib ]; + propagatedBuildInputs = [ absl-py chex - jaxlib numpy ]; |