| Name | Version | Summary | date |
| tbp-nightly |
2.21.8a20251030 |
XProf Profiler Plugin |
2025-10-30 09:44:54 |
| xprof-nightly |
2.21.8a20251030 |
XProf Profiler Plugin |
2025-10-30 09:44:51 |
| gpjax |
0.13.2 |
Gaussian processes in JAX. |
2025-10-30 09:29:04 |
| tensorboard-plugin-profile |
2.20.8 |
XProf Profiler Plugin |
2025-10-30 01:07:29 |
| xprof |
2.20.8 |
XProf Profiler Plugin |
2025-10-30 01:07:26 |
| bbob-jax |
0.5.0 |
BBOB Benchmark function implemented in JAX |
2025-10-27 11:15:39 |
| eformer |
0.0.80 |
(EasyDel Former) is a utility library designed to simplify and enhance the development in JAX |
2025-10-21 15:30:48 |
| NetKet |
3.20.2 |
Netket : Machine Learning toolbox for many-body quantum systems. |
2025-10-20 21:51:53 |
| lovely-jax |
0.1.4 |
💘 Lovely JAX |
2025-10-19 03:52:56 |
| feax |
0.0.5 |
A GPU-accelerated finite element analysis framework with JAX. |
2025-10-15 05:24:50 |
| transformers |
4.57.1 |
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-10-14 15:39:26 |
| transformers-usf |
4.56.0.post11 |
Enhanced Transformers library with Omega3 model support - State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow |
2025-10-14 00:57:45 |
| jraphx |
0.0.3 |
Graph Neural Networks with JAX/NNX |
2025-09-05 01:56:30 |
| distrax |
0.1.7 |
Distrax: Probability distributions in JAX. |
2025-09-01 23:06:00 |
| chex |
0.1.91 |
Chex: Testing made fun, in JAX! |
2025-09-01 21:49:32 |
| folx |
0.2.19 |
Forward Laplacian for JAX |
2025-08-28 13:53:49 |
| tf2jax |
0.3.8 |
TF2JAX: Convert TensorFlow to JAX |
2025-08-21 13:18:24 |
| diffusers |
0.35.1 |
State-of-the-art diffusion in PyTorch and JAX. |
2025-08-20 04:16:10 |
| sif2jax |
0.0.6 |
Functionally pure definitions of optimisation problems extracted from Standard Input Format (SIF). |
2025-08-17 16:54:50 |
| brax |
0.13.0 |
A differentiable physics engine written in JAX. |
2025-08-15 18:50:22 |