| Name | Version | Summary | date |
| jax-optix |
0.1.8 |
Zero-overhead functional lensing for JAX PyTrees |
2025-02-13 16:06:02 |
| sympy2jax |
0.0.6 |
Turn SymPy expressions into trainable JAX expressions. |
2025-01-02 00:46:32 |
| paramax |
0.0.0 |
Parameterizations and parameter constraints for JAX PyTrees. |
2024-10-21 12:19:07 |
| quax |
0.0.5 |
Multiple dispatch in JAX via custom interpreters. |
2024-09-17 09:32:02 |
| 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 |