| Name | Version | Summary | date |
| jaxmod |
0.2.3 |
Lightweight utility functions for JAX arrays, batching, and pytrees |
2025-10-25 09:38:31 |
| flowjax |
17.2.1 |
Easy to use distributions, bijections and normalizing flows in JAX. |
2025-10-19 18:25:21 |
| quax |
0.2.1 |
Multiple dispatch in JAX via custom interpreters. |
2025-10-12 18:58:27 |
| 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 |
| 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 |