| Name | Version | Summary | date |
| jax-tqdm |
0.2.0 |
Tqdm progress bar for JAX scans and loops |
2024-04-26 10:40:17 |
| jax-tools |
1.0.112 |
Jax common tools library |
2024-04-26 04:15:18 |
| jaxutils-nightly |
0.0.8.dev20240426 |
Utility functions for JaxGaussianProcesses |
2024-04-26 00:06:53 |
| jaxkern-nightly |
0.0.5.dev20240426 |
Kernels in Jax. |
2024-04-26 00:05:01 |
| folx |
0.2.6 |
Forward Laplacian for JAX |
2024-04-25 12:37:51 |
| transformers |
4.40.1 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-04-23 22:01:15 |
| flowjax |
12.1.2 |
Easy to use distributions, bijections and normalizing flows in JAX. |
2024-04-22 12:30:57 |
| jax-fem |
0.0.4 |
GPU accelerated Finite element analysis package in JAX. |
2024-04-20 08:12:16 |
| keras-vision |
0.3 |
Building Vision models in Keras3 for framework agnostic training and inference. |
2024-04-19 21:12:04 |
| 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-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 |
| 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 |