13 packages found. Page 1 of 1.

Name Version Votes Popularity? Description Maintainer Last Updated
python-jaxlib-cuda 0.4.26-1 5 1.73 XLA library for JAX daskol 2024-04-04 09:49 (UTC)
python-jaxlib 0.4.26-1 6 0.89 XLA library for JAX daskol 2024-04-03 23:32 (UTC)
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-flax 0.8.2-1 2 0.87 A neural network library and ecosystem for JAX designed for flexibility wjhandley 2024-03-14 13:07 (UTC)
python-jax 0.4.27-1 6 0.87 Composable transformations of Python+NumPy programs daskol 2024-05-08 11:11 (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.28-1 1 0.00 Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. daskol 2024-03-10 11:39 (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-1 1 0.00 A library of utilities for helping to write reliable JAX code daskol 2024-04-17 12:38 (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.4.8-1 0 0.00 Orbax provides common utility libraries for JAX users (checkpoint). daskol 2024-01-01 16:11 (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.