Index

A

as_j64() (in module jaxmod.utils) asdict() (jaxmod.solvers.MultiAttemptSolution method) ATMOSPHERE (in module jaxmod.constants) atmosphere_to_bar (jaxmod.units.UnitConversion attribute)

atol (jaxmod.solvers.RootFindParameters attribute) attempts (jaxmod.solvers.MultiAttemptSolution property) aux (jaxmod.solvers.MultiAttemptSolution property) AVOGADRO (in module jaxmod.constants)

B

bar_to_GPa (jaxmod.units.UnitConversion attribute) bar_to_MPa (jaxmod.units.UnitConversion attribute) bar_to_Pa (jaxmod.units.UnitConversion attribute)

batch_size (jaxmod.solvers.MultiAttemptSolution property) BOLTZMANN_CONSTANT (in module jaxmod.constants) BOLTZMANN_CONSTANT_BAR (in module jaxmod.constants)

C

cm3_to_m3 (jaxmod.units.UnitConversion attribute)

converged (jaxmod.solvers.MultiAttemptSolution property)

D

debug_logger() (in module jaxmod)

E

EARTH_MASS (in module jaxmod.constants)

F

fraction_to_ppm (jaxmod.units.UnitConversion attribute)

G

g_to_kg (jaxmod.units.UnitConversion attribute) GAS_CONSTANT (in module jaxmod.constants) GAS_CONSTANT_BAR (in module jaxmod.constants) get_batch_axis() (in module jaxmod.utils)

get_batch_size() (in module jaxmod.utils) get_solver_instance() (jaxmod.solvers.RootFindParameters method) GPa_to_bar (jaxmod.units.UnitConversion attribute) GRAVITATIONAL_CONSTANT (in module jaxmod.constants)

I

is_hashable() (in module jaxmod.utils)

is_jax_array() (in module jaxmod.utils)

J

J_to_m3_bar (jaxmod.units.UnitConversion attribute) jac (jaxmod.solvers.RootFindParameters attribute) jaxmod module jaxmod.constants module jaxmod.solvers module

jaxmod.type_aliases module jaxmod.units module jaxmod.utils module

L

linear_solver (jaxmod.solvers.RootFindParameters attribute)

litre_to_m3 (jaxmod.units.UnitConversion attribute)

M

m3_bar_to_J (jaxmod.units.UnitConversion attribute) m3_to_cm3 (jaxmod.units.UnitConversion attribute) make_batch_retry_solver() (in module jaxmod.solvers) MAX_EXP_INPUT (in module jaxmod) max_norm() (in module jaxmod.solvers) max_steps (jaxmod.solvers.RootFindParameters attribute) MIN_EXP_INPUT (in module jaxmod) module jaxmod jaxmod.constants jaxmod.solvers jaxmod.type_aliases jaxmod.units jaxmod.utils

MPa_to_bar (jaxmod.units.UnitConversion attribute) MultiAttemptSolution (class in jaxmod.solvers)

N

norm() (jaxmod.solvers.RootFindParameters method) NpArray (in module jaxmod.type_aliases) NpBool (in module jaxmod.type_aliases)

NpFloat (in module jaxmod.type_aliases) NpInt (in module jaxmod.type_aliases) num_steps (jaxmod.solvers.MultiAttemptSolution property)

O

OCEAN_MASS_H2 (in module jaxmod.constants) OCEAN_MASS_H2O (in module jaxmod.constants)

OCEAN_MOLES (in module jaxmod.constants) OptxSolver (in module jaxmod.type_aliases)

P

Pa_to_bar (jaxmod.units.UnitConversion attribute) partial_rref() (in module jaxmod.utils) percent_to_ppm (jaxmod.units.UnitConversion attribute)

POSTCHECK_TOLERANCE (in module jaxmod.solvers) power_law() (in module jaxmod.utils) ppm_to_fraction (jaxmod.units.UnitConversion attribute) ppm_to_percent (jaxmod.units.UnitConversion attribute)

R

result (jaxmod.solvers.MultiAttemptSolution property)

RootFindParameters (class in jaxmod.solvers) rtol (jaxmod.solvers.RootFindParameters attribute)

S

safe_exp() (in module jaxmod.utils) Scalar (in module jaxmod.type_aliases) simple_formatter() (in module jaxmod) solution (jaxmod.solvers.MultiAttemptSolution attribute)

solver (jaxmod.solvers.RootFindParameters attribute) solver_success (jaxmod.solvers.MultiAttemptSolution property) state (jaxmod.solvers.MultiAttemptSolution property) stats (jaxmod.solvers.MultiAttemptSolution property)

T

throw (jaxmod.solvers.RootFindParameters attribute)

to_hashable() (in module jaxmod.utils) to_native_floats() (in module jaxmod.utils)

U

UnitConversion (class in jaxmod.units)

V

value (jaxmod.solvers.MultiAttemptSolution property)

vmap_axes_spec() (in module jaxmod.utils)