Index

A

accumulate() (pyzag.nonlinear.RecursiveNonlinearEquationSolver method) adapt_damping() (pyzag.preconditioning.GaussNewtonCurvature method) adjoint_system() (pyzag.operators.base.BlockJacobian method) (pyzag.operators.dense.DenseBlockJacobian method)

apply_preconditioned_update() (in module pyzag.preconditioning) as_adjoint_walk() (pyzag.operators.base.BlockJacobian method) (pyzag.operators.dense.DenseBlockJacobian method)

B

b (pyzag.operators.dense.DensePCRState attribute) BackwardEulerODE (class in pyzag.ode) batch_lu_solve() (in module pyzag.operators.dense) batch_size (pyzag.chunktime.BidiagonalOperator property) (pyzag.operators.base.BlockJacobian property) (pyzag.operators.base.BlockOperator property) (pyzag.operators.base.BlockVector property) (pyzag.operators.dense.DenseBlockJacobian property) (pyzag.operators.dense.DenseBlockOperator property) (pyzag.operators.dense.DenseBlockVector property) begin_step() (pyzag.preconditioning.GaussNewtonCurvature method) BidiagonalForwardOperator (class in pyzag.chunktime) BidiagonalHybridFactorization() (in module pyzag.chunktime) BidiagonalHybridFactorizationImpl (class in pyzag.chunktime)

BidiagonalInverseOperator (class in pyzag.chunktime) BidiagonalOperator (class in pyzag.chunktime) BidiagonalPCRFactorization (class in pyzag.chunktime) BidiagonalThomasFactorization (class in pyzag.chunktime) block_size (pyzag.operators.base.BlockJacobian property) (pyzag.operators.base.BlockVector property) (pyzag.operators.dense.DenseBlockJacobian property) (pyzag.operators.dense.DenseBlockVector property) block_update() (pyzag.nonlinear.RecursiveNonlinearEquationSolver method) block_update_adjoint() (pyzag.nonlinear.RecursiveNonlinearEquationSolver method) BlockJacobian (class in pyzag.operators.base) BlockOperator (class in pyzag.operators.base) blocks (pyzag.stochastic.GaussianResidual attribute) BlockVector (class in pyzag.operators.base)

C

cat() (pyzag.operators.base.BlockVector class method) (pyzag.operators.dense.DenseBlockVector class method) check_optimizer_is_memoryless() (in module pyzag.preconditioning) check_optimizer_respects_gradient_scale() (in module pyzag.preconditioning) ChunkNewtonRaphson (class in pyzag.chunktime) ChunkNewtonRaphsonLineSearch (class in pyzag.chunktime) ChunkOp (class in pyzag.nonlinear)

clone() (pyzag.operators.base.BlockOperator method) (pyzag.operators.base.BlockVector method) (pyzag.operators.dense.DenseBlockOperator method) (pyzag.operators.dense.DenseBlockVector method) couple_prev_chunk() (pyzag.operators.base.BlockJacobian method) (pyzag.operators.dense.DenseBlockJacobian method) CurvatureEstimator (class in pyzag.curvature) CurvatureRescale (class in pyzag.reparametrization)

D

DenseBlockJacobian (class in pyzag.operators.dense) DenseBlockOperator (class in pyzag.operators.dense) DenseBlockVector (class in pyzag.operators.dense) DenseODEWrapper (class in pyzag.ode) DensePCRState (class in pyzag.operators.dense) device (pyzag.chunktime.BidiagonalOperator property) (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix property) (pyzag.operators.base.BlockJacobian property) (pyzag.operators.base.BlockOperator property) (pyzag.operators.base.BlockVector property) (pyzag.operators.dense.DenseBlockJacobian property) (pyzag.operators.dense.DenseBlockOperator property) (pyzag.operators.dense.DenseBlockVector property)

dtype (pyzag.chunktime.BidiagonalOperator property) (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix property) (pyzag.operators.base.BlockJacobian property) (pyzag.operators.base.BlockOperator property) (pyzag.operators.base.BlockVector property) (pyzag.operators.dense.DenseBlockJacobian property) (pyzag.operators.dense.DenseBlockOperator property) (pyzag.operators.dense.DenseBlockVector property)

E

empty() (pyzag.operators.dense.DenseBlockVector class method) estimate() (pyzag.curvature.CurvatureEstimator method) evaluate_raw() (pyzag.nonlinear.NonlinearFunctionOperatorFactory method) (pyzag.ode.BackwardEulerODE method) (pyzag.ode.ForwardEulerODE method)

ExtrapolatingPredictor (class in pyzag.nonlinear)

F

factored() (pyzag.operators.dense.DenseBlockOperator class method) flat (pyzag.stochastic.GaussianResidual attribute) flatten() (pyzag.operators.base.BlockVector method) (pyzag.operators.dense.DenseBlockVector method) flip() (pyzag.operators.base.BlockVector method) (pyzag.operators.dense.DenseBlockVector method) forward() (pyzag.chunktime.BidiagonalForwardOperator method) (pyzag.chunktime.BidiagonalInverseOperator method) (pyzag.nonlinear.RecursiveNonlinearEquationSolver method) (pyzag.reparametrization.CurvatureRescale method) (pyzag.reparametrization.RangeRescale method) (pyzag.stochastic.HierarchicalStatisticalModel method)

forward_std_dev() (pyzag.reparametrization.CurvatureRescale method) (pyzag.reparametrization.RangeRescale method) forward_system() (pyzag.operators.base.BlockJacobian method) (pyzag.operators.dense.DenseBlockJacobian method) ForwardEulerODE (class in pyzag.ode) from_diagonal() (pyzag.operators.dense.DenseBlockOperator class method) from_stacked() (pyzag.operators.dense.DenseBlockJacobian class method) FullTrajectoryPredictor (class in pyzag.nonlinear)

G

gain_ratio() (pyzag.preconditioning.GaussNewtonCurvature method) gauss_newton_rescalers() (in module pyzag.preconditioning) gaussian_map_residual() (in module pyzag.stochastic)

GaussianResidual (class in pyzag.stochastic) GaussNewtonCurvature (class in pyzag.preconditioning) GaussNewtonPreconditioner (class in pyzag.preconditioning) get_state() (pyzag.stochastic.PyroGaussNewtonOptim method)

H

H (pyzag.preconditioning.GaussNewtonCurvature property)

hierarchical_gn_diagonal() (in module pyzag.stochastic) HierarchicalStatisticalModel (class in pyzag.stochastic)

I

identity() (pyzag.operators.dense.DenseBlockOperator class method) InitialOffsetStepGenerator (class in pyzag.nonlinear)

IntegrateODE (class in pyzag.ode) inverse() (pyzag.chunktime.BidiagonalForwardOperator method)

L

lam (pyzag.preconditioning.GaussNewtonPreconditioner property) LastStepPredictor (class in pyzag.nonlinear)

lookback (pyzag.nonlinear.NonlinearFunctionOperatorFactory property) (pyzag.ode.IntegrateODE property) lu (pyzag.operators.dense.DensePCRState attribute)

M

make_operator() (pyzag.nonlinear.NonlinearFunctionOperatorFactory method) (pyzag.ode.BackwardEulerODE method) (pyzag.ode.ForwardEulerODE method) MapNormal (class in pyzag.stochastic) matvec() (pyzag.chunktime.BidiagonalForwardOperator method) (pyzag.chunktime.BidiagonalHybridFactorizationImpl method) (pyzag.chunktime.BidiagonalPCRFactorization method) (pyzag.chunktime.BidiagonalThomasFactorization method) (pyzag.operators.base.BlockOperator method) (pyzag.operators.dense.DenseBlockOperator method)

mode (pyzag.preconditioning.GaussNewtonPreconditioner property) module pyzag.chunktime pyzag.curvature pyzag.nonlinear pyzag.ode pyzag.preconditioning pyzag.reparametrization pyzag.stochastic

N

n (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix property) n_refresh_sweeps (pyzag.preconditioning.GaussNewtonPreconditioner property) n_refreshes (pyzag.preconditioning.GaussNewtonPreconditioner property) n_steps (pyzag.preconditioning.GaussNewtonPreconditioner property) nblk (pyzag.operators.base.BlockOperator property) (pyzag.operators.base.BlockVector property) (pyzag.operators.dense.DenseBlockOperator property) (pyzag.operators.dense.DenseBlockVector property) nblk_steps (pyzag.operators.base.BlockJacobian property) (pyzag.operators.dense.DenseBlockJacobian property)

neg() (pyzag.operators.base.BlockVector method) nnz (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix property) NonlinearFunctionOperator (class in pyzag.nonlinear) NonlinearFunctionOperatorFactory (class in pyzag.nonlinear) norm() (pyzag.operators.base.BlockVector method) (pyzag.operators.dense.DenseBlockVector method) not_converged() (pyzag.chunktime.ChunkNewtonRaphson method) note_refresh() (pyzag.preconditioning.GaussNewtonCurvature method) note_step() (pyzag.preconditioning.GaussNewtonCurvature method) nparam (pyzag.curvature.CurvatureEstimator property)

O

obs_name (pyzag.stochastic.GaussianResidual attribute) obs_plate_axis() (pyzag.stochastic.GaussianResidual method)

obs_scale (pyzag.stochastic.GaussianResidual attribute) obs_shape (pyzag.stochastic.GaussianResidual attribute) ODEWrapper (class in pyzag.ode)

P

pack() (pyzag.curvature.CurvatureEstimator method) pack_grads() (pyzag.curvature.CurvatureEstimator method) pad_front() (pyzag.operators.base.BlockOperator method) (pyzag.operators.dense.DenseBlockOperator method) parameters (pyzag.preconditioning.GaussNewtonPreconditioner property) pcr_finalize() (pyzag.operators.base.SolvableBlockOperator method) (pyzag.operators.dense.DenseBlockOperator method) pcr_init() (pyzag.operators.base.SolvableBlockOperator method) (pyzag.operators.dense.DenseBlockOperator method) pcr_reduce_level() (pyzag.operators.base.SolvableBlockOperator method) (pyzag.operators.dense.DenseBlockOperator method) PCRState (class in pyzag.operators.base) pivots (pyzag.operators.dense.DensePCRState attribute) plates (pyzag.stochastic.GaussianResidual attribute) precondition() (pyzag.preconditioning.GaussNewtonCurvature method) PreconditionedSVI (class in pyzag.stochastic) predict() (pyzag.nonlinear.ExtrapolatingPredictor method) (pyzag.nonlinear.FullTrajectoryPredictor method) (pyzag.nonlinear.LastStepPredictor method) (pyzag.nonlinear.PreviousStepsPredictor method) (pyzag.nonlinear.StepExtrapolatingPredictor method) (pyzag.nonlinear.ZeroPredictor method)

PreviousStepsPredictor (class in pyzag.nonlinear) prior_flat() (pyzag.stochastic.GaussianResidual method) prior_names (pyzag.stochastic.GaussianResidual property) PyroGaussNewtonOptim (class in pyzag.stochastic) pyzag.chunktime module pyzag.curvature module pyzag.nonlinear module pyzag.ode module pyzag.preconditioning module pyzag.reparametrization module pyzag.stochastic module

R

RangeRescale (class in pyzag.reparametrization) record_loss() (pyzag.stochastic.PyroGaussNewtonOptim method) recorded_loss (pyzag.stochastic.PyroGaussNewtonOptim property) recover_from_failed_evaluation() (pyzag.stochastic.PyroGaussNewtonOptim method) RecursiveNonlinearEquationSolver (class in pyzag.nonlinear) refresh() (pyzag.preconditioning.GaussNewtonCurvature method) refresh_steps (pyzag.preconditioning.GaussNewtonPreconditioner property) reject_failed_evaluation() (pyzag.preconditioning.GaussNewtonCurvature method) Reparameterizer (class in pyzag.reparametrization)

request_refresh() (pyzag.preconditioning.GaussNewtonCurvature method) reset_anchor() (pyzag.preconditioning.GaussNewtonCurvature method) reverse() (pyzag.nonlinear.StepGenerator method) (pyzag.reparametrization.CurvatureRescale method) (pyzag.reparametrization.RangeRescale method) reverse_std_dev() (pyzag.reparametrization.CurvatureRescale method) (pyzag.reparametrization.RangeRescale method) rewind() (pyzag.nonlinear.RecursiveNonlinearEquationSolver method) rho (pyzag.preconditioning.GaussNewtonPreconditioner property)

S

scale_batches() (pyzag.operators.base.BlockVector method) (pyzag.operators.dense.DenseBlockVector method) set_H() (pyzag.preconditioning.GaussNewtonCurvature method) set_state() (pyzag.stochastic.PyroGaussNewtonOptim method) setup() (pyzag.chunktime.ChunkNewtonRaphson method) shape (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix property) should_refresh() (pyzag.preconditioning.GaussNewtonCurvature method) SolvableBlockOperator (class in pyzag.operators.base) solve() (in module pyzag.nonlinear) (pyzag.chunktime.ChunkNewtonRaphson method) (pyzag.nonlinear.RecursiveNonlinearEquationSolver method) (pyzag.operators.base.SolvableBlockOperator method) (pyzag.operators.dense.DenseBlockOperator method)

solve_adjoint() (in module pyzag.nonlinear) solve_terminal_adjoint() (pyzag.operators.base.BlockJacobian method) (pyzag.operators.dense.DenseBlockJacobian method) SquareBatchedBlockDiagonalMatrix (class in pyzag.chunktime) step() (pyzag.chunktime.ChunkNewtonRaphson method) (pyzag.chunktime.ChunkNewtonRaphsonLineSearch method) (pyzag.preconditioning.GaussNewtonPreconditioner method) (pyzag.stochastic.PreconditionedSVI method) StepExtrapolatingPredictor (class in pyzag.nonlinear) StepGenerator (class in pyzag.nonlinear)

T

t_matvec() (pyzag.operators.base.BlockOperator method) (pyzag.operators.dense.DenseBlockOperator method) theta() (pyzag.curvature.CurvatureEstimator method) thomas_solve() (in module pyzag.chunktime)

to_batched_coo() (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix method) to_dense() (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix method) to_unrolled_csr() (pyzag.chunktime.SquareBatchedBlockDiagonalMatrix method) training (pyzag.stochastic.HierarchicalStatisticalModel attribute)

U

undo_last_step() (pyzag.preconditioning.GaussNewtonCurvature method)

unwrap_vector() (pyzag.ode.DenseODEWrapper method) (pyzag.ode.ODEWrapper method)

V

v (pyzag.operators.dense.DensePCRState attribute)

vecmat() (pyzag.chunktime.BidiagonalForwardOperator method) vjp() (pyzag.curvature.CurvatureEstimator method)

W

weights_flat() (pyzag.curvature.CurvatureEstimator method) where() (pyzag.operators.base.BlockVector method) (pyzag.operators.dense.DenseBlockVector method) wrap_jacobian() (pyzag.ode.DenseODEWrapper method) (pyzag.ode.ODEWrapper method)

wrap_vector() (pyzag.ode.DenseODEWrapper method) (pyzag.ode.ODEWrapper method) wrapper (pyzag.nonlinear.NonlinearFunctionOperatorFactory property) (pyzag.ode.IntegrateODE property) write_grads() (pyzag.curvature.CurvatureEstimator method)

Z

zero_grad() (pyzag.preconditioning.GaussNewtonPreconditioner method) ZeroPredictor (class in pyzag.nonlinear)

zeros() (pyzag.operators.dense.DenseBlockVector class method) zeros_like() (pyzag.operators.base.BlockVector class method) (pyzag.operators.dense.DenseBlockVector class method)