| Name | Version | Summary | date |
| transformers |
4.55.0 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-08-05 16:13:48 |
| tbp-nightly |
2.21.5a20250805 |
XProf Profiler Plugin |
2025-08-05 09:21:20 |
| xprof-nightly |
2.21.5a20250805 |
XProf Profiler Plugin |
2025-08-05 09:21:18 |
| tensorboard-plugin-profile |
2.20.5 |
XProf Profiler Plugin |
2025-08-01 22:15:00 |
| xprof |
2.20.5 |
XProf Profiler Plugin |
2025-08-01 22:14:57 |
| jsrm |
0.1.0 |
Kinematic and dynamic models of continuum and articulated soft robots. |
2025-07-31 12:38:15 |
| sbx-rl |
0.22.0 |
Jax version of Stable Baselines, implementations of reinforcement learning algorithms. |
2025-07-25 09:34:53 |
| traceax |
1.0.2 |
Stochastic trace estimation in JAX, Lineax, and Equinox |
2025-07-24 04:40:21 |
| chex |
0.1.90 |
Chex: Testing made fun, in JAX! |
2025-07-23 19:50:47 |
| jaxbo |
0.1.2 |
Fork of PredictiveIntelligenceLab/JAX-BO with updates and compatibility improvements for Colab and pipelines |
2025-07-21 07:32:04 |
| eformer |
0.0.45 |
(EasyDel Former) is a utility library designed to simplify and enhance the development in JAX |
2025-07-20 22:24:39 |
| NetKet |
3.18.1 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2025-07-18 10:48:01 |
| mixed-precision-for-JAX |
0.1.8 |
A toolbox for mixed precision training via JAX. |
2025-07-14 10:56:33 |
| jeig |
0.3.0 |
Various eigendecomposition implementations wrapped for jax. |
2025-07-10 14:54:31 |
| kajihs_utils |
0.8.0 |
Fully typed, plausibly practical, and remarkably random utilities for me—and maybe for you too. |
2025-04-20 21:57:55 |
| jaxkan |
0.2.8 |
A JAX implementation of Kolmogorov-Arnold Networks |
2025-04-09 10:55:48 |
| jaxutils-nightly |
0.0.8.dev20250315 |
Utility functions for JaxGaussianProcesses |
2025-03-15 00:06:48 |
| jaxkern-nightly |
0.0.5.dev20250303 |
Kernels in Jax. |
2025-03-03 00:06:09 |
| rehydrax |
0.1.0 |
Rehydrax: A utility library to rehydrate stablehlo into JAX. |
2025-02-26 22:33:53 |
| my-transformers |
4.5.1 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-02-26 10:40:18 |