Apocrypha

dm-haiku

Haiku is a library for building neural networks in JAX.

homepage ↗ github: deepmind/dm-haiku pypi: dm-haiku

Available in

OverlayNewestEbuildsLast activity
blshkv-overlay GitHub ↗ 0.0.12 1 27 h details ›
pypi GitHub ↗ 0.0.16 1 5 mo details ›

Versions & arches

VersionOverlay amd64arm64x86 Committed
0.0.16 pypi amd64 testing arm64 untested x86 testing — view · download · history ↗
0.0.12 ≈ blshkv-overlay amd64 testing arm64 testing x86 testing 8 mo view · download · history ↗

Use flags of 0.0.16

  • flax
  • jax
4 expansion flags (python targets, ABIs, cpu flags…)
  • python_targets_python3_11
  • python_targets_python3_12
  • python_targets_python3_13
  • python_targets_python3_14

Runtime dependencies of 0.0.16

show 25 lines
>=dev-python/absl-py-0.7.1[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]
dev-python/jmp[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]
>=dev-python/numpy-1.18.0[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]
>=dev-python/tabulate-0.8.9[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]
jax? (
dev-python/jax[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]
)
jax? (
dev-python/jaxlib[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]
)
flax? (
dev-python/flax[python_targets_python3_11(-)?,python_targets_python3_12(-)?,python_targets_python3_13(-)?,python_targets_python3_14(-)?]
)
python_targets_python3_11? (
dev-lang/python:3.11
)
python_targets_python3_12? (
dev-lang/python:3.12
)
python_targets_python3_13? (
dev-lang/python:3.13
)
python_targets_python3_14? (
dev-lang/python:3.14
)