Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | Z A A (neml2.data.CrystalGeometry property) A() (neml2.es.LinearSystem method) A_and_B() (neml2.es.LinearSystem method) A_and_b() (neml2.es.LinearSystem method) A_and_B_and_b() (neml2.es.LinearSystem method) abs() (in module neml2.types) add_node() (neml2.models.resolver.DependencyResolver method) align_sub_batch() (in module neml2.types) allow_nonlinear (neml2.schema.HitField attribute) AOTIModel (class in neml2.aoti) apply_chain_rule() (neml2.models.model.Model method) apply_chain_rule_2() (neml2.models.model.Model method) arange() (neml2.types.Scalar class method) as_typed() (neml2.types.Tensor method) assemble() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) (neml2.es.SparseMatrix method) (neml2.es.SparseVector method) AssembledMatrix (class in neml2.es) AssembledVector (class in neml2.es) attr (neml2.schema.HitField attribute) AxisLayout (class in neml2.es) B b() (neml2.es.LinearSystem method) base (neml2.types.Tensor property) (neml2.types.TensorWrapper property) BASE_NDIM (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) base_ndim (neml2.types.Tensor property) BASE_NDIM (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) BASE_SHAPE (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) base_shape (neml2.types.Tensor property) BASE_SHAPE (neml2.types.TensorWrapper attribute) base_shape (neml2.types.TensorWrapper property) BASE_SHAPE (neml2.types.Vec attribute) (neml2.types.WR2 attribute) batch (neml2.types.Tensor property) (neml2.types.TensorWrapper property) batch_ndim (neml2.types.Tensor attribute) batch_shape (neml2.types.Tensor property) (neml2.types.TensorWrapper property) bilinear_interpolation() (in module neml2.types) bilinear_interpolation_slopes() (in module neml2.types) blayout (neml2.es.LinearSystem property) BLOCK_NAME (in module neml2.schema) block_size() (neml2.es.AxisLayout method) C call_by_name() (neml2.models.model.Model method) cat() (in module neml2.types) cells (neml2.es.SparseMatrix attribute) ChainRuleAction (in module neml2.models.chain_rule) (in module neml2.models.model) ChainRuleDict (in module neml2.models.chain_rule) (in module neml2.models.model) change_lag_order() (in module neml2.pyzag) clamp() (in module neml2.types) col_layout (neml2.es.AssembledMatrix attribute) (neml2.es.SparseMatrix attribute) compile_model() (in module neml2.models.export) compose() (in module neml2.types) consumed_items (neml2.models.model.Model property) ConvergenceError cosh() (in module neml2.types) CrystalGeometry (class in neml2.data) CSVScalar (class in neml2.user_tensors) CSVSR2 (class in neml2.user_tensors) CSVVec (class in neml2.user_tensors) CSVWR2 (class in neml2.user_tensors) ctor_name (neml2.schema.HitField property) cubic_symmetry_operators() (in module neml2.data) CubicCrystal (class in neml2.data) D data (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) (neml2.types.Tensor attribute) (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) declare_typed_buffer() (neml2.models.model.Model method) declare_typed_parameter() (neml2.models.model.Model method) default (neml2.schema.HitField attribute) DenseLU (class in neml2.solvers) dependency() (in module neml2.schema) DependencyResolver (class in neml2.models.resolver) derived_input() (in module neml2.schema) derived_output() (in module neml2.schema) det() (in module neml2.types) dev() (in module neml2.types) device (neml2.aoti.Model property) (neml2.types.Tensor property) (neml2.types.TensorWrapper property) dexp_map() (in module neml2.types) diff() (in module neml2.types) disassemble() (neml2.es.AssembledMatrix method) (neml2.es.AssembledVector method) doc (neml2.schema.HitField attribute) drotate() (in module neml2.types) drotate_self() (in module neml2.types) dtype (neml2.aoti.Model property) (neml2.types.Tensor property) (neml2.types.TensorWrapper property) dynamic_batch (neml2.types.TensorWrapper property) dynamic_batch_shape (neml2.types.TensorWrapper property) E emitted_pairs() (neml2.es.IFT method) empty() (neml2.types.PrimitiveTensor class method) equalize_tangent_K() (in module neml2.models.chain_rule) euler_rodrigues() (in module neml2.types) exp() (in module neml2.types) exp_map() (in module neml2.types) export_model_for_aoti() (in module neml2.cli) F factory_getter (neml2.schema.HitField attribute) FAILURE (neml2.solvers.RetCode attribute) fill() (neml2.types.PrimitiveTensor class method) (neml2.types.SR2 class method) flatten_base() (neml2.types.Tensor method) flatten_sub_batch() (neml2.types.Tensor method) flatten_sub_batch_into_first_base_axis() (neml2.types.Tensor method) fold_preserving() (neml2.types.Tensor method) forward() (neml2.aoti.AOTIModel method) (neml2.aoti.Model method) (neml2.es.IFT method) (neml2.es.NewtonStep method) (neml2.es.RHS method) (neml2.pyzag.NEML2PyzagModel method) from_dict() (neml2.es.AssembledVector class method) from_hit() (neml2.aoti.AOTIModel class method) (neml2.data.CubicCrystal class method) (neml2.drivers.TransientDriver class method) (neml2.drivers.TransientRegression class method) (neml2.drivers.Verification class method) (neml2.es.ModelNonlinearSystem class method) (neml2.models.model.Model class method) (neml2.solvers.DenseLU class method) (neml2.solvers.Newton class method) (neml2.solvers.NewtonWithLineSearch class method) (neml2.solvers.SchurComplement class method) (neml2.user_tensors.PythonTensor class method) from_typed() (neml2.types.Tensor class method) from_value() (neml2.types.Scalar class method) full() (neml2.types.PrimitiveTensor class method) (neml2.types.Scalar class method) G g() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) glayout (neml2.es.LinearSystem property) group() (neml2.es.AssembledMatrix method) (neml2.es.AssembledVector method) group_size() (neml2.es.AxisLayout method) group_sub_batch_shape() (neml2.es.AxisLayout method) groups (neml2.es.AxisLayout attribute) gt() (in module neml2.types) H heaviside() (in module neml2.types) hit (neml2.aoti.AOTIModel attribute) (neml2.data.CubicCrystal attribute) (neml2.drivers.TransientDriver attribute) (neml2.drivers.TransientRegression attribute) (neml2.drivers.Verification attribute) (neml2.es.ModelNonlinearSystem attribute) (neml2.models.model.Model attribute) (neml2.solvers.DenseLU attribute) (neml2.solvers.Newton attribute) (neml2.solvers.NewtonWithLineSearch attribute) (neml2.solvers.SchurComplement attribute) (neml2.user_tensors.PythonTensor attribute) HitField (class in neml2.schema) HitSchema (class in neml2.schema) I identity() (neml2.types.R2 class method) (neml2.types.Rot class method) (neml2.types.SR2 class method) (neml2.types.SSR4 class method) (neml2.types.WR2 class method) identity_C1() (neml2.types.SSR4 class method) identity_C2() (neml2.types.SSR4 class method) identity_C3() (neml2.types.SSR4 class method) identity_dev() (neml2.types.SSR4 class method) identity_sym() (neml2.types.SSR4 class method) identity_vol() (neml2.types.SSR4 class method) IFT (class in neml2.es) inbound_items() (neml2.models.resolver.DependencyResolver method) initialize() (neml2.es.ModelNonlinearSystem method) inner() (in module neml2.types) input() (in module neml2.schema) input_base_shapes (neml2.aoti.Model property) input_name (neml2.models.model.NLParam attribute) input_names (neml2.aoti.Model property) input_spec (neml2.models.model.Model attribute) (neml2.schema.HitSchema property) inv() (in module neml2.types) is_input (neml2.schema.HitField property) is_output (neml2.schema.HitField property) items() (neml2.es.SparseVector method) iterations (neml2.solvers.NonlinearResult attribute) J jacobian() (neml2.aoti.Model method) jvp() (neml2.aoti.Model method) jvp_compose() (in module neml2.types) jvp_euler_rodrigues() (in module neml2.types) jvp_exp_map() (in module neml2.types) jvp_linear_interpolation() (in module neml2.types) jvp_rotate() (in module neml2.types) K k (neml2.types.TensorWrapper property) k_ndim (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) (neml2.types.Tensor attribute) (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) k_pairing (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) (neml2.types.Tensor attribute) (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) k_shape (neml2.types.TensorWrapper property) k_state (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) (neml2.types.Tensor attribute) (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) keys() (neml2.es.SparseVector method) kind (neml2.schema.HitField attribute) kwargs_from_hit() (neml2.schema.HitSchema method) L lag_order() (in module neml2.pyzag) layout (neml2.es.AssembledVector attribute) (neml2.es.SparseVector attribute) linear_interpolation() (in module neml2.types) LinearSystem (class in neml2.es) linspace() (neml2.types.Scalar class method) load_input() (in module neml2.factory) load_model() (in module neml2.factory) load_nonlinear_system() (in module neml2.factory) load_package() (in module neml2.models.export) load_string() (in module neml2.factory) log() (in module neml2.types) log10() (in module neml2.types) lookback (neml2.pyzag.NEML2PyzagModel property) lt() (in module neml2.types) M M (neml2.data.CrystalGeometry property) macaulay() (in module neml2.types) materialize() (neml2.types.TensorWrapper method) matvec() (in module neml2.models.chain_rule) MAXITER (neml2.solvers.RetCode attribute) mean() (in module neml2.types) MillerIndex (class in neml2.types) Model (class in neml2.aoti) (class in neml2.models.model) ModelNonlinearSystem (class in neml2.es) module neml2.aoti neml2.cli neml2.data neml2.drivers neml2.es neml2.factory neml2.models neml2.models.chain_rule neml2.models.export neml2.models.model neml2.models.resolver neml2.pyzag neml2.schema neml2.solvers neml2.types neml2.user_tensors N name (neml2.schema.HitField attribute) named_parameters() (neml2.aoti.Model method) ndim (neml2.types.Tensor property) (neml2.types.TensorWrapper property) neml2.aoti module neml2.cli module neml2.data module neml2.drivers module neml2.es module neml2.factory module neml2.models module neml2.models.chain_rule module neml2.models.export module neml2.models.model module neml2.models.resolver module neml2.pyzag module neml2.schema module neml2.solvers module neml2.types module neml2.user_tensors module NEML2PyzagModel (class in neml2.pyzag) new_zeros() (neml2.types.Tensor method) Newton (class in neml2.solvers) NewtonStep (class in neml2.es) NewtonWithLineSearch (class in neml2.solvers) nforce (neml2.pyzag.NEML2PyzagModel property) ngroup (neml2.es.AxisLayout property) NLParam (class in neml2.models.model) NonlinearResult (class in neml2.solvers) NonlinearSystem (class in neml2.es) norm() (in module neml2.es) (in module neml2.types) norm_sq() (in module neml2.es) nslip (neml2.data.CrystalGeometry property) nslip_groups (neml2.data.CrystalGeometry property) nslip_in_group() (neml2.data.CrystalGeometry method) nstate (neml2.pyzag.NEML2PyzagModel property) nvar (neml2.es.AxisLayout property) O ones() (neml2.types.PrimitiveTensor class method) (neml2.types.Scalar class method) opaque_pow() (in module neml2.types) option() (in module neml2.schema) optional_reader (neml2.schema.HitField attribute) outbound_items() (neml2.models.resolver.DependencyResolver method) outer() (in module neml2.types) output() (in module neml2.schema) output_base_shapes (neml2.aoti.Model property) output_names (neml2.aoti.Model property) output_priorities (neml2.models.model.Model attribute) output_spec (neml2.models.model.Model attribute) (neml2.schema.HitSchema property) override (neml2.schema.HitField attribute) P parameter() (in module neml2.schema) parameters() (in module neml2.schema) pow() (in module neml2.types) PrimitiveTensor (class in neml2.types) priority (neml2.schema.HitField attribute) propagate_tangents() (neml2.models.model.Model method) provided_items (neml2.models.model.Model property) provider (neml2.models.model.NLParam attribute) provider_output (neml2.models.model.NLParam attribute) PythonTensor (class in neml2.user_tensors) R R2 (class in neml2.types) r2_from_sr2() (in module neml2.types) r2_from_wr2() (in module neml2.types) reader (neml2.schema.HitField attribute) register_neml2_object() (in module neml2.factory) register_submodule() (in module neml2.models.model) register_typed_buffer() (neml2.models.model.Model method) register_typed_parameter() (neml2.models.model.Model method) reject_unknown_fields() (neml2.schema.HitSchema method) required (neml2.schema.HitField property) resolve() (neml2.models.resolver.DependencyResolver method) result() (neml2.drivers.TransientDriver method) ret (neml2.solvers.NonlinearResult attribute) RetCode (class in neml2.solvers) RHS (class in neml2.es) Rot (class in neml2.types) rotate() (in module neml2.types) row_layout (neml2.es.AssembledMatrix attribute) (neml2.es.SparseMatrix attribute) run() (neml2.drivers.TransientDriver method) (neml2.drivers.TransientRegression method) (neml2.drivers.Verification method) S save_gold() (neml2.drivers.TransientDriver method) Scalar (class in neml2.types) SchurComplement (class in neml2.solvers) SecondOrderChainRuleAction (in module neml2.models.chain_rule) (in module neml2.models.model) SecondOrderChainRuleDict (in module neml2.models.chain_rule) (in module neml2.models.model) SecondOrderTangentAction (in module neml2.models.chain_rule) SECTION (neml2.aoti.AOTIModel attribute) (neml2.data.CubicCrystal attribute) (neml2.es.LinearSystem attribute) (neml2.models.model.Model attribute) (neml2.solvers.DenseLU attribute) (neml2.solvers.Newton attribute) (neml2.solvers.SchurComplement attribute) (neml2.user_tensors.PythonTensor attribute) select_blocks() (neml2.es.AssembledMatrix static method) set_g() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) set_parameter() (neml2.aoti.Model method) set_solver_config() (neml2.aoti.Model method) set_u() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) set_u_from_group_raws() (neml2.es.ModelNonlinearSystem method) setup_blayout() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) setup_glayout() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) setup_ulayout() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) shape (neml2.types.Tensor property) (neml2.types.TensorWrapper property) sign() (in module neml2.types) sinh() (in module neml2.types) skew() (in module neml2.types) slip_directions (neml2.data.CrystalGeometry property) slip_planes (neml2.data.CrystalGeometry property) solve() (neml2.solvers.DenseLU method) (neml2.solvers.Newton method) (neml2.solvers.SchurComplement method) (neml2.types.Tensor method) SparseMatrix (class in neml2.es) SparseVector (class in neml2.es) specs (neml2.es.AxisLayout attribute) sqrt() (in module neml2.types) SR2 (class in neml2.types) SSR4 (class in neml2.types) stack() (in module neml2.types) storage_size() (neml2.es.AxisLayout method) structure (neml2.es.AxisLayout attribute) sub_batch (neml2.types.Tensor property) (neml2.types.TensorWrapper property) sub_batch_meta (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) (neml2.types.Tensor attribute) (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) sub_batch_ndim (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) (neml2.types.Tensor attribute) (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) sub_batch_shape (neml2.types.Tensor property) (neml2.types.TensorWrapper property) sub_batch_shape() (neml2.es.AxisLayout method) sub_batch_shapes (neml2.es.AxisLayout attribute) sub_batch_state (neml2.types.MillerIndex attribute) (neml2.types.R2 attribute) (neml2.types.Rot attribute) (neml2.types.Scalar attribute) (neml2.types.SR2 attribute) (neml2.types.SSR4 attribute) (neml2.types.Tensor attribute) (neml2.types.TensorWrapper attribute) (neml2.types.Vec attribute) (neml2.types.WR2 attribute) sub_layout() (neml2.es.AxisLayout method) SUCCESS (neml2.solvers.RetCode attribute) suffix (neml2.schema.HitField attribute) sum() (in module neml2.types) SUPPORTS_SECOND_ORDER (neml2.models.model.Model attribute) sym() (in module neml2.types) sym_ops (neml2.data.CrystalGeometry property) T tail_index (neml2.models.model.NLParam attribute) TangentAction (in module neml2.models.chain_rule) (in module neml2.models.model) tanh() (in module neml2.types) Tensor (class in neml2.types) tensors (neml2.es.AssembledMatrix attribute) (neml2.es.AssembledVector attribute) TensorWrapper (class in neml2.types) to() (neml2.es.ModelNonlinearSystem method) (neml2.es.SparseMatrix method) (neml2.es.SparseVector method) (neml2.types.Tensor method) (neml2.types.TensorWrapper method) to_sparse() (neml2.es.ModelNonlinearSystem method) tr() (in module neml2.types) TransientDriver (class in neml2.drivers) TransientRegression (class in neml2.drivers) TYPE_NAME (neml2.user_tensors.CSVScalar attribute) (neml2.user_tensors.CSVSR2 attribute) (neml2.user_tensors.CSVVec attribute) (neml2.user_tensors.CSVWR2 attribute) type_of() (neml2.es.AxisLayout method) U u() (neml2.es.LinearSystem method) (neml2.es.ModelNonlinearSystem method) ulayout (neml2.es.LinearSystem property) unflatten_base() (neml2.types.Tensor method) unit() (in module neml2.types) V value_type (neml2.schema.HitField attribute) values (neml2.es.SparseVector attribute) var_inputs() (in module neml2.schema) var_size() (neml2.es.AxisLayout method) var_type (neml2.schema.HitField property) vars() (neml2.es.AxisLayout method) Vec (class in neml2.types) vec_component() (in module neml2.types) vec_from_scalars() (in module neml2.types) Verification (class in neml2.drivers) vol() (in module neml2.types) W W (neml2.data.CrystalGeometry property) where() (in module neml2.types) with_sub_batch_ndim() (neml2.types.TensorWrapper method) with_sub_batch_shapes() (neml2.es.AxisLayout method) WR2 (class in neml2.types) WRAPPER (neml2.user_tensors.CSVScalar attribute) (neml2.user_tensors.CSVSR2 attribute) (neml2.user_tensors.CSVVec attribute) (neml2.user_tensors.CSVWR2 attribute) Z zeros() (neml2.types.PrimitiveTensor class method) (neml2.types.Scalar class method) (neml2.types.Tensor class method) zeros_like() (neml2.types.PrimitiveTensor class method) (neml2.types.Tensor class method)
NEML2