|
python-jaxlib-cuda
|
0.4.16-4 |
2 |
0.04
|
XLA library for JAX |
daskol
|
2023-09-22 10:43 (UTC) |
|
python-jaxlib
|
0.4.16-3 |
4 |
0.02
|
XLA library for JAX |
daskol
|
2023-09-22 10:28 (UTC) |
|
python-jaxtyping
|
0.2.21-1 |
1 |
0.01
|
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. |
daskol
|
2023-08-17 16:27 (UTC) |
|
python-orbax
|
0.1.7-1 |
1 |
0.01
|
Orbax is a library providing common utilities for JAX users. |
daskol
|
2023-05-23 08:33 (UTC) |
|
python-optax
|
0.1.7-1 |
1 |
0.01
|
A gradient processing and optimization library for JAX |
daskol
|
2023-07-26 22:26 (UTC) |
|
python-chex
|
0.1.83-1 |
1 |
0.01
|
A library of utilities for helping to write reliable JAX code |
daskol
|
2023-09-21 10:31 (UTC) |
|
python-jmp
|
0.0.4-3 |
1 |
0.01
|
A mixed precision library for JAX |
daskol
|
2023-06-15 11:40 (UTC) |
|
python-dm-haiku
|
0.0.10-2 |
1 |
0.01
|
A library for building neural networks in JAX |
daskol
|
2023-07-24 14:04 (UTC) |
|
python-flax
|
0.7.1-5 |
1 |
0.00
|
A neural network library and ecosystem for JAX designed for flexibility |
wjhandley
|
2023-08-01 03:08 (UTC) |
|
python-jax
|
0.4.16-1 |
5 |
0.00
|
Differentiate, compile, and transform Numpy code. |
daskol
|
2023-09-21 10:36 (UTC) |
|
python-orbax-export
|
0.0.2-2 |
0 |
0.00
|
Orbax provides common utility libraries for JAX users (export). |
daskol
|
2023-07-10 20:47 (UTC) |
|
python-orbax-checkpoint
|
0.2.7-2 |
0 |
0.00
|
Orbax provides common utility libraries for JAX users (checkpoint). |
daskol
|
2023-07-10 20:44 (UTC) |
|
python-ml-dtypes
|
0.3.0-1 |
0 |
0.00
|
A stand-alone implementation of several NumPy dtype extensions used in ML. |
daskol
|
2023-09-20 14:28 (UTC) |
|
python-deepxde
|
1.9.3-1 |
0 |
0.00
|
A library for scientific machine learning |
carlosal1015
|
2023-09-06 01:31 (UTC) |