| Name | Version | Summary | date |
| jaxutils-nightly |
0.0.8.dev20240727 |
Utility functions for JaxGaussianProcesses |
2024-07-27 00:06:44 |
| jaxkern-nightly |
0.0.5.dev20240727 |
Kernels in Jax. |
2024-07-27 00:05:05 |
| gwkokab |
0.1.0 |
A JAX-based gravitational-wave population inference |
2024-07-26 21:07:52 |
| transformers |
4.43.3 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-07-26 15:30:35 |
| flowjax |
13.0.1 |
Easy to use distributions, bijections and normalizing flows in JAX. |
2024-07-24 14:52:56 |
| fjformer |
0.0.73 |
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-07-20 14:34:37 |
| jax-tqdm |
0.2.2 |
Tqdm progress bar for JAX scans and loops |
2024-07-17 20:30:25 |
| thermox |
0.0.2 |
Exact OU processes with JAX |
2024-07-16 13:57:40 |
| jax-tools |
1.0.147 |
Jax common tools library |
2024-07-12 01:47:12 |
| jaxtra |
0.0.2 |
JAXTRA |
2024-07-09 04:41:43 |
| transformers-backport |
4.33.4 |
transformers backport - version 4.33.x |
2024-07-04 23:15:53 |
| easydel |
0.0.69 |
An open-source library to make training faster and more optimized in Jax/Flax |
2024-07-04 15:28:39 |
| NetKet |
3.12.4 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2024-07-04 14:37:56 |
| roger |
3.0.8 |
Runoff Generation Research - a process-based hydrological toolbox model in Python |
2024-07-03 10:47:23 |
| jax-dataloader |
0.1.1 |
Dataloader for jax |
2024-06-29 00:40:22 |
| diffusers |
0.29.2 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-06-27 03:58:41 |
| inox |
0.6.1 |
Stainless neural networks in JAX |
2024-06-23 08:51:32 |
| mewtax |
0.0.1 |
Differentiable minimization in jax using Newton's method. |
2024-06-14 19:52:51 |
| FJFormer |
0.0.66 |
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-06-12 13:32:40 |
| jeig |
0.0.2 |
Various eigendecomposition implementations wrapped for jax. |
2024-06-11 18:35:53 |