python-dm-haiku 0.0.13
Propagated dependencies: python-absl-py@1.4.0 python-chex@0.1.88 python-cloudpickle@3.1.0 python-dill@0.3.9 python-dm-tree@0.1.8 python-flax@0.8.0 python-jax@0.4.28 python-jaxlib@0.4.28 python-jmp@0.0.4 python-numpy@1.26.2 python-optax@0.1.5 python-tabulate@0.9.0 python-tensorflow@2.13.1 python-virtualenv@20.29.1
Channel: guix-science
Home page: https://github.com/google-deepmind/dm-haiku
Licenses: ASL 2.0
Synopsis: Sonnet for JAX
Description:
Haiku is a simple neural network library for JAX. It is developed by some of the authors of Sonnet, a neural network library for TensorFlow.
Total results: 1