| Name | Version | Summary | date |
| jaxutils-nightly |
0.0.8.dev20241227 |
Utility functions for JaxGaussianProcesses |
2024-12-27 00:07:09 |
| jaxkern-nightly |
0.0.5.dev20241227 |
Kernels in Jax. |
2024-12-27 00:05:00 |
| jpviz |
0.1.7 |
Jaxpr Visualisation Tool |
2024-12-20 00:31:04 |
| jeig |
0.2.1 |
Various eigendecomposition implementations wrapped for jax. |
2024-12-17 16:30:10 |
| transformers |
4.47.1 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-12-17 15:42:41 |
| rex-lib |
0.0.9 |
Robotic Environments with jaX (REX) |
2024-12-14 21:44:30 |
| brax |
0.12.1 |
A differentiable physics engine written in JAX. |
2024-12-11 18:53:44 |
| chex |
0.1.88 |
Chex: Testing made fun, in JAX! |
2024-12-11 10:34:03 |
| dm-reverb-macos |
0.11.0.dev0 |
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research. |
2024-12-08 03:09:07 |
| dm-reverb-macos-nightly |
0.11.20241207 |
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research. |
2024-12-08 03:04:02 |
| NetKet |
3.15.1 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2024-12-05 21:13:10 |
| diffusers-qaic |
0.24.0 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-12-05 15:19:37 |
| easydel |
0.0.80 |
An open-source library to make training faster and more optimized in JAX |
2024-12-04 15:22:10 |
| jax-hpc-profiler |
0.2.9 |
HPC Plotter and profiler for benchmarking data made for JAX |
2024-11-29 19:38:37 |
| matrax |
0.0.5 |
Matrix games in JAX |
2024-11-28 09:02:43 |
| jax-scipy-spatial |
0.2.5 |
Scipy spatial API for JAX |
2024-11-27 16:26:13 |
| jsrm |
0.0.13 |
Kinematic and dynamic models of continuum and articulated soft robots. |
2024-11-22 23:56:38 |
| fjformer |
0.0.91 |
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer in JAX |
2024-11-14 10:11:07 |
| apebench |
0.1.1 |
Benchmark suite for Autoregressive Neural Emulators of PDEs in JAX. |
2024-11-09 08:49:25 |
| gpjax |
0.9.3 |
Gaussian processes in JAX. |
2024-11-01 19:42:49 |