| Name | Version | Summary | date |
| jax-tools |
1.0.140 |
Jax common tools library |
2024-05-07 09:24:11 |
| jaxutils-nightly |
0.0.8.dev20240507 |
Utility functions for JaxGaussianProcesses |
2024-05-07 00:06:33 |
| jaxkern-nightly |
0.0.5.dev20240507 |
Kernels in Jax. |
2024-05-07 00:05:04 |
| transformers |
4.40.2 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2024-05-06 16:08:02 |
| FJFormer |
0.0.51 |
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-06 10:40:41 |
| keras-vision |
0.4.3 |
Building Vision models in Keras3 for framework agnostic training and inference. |
2024-05-05 19:01:00 |
| jax-metal |
0.0.7 |
JAX acceleration for Mac GPUs. |
2024-05-04 00:36:59 |
| jsrm |
0.0.7 |
Kinematic and dynamic models of continuum and articulated soft robots. |
2024-05-03 09:40:07 |
| 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 |
| EasyDeL |
0.0.63 |
An open-source library to make training faster and more optimized in Jax/Flax |
2024-04-27 12:48:14 |
| jax-tqdm |
0.2.0 |
Tqdm progress bar for JAX scans and loops |
2024-04-26 10:40:17 |
| folx |
0.2.6 |
Forward Laplacian for JAX |
2024-04-25 12:37:51 |
| 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 |
| jax-fdm |
0.8.3 |
Auto-differentiable and hardware-accelerated force density method |
2024-04-18 20:12:45 |
| 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 |
| 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 |