| Name | Version | Summary | date |
| jaxutils-nightly |
0.0.8.dev20240419 |
Utility functions for JaxGaussianProcesses |
2024-04-19 00:06:48 |
| jaxkern-nightly |
0.0.5.dev20240419 |
Kernels in Jax. |
2024-04-19 00:05:13 |
| jax-fdm |
0.8.3 |
Auto-differentiable and hardware-accelerated force density method |
2024-04-18 20:12:45 |
| EasyDeL |
0.0.61 |
An open-source library to make training faster and more optimized in Jax/Flax |
2024-04-17 15:46:57 |
| FJFormer |
0.0.50 |
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal of custom Jax Flax Functions and Utils that elevate your AI endeavors to new heights! |
2024-04-16 13:18:59 |
| NetKet |
3.11.4 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2024-04-11 11:44:34 |
| clu |
0.0.12 |
Set of libraries for ML training loops in JAX. |
2024-04-10 20:36:06 |
| jax-tools |
1.0.110 |
Jax common tools library |
2024-04-08 01:48:52 |
| jax-relax |
0.2.6 |
JAX-based Recourse Explanation Library |
2024-04-04 15:26:01 |
| sobek |
0.0.0 |
Computer vision for JAX |
2024-04-03 13:41:29 |
| jax-scipy-spatial |
0.2.3 |
Scipy spatial API for JAX |
2024-04-02 18:30:49 |
| transformers |
4.39.3 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-04-02 09:36:06 |
| sepes |
0.12.0 |
Visualize, create, and operate on nested trees in the most intuitive way possible. |
2024-03-30 17:21:49 |
| fenbux |
0.1.0 |
A Simple Statistical Distribution Library In JAX |
2024-03-30 07:02:44 |
| jumanji |
1.0.1 |
A diverse suite of scalable reinforcement learning environments in JAX |
2024-03-29 11:24:18 |
| autobnn |
0.0.2.dev0 |
Package for training Gaussian process-like Bayesian Neural Networks with composite structure. |
2024-03-28 00:19:39 |
| tf2jax |
0.3.6 |
TF2JAX: Convert TensorFlow to JAX |
2024-03-26 10:38:21 |
| souJpg-diffusers |
0.27.2 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-03-25 00:20:06 |
| hydrax |
0.2.2 |
A zero-copy multiprocess dataloader for JAX. |
2024-03-24 19:43:42 |
| xminigrid |
0.7.0 |
JAX-accelerated meta-reinforcement learning environments inspired by XLand and MiniGrid |
2024-03-24 11:00:37 |