| Name | Version | Summary | date |
| tbp-nightly |
2.20.0a20250202 |
Profile Tensorboard Plugin |
2025-02-02 10:44:58 |
| jaxutils-nightly |
0.0.8.dev20250202 |
Utility functions for JaxGaussianProcesses |
2025-02-02 00:06:44 |
| jaxkern-nightly |
0.0.5.dev20250202 |
Kernels in Jax. |
2025-02-02 00:05:04 |
| folx |
0.2.14 |
Forward Laplacian for JAX |
2025-01-30 13:06:08 |
| tensorboard-plugin-profile |
2.19.0 |
Profile Tensorboard Plugin |
2025-01-29 19:44:43 |
| 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 |
| transformers |
4.48.1 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-01-20 16:36:07 |
| jeig |
0.2.4 |
Various eigendecomposition implementations wrapped for jax. |
2025-01-15 04:47:36 |
| pytreeclass |
0.11.0 |
Visualize, create, and operate on pytrees in the most intuitive way possible. |
2025-01-11 19:13:53 |
| jaxkan |
0.2.0 |
A JAX implementation of Kolmogorov-Arnold Networks |
2025-01-07 16:35:47 |
| t-draft-123 |
4.48.0.dev0 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-01-07 14:25:43 |
| rax |
0.4.0 |
Learning-to-Rank using JAX. |
2025-01-03 19:06:42 |
| sepes |
0.12.1 |
Visualize, create, and operate on nested trees in the most intuitive way possible. |
2024-12-30 16:55:23 |
| jpviz |
0.1.7 |
Jaxpr Visualisation Tool |
2024-12-20 00:31:04 |
| rex-lib |
0.0.9 |
Robotic Environments with jaX (REX) |
2024-12-14 21:44:30 |
| brax |
0.12.1 |
A differentiable physics engine written in JAX. |
2024-12-11 18:53:44 |
| chex |
0.1.88 |
Chex: Testing made fun, in JAX! |
2024-12-11 10:34:03 |
| dm-reverb-macos |
0.11.0.dev0 |
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research. |
2024-12-08 03:09:07 |
| dm-reverb-macos-nightly |
0.11.20241207 |
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research. |
2024-12-08 03:04:02 |
| NetKet |
3.15.1 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2024-12-05 21:13:10 |