JAXprop

jaxprop provides JAX-compatible thermodynamic property calculations with support for automatic differentiation, vectorization, and JIT compilation.

🔗 Docs: turbo-sim.github.io/jaxprop
📦 PyPI: pypi.org/project/jaxprop

Note: This project is based on the CoolProp library but is not affiliated with or endorsed by the CoolProp project.

Key features

• Compute and plot phase envelopes and spinodal lines for pure fluids.
• Evaluate thermodynamic properties from Helmholtz energy–based equations of state, including metastable states inside the two-phase region.
• Perform flash calculations for any input pair with a custom solver and user-defined initial guesses.
• Work with structured property dictionaries and immutable FluidState objects.
• Evaluate properties over arrays of input conditions for efficient parametric studies and plotting.
• Full JAX compatibility: supports jit, grad, vmap, and parallel evaluation.

Installation

pip install jaxprop