NEOPAX._state

Attributes

JOULE_PER_EV
JOULE_PER_KEV
DEFAULT_TRANSPORT_DENSITY_FLOOR
DEFAULT_TRANSPORT_TEMPERATURE_FLOOR

Classes

TransportState

JAX-compatible transport state for arbitrary number of species.

Functions

_broadcast_species_floor(→ jaxtyping.Array)
safe_density(→ jaxtyping.Array)
safe_temperature(→ jaxtyping.Array)
apply_transport_density_floor(state[, density_floor])
apply_transport_temperature_floor(state[, ...])
get_v_thermal(mass, temperature)

Module Contents

NEOPAX._state.JOULE_PER_EV

NEOPAX._state.JOULE_PER_KEV

NEOPAX._state.DEFAULT_TRANSPORT_DENSITY_FLOOR = 1e-06

NEOPAX._state.DEFAULT_TRANSPORT_TEMPERATURE_FLOOR = None

NEOPAX._state._broadcast_species_floor(reference: jaxtyping.Array, floor) → jaxtyping.Array

NEOPAX._state.safe_density(density: jaxtyping.Array, floor=DEFAULT_TRANSPORT_DENSITY_FLOOR) → jaxtyping.Array

NEOPAX._state.safe_temperature(temperature: jaxtyping.Array, floor) → jaxtyping.Array

NEOPAX._state.apply_transport_density_floor(state, density_floor=DEFAULT_TRANSPORT_DENSITY_FLOOR)

NEOPAX._state.apply_transport_temperature_floor(state, temperature_floor=DEFAULT_TRANSPORT_TEMPERATURE_FLOOR, density_floor=DEFAULT_TRANSPORT_DENSITY_FLOOR)

NEOPAX._state.get_v_thermal(mass, temperature)

class NEOPAX._state.TransportState

JAX-compatible transport state for arbitrary number of species. All fields are JAX arrays for differentiability and vmap support.

density: jaxtyping.Float[jaxtyping.Array, ...]

pressure: jaxtyping.Float[jaxtyping.Array, ...]

Er: jaxtyping.Float[jaxtyping.Array, ...]

property temperature