5 packages found. Page 1 of 1.

Name Version Votes Popularity? Description Maintainer Last Updated
python-optax 0.2.4-1 3 0.49 A gradient processing and optimization library for JAX daskol 2024-11-13 17:55 (UTC)
python-chex 0.1.88-1 2 0.49 A library of utilities for helping to write reliable JAX code daskol 2024-12-11 15:43 (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.13-1 1 0.00 A library for building neural networks in JAX daskol 2024-10-16 13:02 (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.