piwheels

jaxmod

Lightweight utility functions for JAX arrays, batching, and pytrees

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install jaxmod

Dependencies

• equinox
• pandas
• molmass
• jax
• optimistix

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.4.0	2025-12-16
jaxmod-0.4.0-py3-none-any.whl (27 KB)
0.3.0	2025-11-10
jaxmod-0.3.0-py3-none-any.whl (27 KB)
0.2.4	2025-10-31
jaxmod-0.2.4-py3-none-any.whl (25 KB)
0.2.3	2025-10-25
jaxmod-0.2.3-py3-none-any.whl (22 KB)
0.2.2	2025-09-11
jaxmod-0.2.2-py3-none-any.whl (22 KB)
0.2.0	2025-09-10
jaxmod-0.2.0-py3-none-any.whl (22 KB)
0.1.0	2025-09-10
jaxmod-0.1.0-py3-none-any.whl (20 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue