| Name | Version | Summary | date |
| jaxtyping |
0.2.28 |
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. |
2024-03-07 17:31:29 |
| seasons-cal |
1.0.1 |
|
2024-03-05 04:46:43 |
| quax |
0.0.3 |
Multiple dispatch in JAX via custom interpreters. |
2024-02-02 11:11:20 |
| diffrax |
0.5.0 |
GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX. |
2024-01-08 23:23:19 |
| optimistix |
0.0.6 |
Nonlinear optimisation in JAX and Equinox. |
2023-12-27 17:06:26 |
| sympy2jax |
0.0.5 |
Turn SymPy expressions into trainable JAX expressions. |
2023-10-21 21:37:40 |
| ezjaxtyping |
0.2.20 |
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees. |
2023-07-20 18:27:43 |
| atomcloud |
0.0.1 |
Conventional method for fitting atom cloud, BECs and bimodal atom cloud/BEC distributions in 1D and 2D. |
2023-06-23 13:32:15 |