| Name | Version | Summary | date |
| FJFormer |
0.0.61 |
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-05-28 21:42:37 |
| jaxutils-nightly |
0.0.8.dev20240528 |
Utility functions for JaxGaussianProcesses |
2024-05-28 00:06:47 |
| jaxkern-nightly |
0.0.5.dev20240528 |
Kernels in Jax. |
2024-05-28 00:05:05 |
| diffusers |
0.28.0 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-05-27 12:08:44 |
| bosonic |
0.1.1 |
Bosonic JAX |
2024-05-27 05:03:16 |
| nanodl |
1.2.5.dev1 |
A Jax-based library for designing and training transformer models from scratch. |
2024-05-26 21:28:50 |
| transformers |
4.41.1 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-05-22 20:36:46 |
| jsrm |
0.0.9 |
Kinematic and dynamic models of continuum and articulated soft robots. |
2024-05-18 06:27:51 |
| diffusersv |
0.28.0.dev0 |
State-of-the-art diffusion in PyTorch and JAX. |
2024-05-17 02:57:47 |
| flowjax |
12.2.0 |
Easy to use distributions, bijections and normalizing flows in JAX. |
2024-05-16 16:49:00 |
| EasyDeL |
0.0.65 |
An open-source library to make training faster and more optimized in Jax/Flax |
2024-05-16 09:39:43 |
| jax-tqdm |
0.2.1 |
Tqdm progress bar for JAX scans and loops |
2024-05-15 08:39:38 |
| autoray |
0.6.11 |
Abstract your array operations. |
2024-05-14 18:21:42 |
| brax |
0.10.4 |
A differentiable physics engine written in JAX. |
2024-05-13 18:17:59 |
| jax-fdm |
0.8.4 |
Auto-differentiable and hardware-accelerated force density method |
2024-05-09 14:21:29 |
| jax-tools |
1.0.142 |
Jax common tools library |
2024-05-08 08:40:24 |
| jax-metal |
0.0.7 |
JAX acceleration for Mac GPUs. |
2024-05-04 00:36:59 |
| xminigrid |
0.7.1 |
JAX-accelerated meta-reinforcement learning environments inspired by XLand and MiniGrid |
2024-05-01 13:25:55 |
| jax-relax |
0.2.7 |
JAX-based Recourse Explanation Library |
2024-04-27 19:20:08 |
| folx |
0.2.6 |
Forward Laplacian for JAX |
2024-04-25 12:37:51 |