5 packages found. Page 1 of 1.

Name Version Votes Popularity? Description Maintainer Last Updated
python-chex 0.1.86-2 1 0.00 A library of utilities for helping to write reliable JAX code daskol 2024-06-26 21:10 (UTC)
python-dm-haiku 0.0.12-1 1 0.00 A library for building neural networks in JAX daskol 2024-02-29 10:50 (UTC)
python-jmp 0.0.4-3 1 0.00 A mixed precision library for JAX daskol 2023-06-15 11:40 (UTC)
python-optax 0.2.2-2 2 0.32 A gradient processing and optimization library for JAX daskol 2024-06-26 21:11 (UTC)
python-pycolab-git r21.1d808c2-1 0 0.00 DeepMind pycolab sajaysurya 2019-01-17 11:49 (UTC)

5 packages found. Page 1 of 1.