| Name | Version | Summary | date |
| mixed-precision-for-JAX |
0.1.8 |
A toolbox for mixed precision training via JAX. |
2025-07-14 10:56:33 |
| tbp-nightly |
2.21.2a20250714 |
XProf Profiler Plugin |
2025-07-14 09:32:14 |
| xprof-nightly |
2.21.2a20250714 |
XProf Profiler Plugin |
2025-07-14 09:32:09 |
| tensorboard-plugin-profile |
2.20.2 |
XProf Profiler Plugin |
2025-07-11 23:04:46 |
| xprof |
2.20.2 |
XProf Profiler Plugin |
2025-07-11 23:04:45 |
| transformers |
4.53.2 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-07-11 12:39:08 |
| 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 |
| eformer |
0.0.14 |
(EasyDel Former) is a utility library designed to simplify and enhance the development in JAX |
2025-02-26 19:12:42 |
| my-transformers |
4.5.1 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-02-26 10:40:18 |
| sbx-rl |
0.20.0 |
Jax version of Stable Baselines, implementations of reinforcement learning algorithms. |
2025-02-14 11:03:47 |
| NetKet |
3.16.1.post1 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2025-02-07 11:03:15 |
| folx |
0.2.14 |
Forward Laplacian for JAX |
2025-01-30 13:06:08 |
| inference-gym |
0.0.5 |
The Inference Gym is the place to exercise inference methods to help make them faster, leaner and more robust. |
2025-01-24 19:56:04 |
| pytreeclass |
0.11.0 |
Visualize, create, and operate on pytrees in the most intuitive way possible. |
2025-01-11 19:13:53 |
| t-draft-123 |
4.48.0.dev0 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-01-07 14:25:43 |