| Name | Version | Summary | date |
| lineax |
0.0.7 |
Linear solvers in JAX and Equinox. |
2024-10-21 10:54:53 |
| optimistix |
0.0.9 |
Nonlinear optimisation in JAX and Equinox. |
2024-10-21 09:20:27 |
| equinox |
0.11.8 |
Elegant easy-to-use neural networks in JAX. |
2024-10-18 17:19:34 |
| flowjax |
16.0.0 |
Easy to use distributions, bijections and normalizing flows in JAX. |
2024-10-17 11:00:06 |
| seasons-cal |
1.0.2 |
None |
2024-10-15 15:21:15 |
| distreqx |
0.0.1 |
Lightweight JAX library of probability distributions and bijectors. |
2024-09-15 06:13:45 |
| cryojax |
0.3.1 |
Cryo-EM image simulation and analysis powered by JAX |
2024-09-11 15:03:19 |
| jaxtyping |
0.2.34 |
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. |
2024-09-01 14:44:44 |
| diffrax |
0.6.0 |
GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX. |
2024-07-01 09:16:43 |