Index

A

action() (tlm_adjoint.cached_hessian.CachedGaussNewton method) (tlm_adjoint.cached_hessian.CachedHessian method) (tlm_adjoint.hessian.GaussNewton method) (tlm_adjoint.hessian.GeneralHessian method) (tlm_adjoint.hessian.Hessian method) action_space (tlm_adjoint.block_system.Matrix property) add() (tlm_adjoint.caches.Cache method) (tlm_adjoint.caches.Caches method) add_equation() (tlm_adjoint.checkpointing.CheckpointStorage method) (tlm_adjoint.tlm_adjoint.EquationManager method) add_equation_data() (tlm_adjoint.checkpointing.CheckpointStorage method) add_forward() (tlm_adjoint.equations.DotProductRHS method) (tlm_adjoint.equations.InnerProductRHS method) (tlm_adjoint.linear_equation.RHS method) add_initial_condition() (tlm_adjoint.checkpointing.CheckpointStorage method) (tlm_adjoint.tlm_adjoint.EquationManager method) add_interface() (in module tlm_adjoint.interface) addto() (tlm_adjoint.functional.Functional method) (tlm_adjoint.jax.Vector method) (tlm_adjoint.overloaded_float.SymbolicFloat method) adj_X_type() (tlm_adjoint.equation.Equation method) adj_x_type() (tlm_adjoint.equation.Equation method) adjoint() (tlm_adjoint.equation.Equation method) adjoint_action() (tlm_adjoint.linear_equation.Matrix method) adjoint_cached() (tlm_adjoint.equation.Equation method) adjoint_derivative_action() (tlm_adjoint.equation.Equation method) (tlm_adjoint.equations.Assignment method) (tlm_adjoint.equations.Conversion method) (tlm_adjoint.equations.LinearCombination method) (tlm_adjoint.fenics.assembly.Assembly method) (tlm_adjoint.fenics.interpolation.ExprInterpolation method) (tlm_adjoint.firedrake.assignment.ExprAssignment method) (tlm_adjoint.firedrake.interpolation.ExprInterpolation method) (tlm_adjoint.linear_equation.LinearEquation method) (tlm_adjoint.linear_equation.Matrix method) (tlm_adjoint.markers.AdjointActionMarker method) (tlm_adjoint.markers.FunctionalMarker method) adjoint_has_initial_condition() (tlm_adjoint.linear_equation.Matrix method) adjoint_initial_condition_dependencies() (tlm_adjoint.equation.Equation method) adjoint_jacobian_solve() (tlm_adjoint.equation.Equation method) (tlm_adjoint.equation.ZeroAssignment method) (tlm_adjoint.equations.Assignment method) (tlm_adjoint.equations.Conversion method) (tlm_adjoint.equations.LinearCombination method) (tlm_adjoint.fenics.assembly.Assembly method) (tlm_adjoint.fenics.interpolation.ExprInterpolation method) (tlm_adjoint.fenics.solve.EquationSolver method) (tlm_adjoint.firedrake.assembly.Assembly method) (tlm_adjoint.firedrake.assignment.ExprAssignment method) (tlm_adjoint.firedrake.interpolation.ExprInterpolation method) (tlm_adjoint.firedrake.solve.EquationSolver method) (tlm_adjoint.fixed_point.FixedPointSolver method) (tlm_adjoint.instructions.GarbageCollection method) (tlm_adjoint.instructions.Instruction method) (tlm_adjoint.jax.VectorEquation method) (tlm_adjoint.linear_equation.LinearEquation method) (tlm_adjoint.markers.AdjointActionMarker method) (tlm_adjoint.markers.ControlsMarker method) (tlm_adjoint.markers.FunctionalMarker method) (tlm_adjoint.overloaded_float.FloatEquation method) (tlm_adjoint.storage.Storage method)

adjoint_solve() (tlm_adjoint.linear_equation.Matrix method) AdjointActionMarker (class in tlm_adjoint.markers) AdjointBlockRHS (class in tlm_adjoint.adjoint) AdjointEquationRHS (class in tlm_adjoint.adjoint) AdjointModelRHS (class in tlm_adjoint.adjoint) AdjointRHS (class in tlm_adjoint.adjoint) Alias (class in tlm_adjoint.alias) annotation_enabled() (in module tlm_adjoint.manager) (tlm_adjoint.tlm_adjoint.EquationManager method) apply_nullspace_transformation_lhs_left() (tlm_adjoint.block_system.BlockNullspace method) (tlm_adjoint.block_system.NoneNullspace method) (tlm_adjoint.block_system.Nullspace method) (tlm_adjoint.firedrake.block_system.ConstantNullspace method) (tlm_adjoint.firedrake.block_system.DirichletBCNullspace method) (tlm_adjoint.firedrake.block_system.UnityNullspace method) apply_nullspace_transformation_lhs_right() (tlm_adjoint.block_system.BlockNullspace method) (tlm_adjoint.block_system.NoneNullspace method) (tlm_adjoint.block_system.Nullspace method) (tlm_adjoint.firedrake.block_system.ConstantNullspace method) (tlm_adjoint.firedrake.block_system.DirichletBCNullspace method) (tlm_adjoint.firedrake.block_system.UnityNullspace method) arg_space (tlm_adjoint.block_system.Matrix property) args (tlm_adjoint.checkpoint_schedules.schedule.CheckpointAction property) assemble() (tlm_adjoint.fenics.caches.AssemblyCache method) (tlm_adjoint.firedrake.caches.AssemblyCache method) Assembly (class in tlm_adjoint.fenics.assembly) (class in tlm_adjoint.firedrake.assembly) assembly_cache() (in module tlm_adjoint.fenics.caches) (in module tlm_adjoint.firedrake.caches) AssemblyCache (class in tlm_adjoint.fenics.caches) (class in tlm_adjoint.firedrake.caches) assign() (tlm_adjoint.functional.Functional method) (tlm_adjoint.jax.Vector method) (tlm_adjoint.overloaded_float.SymbolicFloat method) Assignment (class in tlm_adjoint.equations) Axpy (class in tlm_adjoint.equations)

B

B() (tlm_adjoint.adjoint.AdjointEquationRHS method) b() (tlm_adjoint.adjoint.AdjointEquationRHS method) (tlm_adjoint.adjoint.AdjointRHS method)

B_orthonormality_test() (tlm_adjoint.block_system.Eigensolver method) BlockMatrix (class in tlm_adjoint.block_system) (class in tlm_adjoint.firedrake.block_system) BlockNullspace (class in tlm_adjoint.block_system)

C

Cache (class in tlm_adjoint.caches) CachedGaussNewton (class in tlm_adjoint.cached_hessian) CachedHessian (class in tlm_adjoint.cached_hessian) CacheRef (class in tlm_adjoint.caches) Caches (class in tlm_adjoint.caches) call_jax() (in module tlm_adjoint.jax) check_space_type() (in module tlm_adjoint.interface) check_space_types() (in module tlm_adjoint.interface) check_space_types_conjugate() (in module tlm_adjoint.interface) check_space_types_conjugate_dual() (in module tlm_adjoint.interface) check_space_types_dual() (in module tlm_adjoint.interface) checkpoint_data() (tlm_adjoint.checkpointing.CheckpointStorage method) CheckpointAction (class in tlm_adjoint.checkpoint_schedules.schedule) Checkpoints (class in tlm_adjoint.checkpointing) CheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.schedule) CheckpointStorage (class in tlm_adjoint.checkpointing) Clear (class in tlm_adjoint.checkpoint_schedules.schedule) clear() (tlm_adjoint.caches.Cache method) (tlm_adjoint.caches.CacheRef method) (tlm_adjoint.caches.Caches method) (tlm_adjoint.checkpointing.CheckpointStorage method) clear_caches() (in module tlm_adjoint.caches) clear_data (tlm_adjoint.checkpoint_schedules.schedule.Clear property) clear_ics (tlm_adjoint.checkpoint_schedules.schedule.Clear property) Cofunction (class in tlm_adjoint.firedrake.variables) comm (tlm_adjoint.jax.VectorSpace property) (tlm_adjoint.overloaded_float.FloatSpace property) (tlm_adjoint.tlm_adjoint.EquationManager property)

comm_dup_cached() (in module tlm_adjoint.interface) compute_gradient() (in module tlm_adjoint.manager) (tlm_adjoint.tlm_adjoint.EquationManager method) Configure (class in tlm_adjoint.checkpoint_schedules.schedule) configure() (tlm_adjoint.checkpointing.CheckpointStorage method) configure_checkpointing() (in module tlm_adjoint.manager) (tlm_adjoint.tlm_adjoint.EquationManager method) configure_tlm() (in module tlm_adjoint.manager) (tlm_adjoint.tlm_adjoint.EquationManager method) conjugate_dual_space_type() (in module tlm_adjoint.interface) conjugate_space_type() (in module tlm_adjoint.interface) Constant (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables) ConstantNullspace (class in tlm_adjoint.firedrake.block_system) constraint_correct_lhs() (tlm_adjoint.block_system.BlockNullspace method) (tlm_adjoint.block_system.NoneNullspace method) (tlm_adjoint.block_system.Nullspace method) (tlm_adjoint.firedrake.block_system.ConstantNullspace method) (tlm_adjoint.firedrake.block_system.DirichletBCNullspace method) (tlm_adjoint.firedrake.block_system.UnityNullspace method) ControlsMarker (class in tlm_adjoint.markers) Conversion (class in tlm_adjoint.equations) correct_rhs() (tlm_adjoint.block_system.Nullspace method) correct_soln() (tlm_adjoint.block_system.Nullspace method) CustomNormSq (class in tlm_adjoint.fixed_point)

D

delete (tlm_adjoint.checkpoint_schedules.schedule.Read property) delete() (tlm_adjoint.checkpointing.Checkpoints method) (tlm_adjoint.checkpointing.HDF5Checkpoints method) (tlm_adjoint.checkpointing.PickleCheckpoints method) dependencies() (tlm_adjoint.equation.Equation method) (tlm_adjoint.linear_equation.RHS method) DirichletBCNullspace (class in tlm_adjoint.firedrake.block_system) dM (tlm_adjoint.tangent_linear.TangentLinearMap property) DotProduct (class in tlm_adjoint.equations) DotProductRHS (class in tlm_adjoint.equations) drop_references() (tlm_adjoint.equation.Equation method) (tlm_adjoint.equations.DotProductRHS method) (tlm_adjoint.equations.InnerProductRHS method) (tlm_adjoint.fenics.assembly.Assembly method) (tlm_adjoint.fenics.interpolation.ExprInterpolation method) (tlm_adjoint.fenics.solve.EquationSolver method) (tlm_adjoint.firedrake.assembly.Assembly method) (tlm_adjoint.firedrake.assignment.ExprAssignment method) (tlm_adjoint.firedrake.interpolation.ExprInterpolation method) (tlm_adjoint.firedrake.solve.EquationSolver method) (tlm_adjoint.fixed_point.FixedPointSolver method) (tlm_adjoint.linear_equation.LinearEquation method) (tlm_adjoint.linear_equation.Matrix method) (tlm_adjoint.linear_equation.RHS method) (tlm_adjoint.tlm_adjoint.EquationManager method)

dtype (tlm_adjoint.jax.VectorSpace property) (tlm_adjoint.overloaded_float.FloatSpace property) dual_sample() (tlm_adjoint.firedrake.block_system.WhiteNoiseSampler method) dual_space_type() (in module tlm_adjoint.interface)

E

eigenpairs() (tlm_adjoint.block_system.Eigensolver method) Eigensolver (class in tlm_adjoint.block_system) eigenvalues() (tlm_adjoint.block_system.Eigensolver method) eliminate_zeros() (in module tlm_adjoint.fenics.expr) (in module tlm_adjoint.firedrake.expr) EmptyEquation (class in tlm_adjoint.equations) EndForward (class in tlm_adjoint.checkpoint_schedules.schedule) EndReverse (class in tlm_adjoint.checkpoint_schedules.schedule) eps (tlm_adjoint.block_system.Eigensolver property)

equals() (tlm_adjoint.firedrake.variables.Cofunction method) Equation (class in tlm_adjoint.equation) EquationManager (class in tlm_adjoint.tlm_adjoint) EquationSolver (class in tlm_adjoint.fenics.solve) (class in tlm_adjoint.firedrake.solve) exhausted (tlm_adjoint.checkpoint_schedules.schedule.EndReverse property) ExprAssignment (class in tlm_adjoint.firedrake.assignment) ExprInterpolation (class in tlm_adjoint.fenics.interpolation) (class in tlm_adjoint.firedrake.interpolation)

F

finalize() (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule method) (tlm_adjoint.tlm_adjoint.EquationManager method) FixedPointSolver (class in tlm_adjoint.fixed_point) flattened_space (tlm_adjoint.block_system.MixedSpace property) Float (class in tlm_adjoint.overloaded_float) float_cls (tlm_adjoint.overloaded_float.FloatSpace property) FloatEquation (class in tlm_adjoint.overloaded_float) FloatSpace (class in tlm_adjoint.overloaded_float) form_matrix() (in module tlm_adjoint.firedrake.block_system) Forward (class in tlm_adjoint.checkpoint_schedules.schedule) forward() (tlm_adjoint.equation.Equation method) forward_action() (tlm_adjoint.linear_equation.Matrix method) forward_solve() (tlm_adjoint.equation.Equation method) (tlm_adjoint.equation.ZeroAssignment method) (tlm_adjoint.equations.Assignment method) (tlm_adjoint.equations.Conversion method) (tlm_adjoint.equations.EmptyEquation method) (tlm_adjoint.equations.LinearCombination method) (tlm_adjoint.fenics.assembly.Assembly method) (tlm_adjoint.fenics.interpolation.ExprInterpolation method) (tlm_adjoint.fenics.solve.EquationSolver method) (tlm_adjoint.firedrake.assembly.Assembly method) (tlm_adjoint.firedrake.assignment.ExprAssignment method) (tlm_adjoint.firedrake.interpolation.ExprInterpolation method) (tlm_adjoint.firedrake.solve.EquationSolver method) (tlm_adjoint.fixed_point.FixedPointSolver method) (tlm_adjoint.instructions.GarbageCollection method) (tlm_adjoint.jax.VectorEquation method) (tlm_adjoint.linear_equation.LinearEquation method) (tlm_adjoint.linear_equation.Matrix method) (tlm_adjoint.markers.AdjointActionMarker method) (tlm_adjoint.overloaded_float.FloatEquation method) (tlm_adjoint.storage.Storage method)

from_torch_tensors() (in module tlm_adjoint.torch) Function (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables) function_tlm() (tlm_adjoint.tlm_adjoint.EquationManager method) Functional (class in tlm_adjoint.functional) FunctionalMarker (class in tlm_adjoint.markers)

G

garbage_cleanup() (in module tlm_adjoint.interface) GarbageCollection (class in tlm_adjoint.instructions) GaussNewton (class in tlm_adjoint.hessian) gc_disabled() (in module tlm_adjoint.alias)

GeneralGaussNewton (class in tlm_adjoint.hessian) GeneralHessian (class in tlm_adjoint.hessian) get() (tlm_adjoint.caches.Cache method) global_size (tlm_adjoint.jax.VectorSpace property)

H

has_initial_condition() (tlm_adjoint.linear_equation.Matrix method) HDF5Checkpoints (class in tlm_adjoint.checkpointing) HDF5Storage (class in tlm_adjoint.storage)

Hessian (class in tlm_adjoint.hessian) HessianEigensolver (class in tlm_adjoint.hessian_system) HessianLinearSolver (class in tlm_adjoint.hessian_system) HRevolveCheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.h_revolve)

I

id (tlm_adjoint.caches.Cache property) (tlm_adjoint.tangent_linear.TangentLinearMap property) info() (tlm_adjoint.tlm_adjoint.EquationManager method) initial_condition() (tlm_adjoint.checkpointing.CheckpointStorage method) initial_condition_dependencies() (tlm_adjoint.equation.Equation method) initial_conditions() (tlm_adjoint.checkpointing.CheckpointStorage method) InnerProduct (class in tlm_adjoint.equations) InnerProductRHS (class in tlm_adjoint.equations) Instruction (class in tlm_adjoint.instructions) is_active() (tlm_adjoint.checkpointing.ReplayStorage method) is_empty() (tlm_adjoint.adjoint.AdjointBlockRHS method) (tlm_adjoint.adjoint.AdjointModelRHS method) is_exhausted (tlm_adjoint.checkpoint_schedules.binomial.MultistageCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.binomial.TwoLevelCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.h_revolve.HRevolveCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.memory.MemoryCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.mixed.MixedCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.none.NoneCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.periodic.PeriodicDiskCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule property)

is_hermitian_and_positive() (tlm_adjoint.block_system.Eigensolver method) is_running (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule property) is_saved() (tlm_adjoint.storage.HDF5Storage method) (tlm_adjoint.storage.MemoryStorage method) (tlm_adjoint.storage.Storage method) is_space() (in module tlm_adjoint.interface) is_var() (in module tlm_adjoint.interface) iter() (tlm_adjoint.checkpoint_schedules.binomial.MultistageCheckpointSchedule method) (tlm_adjoint.checkpoint_schedules.binomial.TwoLevelCheckpointSchedule method) (tlm_adjoint.checkpoint_schedules.h_revolve.HRevolveCheckpointSchedule method) (tlm_adjoint.checkpoint_schedules.memory.MemoryCheckpointSchedule method) (tlm_adjoint.checkpoint_schedules.mixed.MixedCheckpointSchedule method) (tlm_adjoint.checkpoint_schedules.none.NoneCheckpointSchedule method) (tlm_adjoint.checkpoint_schedules.periodic.PeriodicDiskCheckpointSchedule method) (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule method)

K

key (tlm_adjoint.storage.Storage property)

ksp (tlm_adjoint.block_system.LinearSolver property)

L

linear_solver() (in module tlm_adjoint.fenics.backend_interface) (in module tlm_adjoint.firedrake.backend_interface) (tlm_adjoint.fenics.caches.LinearSolverCache method) (tlm_adjoint.firedrake.caches.LinearSolverCache method) linear_solver_cache() (in module tlm_adjoint.fenics.caches) (in module tlm_adjoint.firedrake.caches) LinearCombination (class in tlm_adjoint.equations) LinearEquation (class in tlm_adjoint.linear_equation) LinearSolver (class in tlm_adjoint.block_system) (class in tlm_adjoint.firedrake.block_system) LinearSolverCache (class in tlm_adjoint.fenics.caches) (class in tlm_adjoint.firedrake.caches) load() (tlm_adjoint.storage.HDF5Storage method) (tlm_adjoint.storage.MemoryStorage method) (tlm_adjoint.storage.Storage method)

local_caches() (in module tlm_adjoint.caches) local_size (tlm_adjoint.jax.VectorSpace property) local_solver() (tlm_adjoint.fenics.caches.LocalSolverCache method) (tlm_adjoint.firedrake.caches.LocalSolverCache method) local_solver_cache() (in module tlm_adjoint.fenics.caches) (in module tlm_adjoint.firedrake.caches) LocalEquationSolver (class in tlm_adjoint.fenics.solve) (class in tlm_adjoint.firedrake.solve) LocalProjection (class in tlm_adjoint.fenics.projection) (class in tlm_adjoint.firedrake.projection) LocalSolverCache (class in tlm_adjoint.fenics.caches) (class in tlm_adjoint.firedrake.caches)

M

M (tlm_adjoint.tangent_linear.TangentLinearMap property) manager() (in module tlm_adjoint.manager) manager_disabled() (in module tlm_adjoint.manager) manager_info() (in module tlm_adjoint.manager) Matrix (class in tlm_adjoint.block_system) (class in tlm_adjoint.linear_equation) MatrixFreeMatrix (class in tlm_adjoint.block_system) MatrixFunctionSolver (class in tlm_adjoint.block_system) max_n (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule property) MemoryCheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.memory) MemoryStorage (class in tlm_adjoint.storage) mfn (tlm_adjoint.block_system.MatrixFunctionSolver property) minimize_scipy() (in module tlm_adjoint.optimization) minimize_tao() (in module tlm_adjoint.optimization) MixedCheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.mixed) MixedSpace (class in tlm_adjoint.block_system) module tlm_adjoint tlm_adjoint.adjoint tlm_adjoint.alias tlm_adjoint.block_system tlm_adjoint.cached_hessian tlm_adjoint.caches tlm_adjoint.checkpoint_schedules.binomial tlm_adjoint.checkpoint_schedules.checkpoint_schedules tlm_adjoint.checkpoint_schedules.h_revolve tlm_adjoint.checkpoint_schedules.memory tlm_adjoint.checkpoint_schedules.mixed tlm_adjoint.checkpoint_schedules.none tlm_adjoint.checkpoint_schedules.periodic tlm_adjoint.checkpoint_schedules.schedule tlm_adjoint.checkpointing tlm_adjoint.equation tlm_adjoint.equations tlm_adjoint.fenics.assembly tlm_adjoint.fenics.backend_interface tlm_adjoint.fenics.caches tlm_adjoint.fenics.expr tlm_adjoint.fenics.interpolation tlm_adjoint.fenics.projection tlm_adjoint.fenics.solve tlm_adjoint.fenics.variables tlm_adjoint.firedrake.assembly tlm_adjoint.firedrake.assignment tlm_adjoint.firedrake.backend_interface tlm_adjoint.firedrake.block_system tlm_adjoint.firedrake.caches tlm_adjoint.firedrake.expr tlm_adjoint.firedrake.interpolation tlm_adjoint.firedrake.projection tlm_adjoint.firedrake.solve tlm_adjoint.firedrake.variables tlm_adjoint.fixed_point tlm_adjoint.functional tlm_adjoint.hessian tlm_adjoint.hessian_system tlm_adjoint.instructions tlm_adjoint.interface tlm_adjoint.jax tlm_adjoint.linear_equation tlm_adjoint.manager tlm_adjoint.markers tlm_adjoint.optimization tlm_adjoint.overloaded_float tlm_adjoint.storage tlm_adjoint.tangent_linear tlm_adjoint.tlm_adjoint tlm_adjoint.torch tlm_adjoint.verification

mult() (tlm_adjoint.block_system.Matrix method) (tlm_adjoint.block_system.MatrixFreeMatrix method) mult_add() (tlm_adjoint.block_system.BlockMatrix method) (tlm_adjoint.block_system.Matrix method) (tlm_adjoint.firedrake.block_system.PETScMatrix method) MultistageCheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.binomial)

N

n (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.schedule.Read property) (tlm_adjoint.checkpoint_schedules.schedule.Write property) n0 (tlm_adjoint.checkpoint_schedules.schedule.Forward property) (tlm_adjoint.checkpoint_schedules.schedule.Reverse property) n1 (tlm_adjoint.checkpoint_schedules.schedule.Forward property) (tlm_adjoint.checkpoint_schedules.schedule.Reverse property) name (tlm_adjoint.jax.Vector property) new() (tlm_adjoint.block_system.MixedSpace method) (tlm_adjoint.block_system.TypedSpace method) (tlm_adjoint.jax.Vector method) (tlm_adjoint.overloaded_float.SymbolicFloat method) (tlm_adjoint.tlm_adjoint.EquationManager method) new_adj_X() (tlm_adjoint.equation.Equation method)

new_adj_x() (tlm_adjoint.equation.Equation method) new_block() (in module tlm_adjoint.manager) (tlm_adjoint.tlm_adjoint.EquationManager method) new_jax() (in module tlm_adjoint.jax) new_jax_float() (in module tlm_adjoint.jax) no_float_overloading() (in module tlm_adjoint.overloaded_float) no_space_type_checking() (in module tlm_adjoint.interface) NoneCheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.none) NoneNullspace (class in tlm_adjoint.block_system) nonlinear_dependencies() (tlm_adjoint.equation.Equation method) (tlm_adjoint.linear_equation.Matrix method) (tlm_adjoint.linear_equation.RHS method) NullSolver (class in tlm_adjoint.equation) Nullspace (class in tlm_adjoint.block_system)

O

OverloadedFloat (class in tlm_adjoint.overloaded_float)

ownership_range (tlm_adjoint.jax.VectorSpace property)

P

paused() (tlm_adjoint.tlm_adjoint.EquationManager method) paused_float_overloading() (in module tlm_adjoint.overloaded_float) paused_manager() (in module tlm_adjoint.manager) paused_space_type_checking() (in module tlm_adjoint.interface) pc_constraint_correct_soln() (tlm_adjoint.block_system.BlockNullspace method) (tlm_adjoint.block_system.NoneNullspace method) (tlm_adjoint.block_system.Nullspace method) (tlm_adjoint.firedrake.block_system.ConstantNullspace method) (tlm_adjoint.firedrake.block_system.DirichletBCNullspace method) (tlm_adjoint.firedrake.block_system.UnityNullspace method) pc_post_mult_correct() (tlm_adjoint.block_system.Nullspace method)

pc_pre_mult_correct() (tlm_adjoint.block_system.Nullspace method) PeriodicDiskCheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.periodic) PETScMatrix (class in tlm_adjoint.firedrake.block_system) PickleCheckpoints (class in tlm_adjoint.checkpointing) pop() (tlm_adjoint.adjoint.AdjointBlockRHS method) (tlm_adjoint.adjoint.AdjointModelRHS method) popleft() (tlm_adjoint.checkpointing.ReplayStorage method) post_mult_correct_lhs() (tlm_adjoint.block_system.Nullspace method) pre_mult_correct_lhs() (tlm_adjoint.block_system.Nullspace method) Projection (class in tlm_adjoint.fenics.projection) (class in tlm_adjoint.firedrake.projection)

R

r (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule property) rdtype() (tlm_adjoint.jax.VectorSpace method) (tlm_adjoint.overloaded_float.FloatSpace method) Read (class in tlm_adjoint.checkpoint_schedules.schedule) read() (tlm_adjoint.checkpointing.Checkpoints method) (tlm_adjoint.checkpointing.HDF5Checkpoints method) (tlm_adjoint.checkpointing.PickleCheckpoints method) relative_space_type() (in module tlm_adjoint.interface) remove() (tlm_adjoint.caches.Caches method) Replacement (class in tlm_adjoint.fenics.expr) (class in tlm_adjoint.firedrake.expr) ReplacementCofunction (class in tlm_adjoint.firedrake.variables) ReplacementConstant (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables)

ReplacementFunction (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables) ReplacementZeroConstant (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables) ReplacementZeroFunction (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables) ReplayStorage (class in tlm_adjoint.checkpointing) reset() (tlm_adjoint.tlm_adjoint.EquationManager method) reset_manager() (in module tlm_adjoint.manager) restore_manager() (in module tlm_adjoint.manager) Reverse (class in tlm_adjoint.checkpoint_schedules.schedule) RHS (class in tlm_adjoint.linear_equation)

S

sample() (tlm_adjoint.firedrake.block_system.WhiteNoiseSampler method) save() (tlm_adjoint.storage.HDF5Storage method) (tlm_adjoint.storage.MemoryStorage method) (tlm_adjoint.storage.Storage method) set_assembly_cache() (in module tlm_adjoint.fenics.caches) (in module tlm_adjoint.firedrake.caches) set_default_float_dtype() (in module tlm_adjoint.overloaded_float) set_default_jax_dtype() (in module tlm_adjoint.jax) set_linear_solver_cache() (in module tlm_adjoint.fenics.caches) (in module tlm_adjoint.firedrake.caches) set_local_solver_cache() (in module tlm_adjoint.fenics.caches) (in module tlm_adjoint.firedrake.caches) set_manager() (in module tlm_adjoint.manager) solve() (tlm_adjoint.block_system.Eigensolver method) (tlm_adjoint.block_system.LinearSolver method) (tlm_adjoint.block_system.MatrixFunctionSolver method) (tlm_adjoint.equation.Equation method) (tlm_adjoint.hessian_system.HessianLinearSolver method) (tlm_adjoint.jax.VectorEquation method) (tlm_adjoint.optimization.TAOSolver method) space (tlm_adjoint.jax.Vector property) space_comm() (in module tlm_adjoint.interface) space_default_space_type() (in module tlm_adjoint.interface) space_dtype() (in module tlm_adjoint.interface) space_eq() (in module tlm_adjoint.interface) space_global_size() (in module tlm_adjoint.interface) space_id() (in module tlm_adjoint.interface) space_local_indices() (in module tlm_adjoint.interface) space_local_size() (in module tlm_adjoint.interface) space_new() (in module tlm_adjoint.interface)

space_type (tlm_adjoint.jax.Vector property) SpaceInterface (class in tlm_adjoint.interface) SpaceTypeError spectral_approximation_solve() (tlm_adjoint.hessian_system.HessianEigensolver method) spectral_pc_fn() (tlm_adjoint.hessian_system.HessianEigensolver method) split_space (tlm_adjoint.block_system.MixedSpace property) start() (tlm_adjoint.tlm_adjoint.EquationManager method) start_manager() (in module tlm_adjoint.manager) stop() (tlm_adjoint.tlm_adjoint.EquationManager method) stop_manager() (in module tlm_adjoint.manager) Storage (class in tlm_adjoint.storage) storage (tlm_adjoint.checkpoint_schedules.schedule.Read property) (tlm_adjoint.checkpoint_schedules.schedule.Write property) store_data (tlm_adjoint.checkpoint_schedules.schedule.Configure property) (tlm_adjoint.checkpointing.CheckpointStorage property) store_ics (tlm_adjoint.checkpoint_schedules.schedule.Configure property) (tlm_adjoint.checkpointing.CheckpointStorage property) sub() (tlm_adjoint.adjoint.AdjointRHS method) subtract_adjoint_derivative_action() (in module tlm_adjoint.interface) (tlm_adjoint.equations.DotProductRHS method) (tlm_adjoint.equations.InnerProductRHS method) (tlm_adjoint.linear_equation.RHS method) subtract_adjoint_derivative_actions() (tlm_adjoint.equation.Equation method) (tlm_adjoint.fenics.solve.EquationSolver method) (tlm_adjoint.firedrake.assembly.Assembly method) (tlm_adjoint.firedrake.solve.EquationSolver method) (tlm_adjoint.fixed_point.FixedPointSolver method) (tlm_adjoint.jax.VectorEquation method) (tlm_adjoint.overloaded_float.FloatEquation method) SymbolicFloat (class in tlm_adjoint.overloaded_float)

T

tangent_linear() (tlm_adjoint.equation.Equation method) (tlm_adjoint.equation.ZeroAssignment method) (tlm_adjoint.equations.Assignment method) (tlm_adjoint.equations.Conversion method) (tlm_adjoint.equations.LinearCombination method) (tlm_adjoint.fenics.assembly.Assembly method) (tlm_adjoint.fenics.interpolation.ExprInterpolation method) (tlm_adjoint.fenics.solve.EquationSolver method) (tlm_adjoint.fenics.solve.LocalEquationSolver method) (tlm_adjoint.firedrake.assembly.Assembly method) (tlm_adjoint.firedrake.assignment.ExprAssignment method) (tlm_adjoint.firedrake.interpolation.ExprInterpolation method) (tlm_adjoint.firedrake.solve.EquationSolver method) (tlm_adjoint.firedrake.solve.LocalEquationSolver method) (tlm_adjoint.fixed_point.FixedPointSolver method) (tlm_adjoint.instructions.GarbageCollection method) (tlm_adjoint.jax.VectorEquation method) (tlm_adjoint.linear_equation.LinearEquation method) (tlm_adjoint.markers.AdjointActionMarker method) (tlm_adjoint.overloaded_float.FloatEquation method) (tlm_adjoint.storage.Storage method) tangent_linear_rhs() (tlm_adjoint.equations.DotProductRHS method) (tlm_adjoint.equations.InnerProductRHS method) (tlm_adjoint.linear_equation.Matrix method) (tlm_adjoint.linear_equation.RHS method) TangentLinearMap (class in tlm_adjoint.tangent_linear) tao (tlm_adjoint.optimization.TAOSolver property) TAOSolver (class in tlm_adjoint.optimization) taylor_test() (in module tlm_adjoint.verification) taylor_test_tlm() (in module tlm_adjoint.verification) taylor_test_tlm_adjoint() (in module tlm_adjoint.verification) tlm_adjoint module tlm_adjoint.adjoint module tlm_adjoint.alias module tlm_adjoint.block_system module tlm_adjoint.cached_hessian module tlm_adjoint.caches module tlm_adjoint.checkpoint_schedules.binomial module tlm_adjoint.checkpoint_schedules.checkpoint_schedules module tlm_adjoint.checkpoint_schedules.h_revolve module tlm_adjoint.checkpoint_schedules.memory module tlm_adjoint.checkpoint_schedules.mixed module tlm_adjoint.checkpoint_schedules.none module tlm_adjoint.checkpoint_schedules.periodic module tlm_adjoint.checkpoint_schedules.schedule module tlm_adjoint.checkpointing module tlm_adjoint.equation module tlm_adjoint.equations module tlm_adjoint.fenics.assembly module tlm_adjoint.fenics.backend_interface module tlm_adjoint.fenics.caches module tlm_adjoint.fenics.expr module

tlm_adjoint.fenics.interpolation module tlm_adjoint.fenics.projection module tlm_adjoint.fenics.solve module tlm_adjoint.fenics.variables module tlm_adjoint.firedrake.assembly module tlm_adjoint.firedrake.assignment module tlm_adjoint.firedrake.backend_interface module tlm_adjoint.firedrake.block_system module tlm_adjoint.firedrake.caches module tlm_adjoint.firedrake.expr module tlm_adjoint.firedrake.interpolation module tlm_adjoint.firedrake.projection module tlm_adjoint.firedrake.solve module tlm_adjoint.firedrake.variables module tlm_adjoint.fixed_point module tlm_adjoint.functional module tlm_adjoint.hessian module tlm_adjoint.hessian_system module tlm_adjoint.instructions module tlm_adjoint.interface module tlm_adjoint.jax module tlm_adjoint.linear_equation module tlm_adjoint.manager module tlm_adjoint.markers module tlm_adjoint.optimization module tlm_adjoint.overloaded_float module tlm_adjoint.storage module tlm_adjoint.tangent_linear module tlm_adjoint.tlm_adjoint module tlm_adjoint.torch module tlm_adjoint.verification module tlm_enabled() (in module tlm_adjoint.manager) (tlm_adjoint.tlm_adjoint.EquationManager method) to_fenics() (in module tlm_adjoint.fenics.variables) to_firedrake() (in module tlm_adjoint.firedrake.variables) to_float() (in module tlm_adjoint.overloaded_float) to_jax() (in module tlm_adjoint.jax) to_torch_tensors() (in module tlm_adjoint.torch) torch_wrapped() (in module tlm_adjoint.torch) tuple_sub() (tlm_adjoint.block_system.MixedSpace method) TwoLevelCheckpointSchedule (class in tlm_adjoint.checkpoint_schedules.binomial) TypedSpace (class in tlm_adjoint.block_system)

U

UnityNullspace (class in tlm_adjoint.firedrake.block_system) update() (tlm_adjoint.caches.Caches method) (tlm_adjoint.checkpointing.CheckpointStorage method) (tlm_adjoint.checkpointing.ReplayStorage method) update_keys() (tlm_adjoint.checkpointing.CheckpointStorage method) uses_disk_storage (tlm_adjoint.checkpoint_schedules.binomial.MultistageCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.binomial.TwoLevelCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.h_revolve.HRevolveCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.memory.MemoryCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.mixed.MixedCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.none.NoneCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.periodic.PeriodicDiskCheckpointSchedule property) (tlm_adjoint.checkpoint_schedules.schedule.CheckpointSchedule property)

V

value (tlm_adjoint.jax.Vector property) (tlm_adjoint.overloaded_float.SymbolicFloat property) var_assign() (in module tlm_adjoint.interface) var_axpy() (in module tlm_adjoint.interface) var_caches() (in module tlm_adjoint.interface) var_comm() (in module tlm_adjoint.interface) var_copy() (in module tlm_adjoint.interface) var_dtype() (in module tlm_adjoint.interface) var_from_petsc() (in module tlm_adjoint.interface) var_get_values() (in module tlm_adjoint.interface) var_global_size() (in module tlm_adjoint.interface) var_id() (in module tlm_adjoint.interface) var_inner() (in module tlm_adjoint.interface) var_is_alias() (in module tlm_adjoint.interface) var_is_cached() (in module tlm_adjoint.interface) var_is_replacement() (in module tlm_adjoint.interface) var_is_scalar() (in module tlm_adjoint.interface) var_is_static() (in module tlm_adjoint.interface) var_linf_norm() (in module tlm_adjoint.interface) var_local_indices() (in module tlm_adjoint.interface) var_local_size() (in module tlm_adjoint.interface) var_lock_state() (in module tlm_adjoint.interface) var_locked() (in module tlm_adjoint.interface)

var_name() (in module tlm_adjoint.interface) var_new() (in module tlm_adjoint.interface) var_new_conjugate() (in module tlm_adjoint.interface) var_new_conjugate_dual() (in module tlm_adjoint.interface) var_new_dual() (in module tlm_adjoint.interface) var_replacement() (in module tlm_adjoint.interface) var_scalar_value() (in module tlm_adjoint.interface) var_set_values() (in module tlm_adjoint.interface) var_space() (in module tlm_adjoint.interface) var_space_type() (in module tlm_adjoint.interface) var_state() (in module tlm_adjoint.interface) var_tlm() (in module tlm_adjoint.manager) (tlm_adjoint.tlm_adjoint.EquationManager method) var_to_petsc() (in module tlm_adjoint.interface) var_update_caches() (in module tlm_adjoint.interface) var_update_state() (in module tlm_adjoint.interface) var_zero() (in module tlm_adjoint.interface) VariableInterface (class in tlm_adjoint.interface) VariableStateLockDictionary (class in tlm_adjoint.interface) Vector (class in tlm_adjoint.jax) vector (tlm_adjoint.jax.Vector property) VectorEquation (class in tlm_adjoint.jax) VectorSpace (class in tlm_adjoint.jax)

W

WeakAlias (class in tlm_adjoint.alias) WhiteNoiseSampler (class in tlm_adjoint.firedrake.block_system) Write (class in tlm_adjoint.checkpoint_schedules.schedule)

write() (tlm_adjoint.checkpointing.Checkpoints method) (tlm_adjoint.checkpointing.HDF5Checkpoints method) (tlm_adjoint.checkpointing.PickleCheckpoints method)

X

X() (tlm_adjoint.equation.Equation method)

x() (tlm_adjoint.equation.Equation method)

Z

Zero (class in tlm_adjoint.fenics.expr) (class in tlm_adjoint.firedrake.expr) ZeroAssignment (class in tlm_adjoint.equation)

ZeroConstant (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables) ZeroFunction (class in tlm_adjoint.fenics.variables) (class in tlm_adjoint.firedrake.variables)