Index

A

AB3Stepper (class in pyqg_jax.steppers) apply_parameterization() (in module pyqg_jax.parameterizations.backscatterbiharmonic) (in module pyqg_jax.parameterizations.noop) (in module pyqg_jax.parameterizations.smagorinsky) (in module pyqg_jax.parameterizations.zannabolton2020)

apply_updates() (pyqg_jax.steppers.AB3Stepper method) (pyqg_jax.steppers.EulerStepper method)

B

BTModel (class in pyqg_jax.bt_model)

C

calc_ispec() (in module pyqg_jax.diagnostics) cfl() (in module pyqg_jax.diagnostics) create_initial_state() (pyqg_jax.bt_model.BTModel method) (pyqg_jax.parameterizations.ParameterizedModel method) (pyqg_jax.qg_model.QGModel method) (pyqg_jax.sqg_model.SQGModel method) (pyqg_jax.steppers.SteppedModel method)

D

dk (pyqg_jax.state.Grid attribute) dl (pyqg_jax.state.Grid attribute) DOUBLE (pyqg_jax.state.Precision attribute) dqdt (pyqg_jax.state.FullPseudoSpectralState attribute)

dqhdt (pyqg_jax.state.FullPseudoSpectralState attribute) dt (pyqg_jax.steppers.AB3Stepper attribute) (pyqg_jax.steppers.EulerStepper attribute) dx (pyqg_jax.state.Grid attribute) dy (pyqg_jax.state.Grid attribute)

E

EulerStepper (class in pyqg_jax.steppers)

F

FullPseudoSpectralState (class in pyqg_jax.state)

G

get_full_state() (pyqg_jax.bt_model.BTModel method) (pyqg_jax.parameterizations.ParameterizedModel method) (pyqg_jax.qg_model.QGModel method) (pyqg_jax.sqg_model.SQGModel method) (pyqg_jax.steppers.SteppedModel method) get_grid() (pyqg_jax.bt_model.BTModel method) (pyqg_jax.qg_model.QGModel method) (pyqg_jax.sqg_model.SQGModel method)

get_kappa() (pyqg_jax.state.Grid method) get_updates() (pyqg_jax.bt_model.BTModel method) (pyqg_jax.parameterizations.ParameterizedModel method) (pyqg_jax.qg_model.QGModel method) (pyqg_jax.sqg_model.SQGModel method) Grid (class in pyqg_jax.state)

H

H (pyqg_jax.state.Grid attribute)

Hi (pyqg_jax.state.Grid attribute)

I

init_param_aux_func (pyqg_jax.parameterizations.ParameterizedModel attribute) initialize_param_state() (pyqg_jax.parameterizations.ParameterizedModel method) initialize_stepper_state() (pyqg_jax.steppers.AB3Stepper method) (pyqg_jax.steppers.EulerStepper method) (pyqg_jax.steppers.SteppedModel method)

ispec_grid() (in module pyqg_jax.diagnostics)

K

ke_spec_vals() (in module pyqg_jax.diagnostics)

L

L (pyqg_jax.state.Grid attribute)

M

model (pyqg_jax.parameterizations.ParameterizedModel attribute) (pyqg_jax.steppers.SteppedModel attribute) model_state (pyqg_jax.parameterizations.ParameterizedModelState attribute) module pyqg_jax pyqg_jax.bt_model pyqg_jax.diagnostics pyqg_jax.parameterizations pyqg_jax.parameterizations.backscatterbiharmonic pyqg_jax.parameterizations.noop pyqg_jax.parameterizations.smagorinsky pyqg_jax.parameterizations.zannabolton2020 pyqg_jax.qg_model pyqg_jax.sqg_model pyqg_jax.state pyqg_jax.steppers

N

nk (pyqg_jax.state.Grid attribute) nl (pyqg_jax.state.Grid attribute) NoStepValue (class in pyqg_jax.steppers)

nx (pyqg_jax.state.Grid attribute) ny (pyqg_jax.state.Grid attribute) nz (pyqg_jax.state.Grid attribute)

P

p (pyqg_jax.state.FullPseudoSpectralState attribute) param_aux (pyqg_jax.parameterizations.ParameterizedModelState attribute) param_func (pyqg_jax.parameterizations.ParameterizedModel attribute) ParameterizedModel (class in pyqg_jax.parameterizations) ParameterizedModelState (class in pyqg_jax.parameterizations) ph (pyqg_jax.state.FullPseudoSpectralState attribute) postprocess_state() (pyqg_jax.bt_model.BTModel method) (pyqg_jax.parameterizations.ParameterizedModel method) (pyqg_jax.qg_model.QGModel method) (pyqg_jax.sqg_model.SQGModel method) Precision (class in pyqg_jax.state) PseudoSpectralState (class in pyqg_jax.state) pyqg_jax module pyqg_jax.bt_model module pyqg_jax.diagnostics module

pyqg_jax.parameterizations module pyqg_jax.parameterizations.backscatterbiharmonic module pyqg_jax.parameterizations.noop module pyqg_jax.parameterizations.smagorinsky module pyqg_jax.parameterizations.zannabolton2020 module pyqg_jax.qg_model module pyqg_jax.sqg_model module pyqg_jax.state module pyqg_jax.steppers module

Q

q (pyqg_jax.state.FullPseudoSpectralState attribute) (pyqg_jax.state.PseudoSpectralState attribute) q_parameterization() (in module pyqg_jax.parameterizations)

QGModel (class in pyqg_jax.qg_model) qh (pyqg_jax.state.FullPseudoSpectralState attribute) (pyqg_jax.state.PseudoSpectralState attribute)

R

real_state_shape (pyqg_jax.state.Grid attribute)

S

SINGLE (pyqg_jax.state.Precision attribute) spectral_state_shape (pyqg_jax.state.Grid attribute) SQGModel (class in pyqg_jax.sqg_model) state (pyqg_jax.state.FullPseudoSpectralState attribute) (pyqg_jax.steppers.StepperState attribute)

step_model() (pyqg_jax.steppers.SteppedModel method) SteppedModel (class in pyqg_jax.steppers) stepper (pyqg_jax.steppers.SteppedModel attribute) StepperState (class in pyqg_jax.steppers)

T

t (pyqg_jax.steppers.StepperState attribute)

tc (pyqg_jax.steppers.StepperState attribute) total_ke() (in module pyqg_jax.diagnostics)

U

u (pyqg_jax.state.FullPseudoSpectralState attribute) Ubg (pyqg_jax.bt_model.BTModel attribute) (pyqg_jax.qg_model.QGModel attribute) (pyqg_jax.sqg_model.SQGModel attribute)

uh (pyqg_jax.state.FullPseudoSpectralState attribute) update() (pyqg_jax.state.FullPseudoSpectralState method) (pyqg_jax.state.PseudoSpectralState method) (pyqg_jax.steppers.StepperState method) uv_parameterization() (in module pyqg_jax.parameterizations)

V

v (pyqg_jax.state.FullPseudoSpectralState attribute)

value (pyqg_jax.steppers.NoStepValue attribute) vh (pyqg_jax.state.FullPseudoSpectralState attribute)

W

W (pyqg_jax.state.Grid attribute)