Jaxmod
About
Jaxmod—short for "JAX for modelling"—is a lightweight "extension pack" for JAX-based scientific computing. It provides convenience utilities, wrappers, and conventions on top of the excellent Equinox and Optimistix libraries, making them even easier to use for real scientific workflows.
The library was originally created to avoid code duplication in thermochemistry applications for planetary science. Many such problems share similar structural patterns—stoichiometric systems, batched equilibrium calculations, constraints, and differentiable solvers. Jaxmod consolidates this boilerplate into reusable components and establishes consistent conventions, whether for vectorising models efficiently or extending solver behaviour in a principled way.
Documentation
The documentation is available online, with options to download it in EPUB or PDF format:
https://jaxmod.readthedocs.io/en/latest/
Quick install
Jaxmod is a Python package that can be installed on a variety of platforms (e.g. Mac, Windows, Linux). It is recommended to install Jaxmod in a dedicated Python environment. Before installation, create and activate the environment, then run:
pip install jaxmod
1 maintainer
Metadata
1 maintainer
Release files for jaxmod 0.4.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| jaxmod-0.4.0.tar.gz | 34.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jaxmod-0.4.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 62.6 kB
Release files / jaxmod-0.4.0.tar.gz
| Download URL | jaxmod-0.4.0.tar.gz |
|---|---|
| Size | 34.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
786950e6847f8776167d26e09e1ad2f24aaa8fe56d449ee70881e5a4b6faf8b4
|
|
BLAKE2b-256 checksum How to use checksums |
f72c0efea976eeaa98c4ff49e39545d7ac58d88ccb9f41d3c39a5636795f81f3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.7.20
|
Release files / jaxmod-0.4.0-py3-none-any.whl
| Download URL | jaxmod-0.4.0-py3-none-any.whl |
|---|---|
| Size | 28.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
c05ada75f81c96601303ad1018b607905ed38e640c6087b26d26c3561b704906
|
|
BLAKE2b-256 checksum How to use checksums |
c2f0580807b4cfeb7d2a1756a72fd8de08a6ed6591bec56865bbb1767177c102
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.7.20
|
