5 packages found. Page 1 of 1.

Name Version Votes Popularity? Description Maintainer Last Updated
python-optax 0.2.2-1 2 0.87 A gradient processing and optimization library for JAX daskol 2024-03-28 12:57 (UTC)
python-chex 0.1.86-1 1 0.00 A library of utilities for helping to write reliable JAX code daskol 2024-04-17 12:38 (UTC)
python-jmp 0.0.4-3 1 0.00 A mixed precision library for JAX daskol 2023-06-15 11:40 (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-pycolab-git r21.1d808c2-1 0 0.00 DeepMind pycolab sajaysurya 2019-01-17 11:49 (UTC)

5 packages found. Page 1 of 1.