Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | 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_promotion (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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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) cross() (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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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) dist() (in module neml2.types) 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) dV() (in module neml2.types) dynamic_batch (neml2.types.TensorWrapper property) dynamic_batch_shape (neml2.types.TensorWrapper property) E emitted_pairs() (neml2.es.IFT method) emitted_param_pairs() (neml2.es.ParamIFT 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.ParamIFT method) (neml2.es.RHS method) (neml2.pyzag.NEML2PyzagModel method) from_axis_angle() (neml2.types.MRP class method) from_axis_angle_standard() (neml2.types.MRP class 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_matrix() (neml2.types.MRP 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) gdist() (in module neml2.types) 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.MRP class method) (neml2.types.Quaternion class method) (neml2.types.R2 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.PromotedParam attribute) input_names (neml2.aoti.Model property) input_spec (neml2.models.model.Model attribute) (neml2.schema.HitSchema property) inv() (in module neml2.types) inverse() (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.AOTIModel method) (neml2.aoti.Model method) jvp() (neml2.aoti.AOTIModel method) (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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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() (in module neml2.types) 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 (neml2.solvers.NonlinearResult attribute) log() (in module neml2.types) log10() (in module neml2.types) logspace() (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 MRP (class in neml2.types) N name (neml2.schema.HitField attribute) named_parameters() (neml2.aoti.AOTIModel method) (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) 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 param_jacobian() (neml2.aoti.AOTIModel method) (neml2.aoti.Model method) (neml2.models.model.Model method) param_vjp() (neml2.aoti.AOTIModel method) (neml2.aoti.Model method) (neml2.models.model.Model method) parameter() (in module neml2.schema) parameter_base_shapes (neml2.aoti.Model property) (neml2.models.model.Model property) parameters() (in module neml2.schema) ParamIFT (class in neml2.es) pow() (in module neml2.types) PrimitiveTensor (class in neml2.types) priority (neml2.schema.HitField attribute) PromotedParam (class in neml2.models.model) propagate_tangents() (neml2.models.model.Model method) provided_items (neml2.models.model.Model property) provider (neml2.models.model.PromotedParam attribute) provider_output (neml2.models.model.PromotedParam attribute) PythonTensor (class in neml2.user_tensors) Q Quaternion (class in neml2.types) quaternion_dist() (in module neml2.types) quaternion_rotation_matrix() (in module neml2.types) R R2 (class in neml2.types) r2_from_sr2() (in module neml2.types) r2_from_wr2() (in module neml2.types) rand() (neml2.types.MRP class method) reader (neml2.schema.HitField attribute) reciprocal() (in module neml2.types) 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) request_AD() (neml2.models.model.Model 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) rotate() (in module neml2.types) rotation_from_to() (neml2.types.MRP class method) 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_result() (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.AOTIModel method) (neml2.aoti.Model method) (neml2.models.model.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) shadow() (in module neml2.types) 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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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.MRP attribute) (neml2.types.Quaternion attribute) (neml2.types.R2 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.PromotedParam 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_quaternion() (in module neml2.types) 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) vdot() (in module neml2.types) 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