NEOPAX._energy_grid_models¶

Modular energy grid models for NEOPAX. - Registry-based selection (like transport_flux_models.py) - JIT-compatible, differentiable (JAX-friendly) - No species or radial grid info (energy grid only)

Attributes¶

ENERGY_GRID_MODEL_REGISTRY

Classes¶

`EnergyGridModelBase`	Abstract base class for energy grid models.
`StandardLaguerreEnergyGrid`	Abstract base class for energy grid models.

Functions¶

`register_energy_grid_model`(name, builder)
`get_energy_grid_model`(→ EnergyGridModelBase)

Module Contents¶

NEOPAX._energy_grid_models.ENERGY_GRID_MODEL_REGISTRY: dict[str, Callable[Ellipsis, EnergyGridModelBase]]¶

NEOPAX._energy_grid_models.register_energy_grid_model(name: str, builder: Callable[Ellipsis, EnergyGridModelBase])¶

NEOPAX._energy_grid_models.get_energy_grid_model(name: str, **kwargs) → EnergyGridModelBase¶

class NEOPAX._energy_grid_models.EnergyGridModelBase¶: Abstract base class for energy grid models.

class NEOPAX._energy_grid_models.StandardLaguerreEnergyGrid(n_x: int, n_order: int = 3, **kwargs)¶

Bases: EnergyGridModelBase

Abstract base class for energy grid models.

n_x: int¶

n_order: int¶

Sonine_expansion: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_order]¶

sonine_indeces: jaxtyping.Int[jaxtyping.Array, StandardLaguerreEnergyGrid.n_order]¶

x: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

xWeights: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

v_norm: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L11_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L12_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L22_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L13_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L23_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L33_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L24_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L25_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L43_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L44_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L45_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶

L55_weight: jaxtyping.Float[jaxtyping.Array, StandardLaguerreEnergyGrid.n_x]¶