13 packages found. Page 1 of 1.

Name Version Votes Popularity? Description Maintainer Last Updated
python-jax 0.4.30-1 7 1.06 Composable transformations of Python+NumPy programs daskol 2024-06-19 07:06 (UTC)
python-jaxlib-cuda 0.4.30-1 5 0.36 XLA library for JAX daskol 2024-06-19 07:37 (UTC)
python-jaxlib 0.4.30-1 6 0.21 XLA library for JAX daskol 2024-06-19 07:50 (UTC)
python-optax 0.2.3-1 2 0.18 A gradient processing and optimization library for JAX daskol 2024-07-10 20:32 (UTC)
python-flax 0.8.5-1 2 0.18 A neural network library and ecosystem for JAX designed for flexibility wjhandley 2024-07-05 08:50 (UTC)
python-orbax 0.1.7-1 1 0.00 Orbax is a library providing common utilities for JAX users. daskol 2023-05-23 08:33 (UTC)
python-jmp 0.0.4-3 1 0.00 A mixed precision library for JAX daskol 2023-06-15 11:40 (UTC)
python-jaxtyping 0.2.32-1 1 0.00 Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. daskol 2024-07-12 10:38 (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-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-orbax-export 0.0.3-1 0 0.00 Orbax provides common utility libraries for JAX users (export). daskol 2024-01-01 16:15 (UTC)
python-orbax-checkpoint 0.5.20-1 0 0.00 Orbax provides common utility libraries for JAX users (checkpoint). daskol 2024-06-27 16:15 (UTC)
python-deepxde 1.11.1-1 0 0.00 Library for scientific machine learning carlosal1015 2024-05-05 16:39 (UTC)

13 packages found. Page 1 of 1.