diff options
Diffstat (limited to 'pkgs/development/python-modules/dm-haiku/default.nix')
-rw-r--r-- | pkgs/development/python-modules/dm-haiku/default.nix | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/dm-haiku/default.nix b/pkgs/development/python-modules/dm-haiku/default.nix index 5468776c72d11..14ceebf0a19b3 100644 --- a/pkgs/development/python-modules/dm-haiku/default.nix +++ b/pkgs/development/python-modules/dm-haiku/default.nix @@ -4,6 +4,7 @@ , dill , dm-tree , fetchFromGitHub +, jaxlib , jmp , lib , pytestCheckHook @@ -31,6 +32,7 @@ buildPythonPackage rec { chex cloudpickle dm-tree + jaxlib pytestCheckHook tensorflow ]; |