Index

A

a (AdaptativeTanh attribute) A (LinearOrder2PDE attribute) a() (Vlasov method) a_laplacian() (in module scimba_torch.physical_models.elliptic_pde.linear_order_2) abs_det_jacobian() (ActivationSymplecticLayer method) (AffineFlowLayer method) (ConstantFlowLayer method) (CouplingLayer method) (GradPotentialSymplecticLayer method) (GSymplecticLayer method) (GSymplecticNet method) (InvertibleLayer method) (InvertibleNet method) (LASymplecticLayer method) (LASymplecticNet method) (LinearSymplecticLayer method) (PeriodicGradPotentialSymplecticLayer method) (PeriodicGSymplecticLayer method) (PeriodicGSymplecticNet method) (PSymplecticLayer method) (PSymplecticNet method) AbstractApproxSpace (class in scimba_torch.approximation_space.abstract_space) AbstractNonlinearProjector (class in scimba_torch.numerical_solvers.abstract_projector) AbstractPreconditioner (class in scimba_torch.numerical_solvers.abstract_preconditioner) AbstractScimbaOptimizer (class in scimba_torch.optimizers.scimba_optimizers) activate_first_optimizer() (OptimizerData method) activate_next_optimizer() (OptimizerData method) activated_optimizer (OptimizerData attribute) activation (GradPotentialSymplecticLayer attribute) (PeriodicGradPotentialSymplecticLayer attribute) (PirateNet attribute) (PirateNetBlock attribute) (SirenLayer attribute) activation_function() (in module scimba_torch.neural_nets.coordinates_based_nets.activation) activation_output (PirateNet attribute) activation_type (GradPotentialSymplecticLayer attribute) (PeriodicGradPotentialSymplecticLayer attribute) ActivationSymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.separated_symplectic_layers) AdaptativeTanh (class in scimba_torch.neural_nets.coordinates_based_nets.activation) adaptive_weights (GenericLosses attribute) add_bc_domain() (VolumetricDomain method) add_hole() (VolumetricDomain method)

add_subdomain() (VolumetricDomain method) AdvectionReactionDiffusion (class in scimba_torch.physical_models.temporal_pde.advection_diffusion_equation) AdvectionReactionDiffusion1DDirichletStrongForm (class in scimba_torch.physical_models.elliptic_pde.advection_diffusion) AdvectionReactionDiffusion2DDirichletStrongForm (class in scimba_torch.physical_models.elliptic_pde.advection_diffusion) AdvectionReactionDiffusionDirichletStrongForm (class in scimba_torch.physical_models.temporal_pde.advection_diffusion_equation) AffineFlowLayer (class in scimba_torch.neural_nets.structure_preserving_nets.affine_ode_layers) alpha (PirateNetBlock attribute) alpha_lr_annealing (GenericLosses attribute) AnagramPinnsElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.pinns) AnagramPreconditioner (class in scimba_torch.numerical_solvers.pinn_preconditioners) (class in scimba_torch.numerical_solvers.pinn_preconditioners.anagram_ng) AnagramPreconditionerProjector (class in scimba_torch.numerical_solvers.preconditioner_projector) AnagramProjector (class in scimba_torch.numerical_solvers.collocation_projector) AnagramTemporalPinns (class in scimba_torch.numerical_solvers.temporal_pde.pinns) AnisotropicRadial (class in scimba_torch.neural_nets.coordinates_based_nets.activation) apply_at_each_step (DiscreteFlowSpace attribute) apply_dict_of_func_to_func() (in module scimba_torch.numerical_solvers.functional_operator) apply_dict_of_vmap_to_label_tensors() (FunctionalOperator method) (in module scimba_torch.numerical_solvers.functional_operator) apply_func_to_dict_of_func() (FunctionalOperator method) (in module scimba_torch.numerical_solvers.functional_operator) apply_to_func() (FunctionalOperator method) ArcCircle2D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_2d) assemble_right_member() (AnagramPreconditioner method), [1] (AnagramPreconditionerProjector method) assemble_right_member_bc() (AnagramPreconditioner method), [1] assemble_right_member_dl() (AnagramPreconditioner method), [1] assemble_right_member_ic() (AnagramPreconditioner method), [1] assembly() (AbstractNonlinearProjector method) (CollocationProjector method) (DeepRitzElliptic method) (FlowTrainer method) (PinnsElliptic method) (TemporalPinns method) assembly_post_sampling() (AbstractNonlinearProjector method) (CollocationProjector method) (DeepRitzElliptic method) (FlowTrainer method) (PinnsElliptic method) (TemporalPinns method) (TimeDiscreteCollocationProjector method)

B

b (LinearOrder2PDE attribute) b_laplacian() (in module scimba_torch.physical_models.elliptic_pde.linear_order_2) backtracking_armijo_line_search() (in module scimba_torch.optimizers.line_search) backtracking_armijo_line_search_with_loss_theta_grad_loss_theta() (in module scimba_torch.optimizers.line_search) backward() (ActivationSymplecticLayer method) (AffineFlowLayer method) (ConstantFlowLayer method) (CouplingLayer method) (GradPotentialSymplecticLayer method) (GSymplecticLayer method) (InvertibleLayer method) (InvertibleNet method) (LASymplecticLayer method) (LinearSymplecticLayer method) (ODESplittedLayer method) (PeriodicGradPotentialSymplecticLayer method) (PeriodicGSymplecticLayer method) (PeriodicGSymplecticNet method) (PSymplecticLayer method) backwards_ode_integrator() (Characteristic static method) BallND (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_nd) bc_bilinearform() (WeakFormEllipticPDE method) bc_linearform() (WeakFormEllipticPDE method) bc_operator() (AdvectionReactionDiffusion1DDirichletStrongForm method) (AdvectionReactionDiffusion2DDirichletStrongForm method) (AdvectionReactionDiffusionDirichletStrongForm method) (CollisionalKineticPDE method) (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongForm method) (HeatEquation2DStrongFormImplicit method) (KineticPDE method) (Laplacian1DDirichletStrongForm method) (Laplacian1DNeumannStrongForm method) (Laplacian2DDirichletRitzForm method) (Laplacian2DDirichletStrongForm method) (LinearOrder2PDE method) (RadiativeTransfer method) (RegEikonalPDE method) (RitzFormEllipticPDE method) (StrongFormEllipticPDE method) (TemporalPDE method) (Transport1D method) (Transport1DImplicit method) (Vlasov method)

bc_rhs() (AdvectionReactionDiffusion1DDirichletStrongForm method) (AdvectionReactionDiffusion2DDirichletStrongForm method) (AdvectionReactionDiffusionDirichletStrongForm method) (CollisionalKineticPDE method) (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongForm method) (HeatEquation2DStrongFormImplicit method) (KineticPDE method) (Laplacian1DDirichletStrongForm method) (Laplacian1DNeumannStrongForm method) (Laplacian2DDirichletRitzForm method) (Laplacian2DDirichletStrongForm method) (LinearOrder2PDE method) (RadiativeTransfer method) (RegEikonalPDE method) (RitzFormEllipticPDE method) (StrongFormEllipticPDE method) (TemporalPDE method) (Transport1D method) (Transport1DImplicit method) (Vlasov method) bc_sample() (DomainSampler method) (HyperSurfaceSampler method) (RectangleMethod method) (TensorizedSampler method) bc_sample_new_points() (DomainSampler method) bc_sample_pre_sampled_points() (DomainSampler method) bean_2d() (ParametricHyperSurface static method) best_approx (AbstractApproxSpace attribute) best_loss (AbstractNonlinearProjector attribute) best_optimizer (AbstractScimbaOptimizer attribute) best_scheduler (AbstractScimbaOptimizer attribute) beta (KernelxSpace attribute) (Swish attribute) bias (FactorizedLinear attribute) (PeriodicEmbedding attribute) bilinearform() (WeakFormEllipticPDE method) blocks (PirateNet attribute) bool_linesearch (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) bool_preconditioner (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) bound (UniformTimeSampler attribute) bounds_postmap (VolumetricDomain attribute)

C

c (LinearOrder2PDE attribute) c_laplacian() (in module scimba_torch.physical_models.elliptic_pde.linear_order_2) call_and_update() (GenericLoss method) (GenericLosses method) CartesianProduct (class in scimba_torch.domain.meshless_domain.domain_nd) cat() (LabelTensor static method) cat_dict_of_tensors() (FunctionalOperator method) cat_dict_of_tensors_with_indexes() (in module scimba_torch.numerical_solvers.functional_operator) cat_tuple_of_tensors() (FunctionalOperator method) cat_tuple_of_tensors_along_flatten_keys() (FunctionalOperator method) centers (KernelxSpace attribute) Characteristic (class in scimba_torch.numerical_solvers.temporal_pde.neural_semilagrangian) check_functional_operator_against_keys() (in module scimba_torch.numerical_solvers.functional_operator) check_sample_size() (DomainSampler method) (ParametricSampler method) circle() (Mapping static method), [1] Circle2D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_2d) coeff (GenericLoss attribute) coeff_history (GenericLoss attribute) collision_kernel() (CollisionalKineticPDE method) CollisionalKineticPDE (class in scimba_torch.physical_models.kinetic_pde.abstract_kinetic_pde) CollocationProjector (class in scimba_torch.numerical_solvers.collocation_projector) compose() (Mapping static method), [1] compose_funcs() (in module scimba_torch.numerical_solvers.functional_operator) compute_all_losses() (GenericLosses method) compute_bounding_box() (in module scimba_torch.geometry.utils) compute_full_loss() (GenericLosses method) compute_full_loss_without_updating() (GenericLosses method) compute_full_preconditioning_matrix() (EnergyNaturalGradientPreconditioner method), [1] compute_model() (NeuralGalerkin method) compute_normals() (SurfacicDomain method) compute_periods() (GenericPeriodicNet method) compute_preconditioning_matrix() (AnagramPreconditioner method), [1] (AnagramPreconditionerProjector method) (EnergyNaturalGradientPreconditioner method), [1], [2] (EnergyNaturalGradientPreconditionerProjector method) (MatrixPreconditionerProjector method) (MatrixPreconditionerSolver method) (NystromNaturalGradientPreconditioner method), [1] (SketchyNaturalGradientPreconditioner method), [1]

compute_preconditioning_matrix_bc() (AnagramPreconditioner method), [1] (EnergyNaturalGradientPreconditioner method), [1], [2] (MatrixPreconditionerSolver method) (NystromNaturalGradientPreconditioner method), [1] (SketchyNaturalGradientPreconditioner method), [1] compute_preconditioning_matrix_dl() (AnagramPreconditioner method), [1] (EnergyNaturalGradientPreconditioner method), [1], [2] (MatrixPreconditionerSolver method) (NystromNaturalGradientPreconditioner method), [1] (SketchyNaturalGradientPreconditioner method), [1] compute_preconditioning_matrix_ic() (AnagramPreconditioner method), [1] (EnergyNaturalGradientPreconditioner method), [1], [2] (MatrixPreconditionerSolver method) (NystromNaturalGradientPreconditioner method), [1] (SketchyNaturalGradientPreconditioner method), [1] compute_relative_error_in_time() (ExplicitTimeDiscreteScheme method) concatenate() (LabelTensor method) ConstantFlowLayer (class in scimba_torch.neural_nets.structure_preserving_nets.affine_ode_layers) construct_lhs_cn() (DiscretePINNImplicit method) construct_lhs_ei() (DiscretePINNImplicit method) construct_lhs_sdirk2_step1() (DiscretePINNImplicit method) construct_lhs_sdirk2_step2() (DiscretePINNImplicit method) construct_rhs() (NeuralSemiLagrangian method) construct_rhs_cn() (DiscretePINNImplicit method) construct_rhs_ee() (DiscretePINN method) construct_rhs_ee_weak_bc() (DiscretePINN method) construct_rhs_ei() (DiscretePINNImplicit method) construct_rhs_rk2() (DiscretePINN method) construct_rhs_rk2_weak_bc() (DiscretePINN method) construct_rhs_rk4() (DiscretePINN method) construct_rhs_sdirk2_step1() (DiscretePINNImplicit method) construct_rhs_sdirk2_step2() (DiscretePINNImplicit method) ContinuousFlowSpace (class in scimba_torch.flows.deep_flows) Cosine (class in scimba_torch.neural_nets.coordinates_based_nets.activation) CouplingLayer (class in scimba_torch.neural_nets.structure_preserving_nets.coupling_layers) create_solution() (in module scimba_torch.flows.create_solution) Cube3D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d) Cuboid (class in scimba_torch.domain.mesh_based_domain.cuboid) Cylinder3D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d)

D

d (LinearOrder2PDE attribute) d_dirichlet() (in module scimba_torch.physical_models.elliptic_pde.linear_order_2) data_linesearch (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) DataLoss (class in scimba_torch.optimizers.losses) DeepRitzElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.deep_ritz) default_lr (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) default_pre_processing() (in module scimba_torch.numerical_solvers.preconditioner_solvers) del_mapping() (SurfacicDomain method) (VolumetricDomain method) detach() (LabelTensor method) dict_for_save() (AbstractApproxSpace method) (AbstractNonlinearProjector method) (AbstractScimbaOptimizer method) (GenericLosses method) (OptimizerData method) dim (LabelTensor attribute) (SurfacicDomain attribute) (UniformVelocitySampler attribute) (UniformVelocitySamplerOnCuboid attribute) (VolumetricDomain attribute) dim_hidden (PirateNet attribute)

dim_parametric (SurfacicDomain attribute) dim_postmap (SurfacicDomain attribute) (VolumetricDomain attribute) DiscontinuousLayer (class in scimba_torch.neural_nets.coordinates_based_nets.discontinuous_mlp) DiscontinuousMLP (class in scimba_torch.neural_nets.coordinates_based_nets.discontinuous_mlp) DiscreteFlowSpace (class in scimba_torch.flows.deep_flows) DiscretePINN (class in scimba_torch.numerical_solvers.temporal_pde.discrete_pinn) DiscretePINNImplicit (class in scimba_torch.numerical_solvers.temporal_pde.discrete_pinn) Disk2D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_2d) Disk3D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d) dist() (PolygonalApproxSignedDistance method) DivAGradUPDE (class in scimba_torch.physical_models.elliptic_pde.linear_order_2) domain (RectangleMethod attribute) (SurfacicSampler attribute) (VolumetricSampler attribute) domain_size (UniformVelocitySamplerOnCuboid attribute) domain_type (SurfacicDomain attribute) (VolumetricDomain attribute) DomainSampler (class in scimba_torch.integration.monte_carlo) dot_product() (PolygonalApproxSignedDistance method) dot_product_of_tensor_and_tuple() (Vlasov method) dot_product_of_tuples() (Vlasov method)

E

e (LinearOrder2PDE attribute) e_dirichlet() (in module scimba_torch.physical_models.elliptic_pde.linear_order_2) EllipticPDE (class in scimba_torch.physical_models.elliptic_pde.abstract_elliptic_pde) embedding (PirateNet attribute) embedding_1 (PirateNet attribute) embedding_2 (PirateNet attribute) energy_matrix() (DivAGradUPDE method) (Laplacian2DDirichletRitzForm method) EnergyNaturalGradientPreconditioner (class in scimba_torch.numerical_solvers.pinn_preconditioners) (class in scimba_torch.numerical_solvers.pinn_preconditioners.energy_ng) (class in scimba_torch.numerical_solvers.preconditioner_deep_ritz) EnergyNaturalGradientPreconditionerProjector (class in scimba_torch.numerical_solvers.preconditioner_projector) EnhancedFeatureNet (class in scimba_torch.neural_nets.coordinates_based_nets.features) ensure_periodicity() (Characteristic method) epochs_adapt (GenericLosses attribute) eps (DiscontinuousLayer attribute) (KernelxSpace attribute) estimate_bounding_box() (ParametricHyperSurface method) euler_midpoint() (in module scimba_torch.flows.integrators_ode) euler_symplectic() (in module scimba_torch.flows.integrators_ode) eval_and_gradx() (EnergyNaturalGradientPreconditioner method) eval_and_gradx_and_jactheta() (EnergyNaturalGradientPreconditioner method) eval_func() (MatrixPreconditionerSolver method)

eval_network() (MatrixPreconditionerSolver method) evaluate() (AbstractApproxSpace method) (ContinuousFlowSpace method) (DeepRitzElliptic method) (DiscreteFlowSpace method) (KernelxSpace method) (NNxSpace method), [1] (NNxSpaceModulation method) (NNxSpaceSplit method), [1] (NNxtSpace method) (NNxvSpace method) (PinnsElliptic method) (SeparatedNNxSpace method), [1] (SeparatedSpectralxSpace method) (SpectralxSpace method) (TemporalPinns method) exact_foot_constant_advection() (Characteristic method) expand_hidden_layers() (GenericMLP method) (NNxvSpace method) ExplicitEulerFlow (class in scimba_torch.flows.discretization_based_flows) ExplicitEulerHamiltonianFlow (class in scimba_torch.flows.discretization_based_flows) ExplicitTimeDiscreteScheme (class in scimba_torch.numerical_solvers.temporal_pde.time_discrete) ExponentialKernel (class in scimba_torch.approximation_space.kernelx_space)

F

f (LinearOrder2PDE attribute) factorized_glorot_normal() (in module scimba_torch.neural_nets.coordinates_based_nets.mlp) FactorizedLinear (class in scimba_torch.neural_nets.coordinates_based_nets.mlp) FirstOrderTemporalPDE (class in scimba_torch.physical_models.temporal_pde.abstract_temporal_pde) flatten_mesh() (Cuboid method) FlippedEmbedding (class in scimba_torch.neural_nets.embeddings.periodic_embedding) FlippedMLP (class in scimba_torch.neural_nets.coordinates_based_nets.features) FlowTrainer (class in scimba_torch.flows.flow_trainer) forward() (ActivationSymplecticLayer method) (AdaptativeTanh method) (AffineFlowLayer method) (AnisotropicRadial method) (ConstantFlowLayer method) (ContinuousFlowSpace method) (Cosine method) (CouplingLayer method) (DiscontinuousLayer method) (DiscontinuousMLP method) (DiscreteFlowSpace method) (EnhancedFeatureNet method) (ExplicitEulerFlow method) (ExplicitEulerHamiltonianFlow method) (ExponentialKernel method) (FactorizedLinear method) (FlippedEmbedding method) (FlippedMLP method) (GaussianKernel method) (GELU method) (GenericFourierNet method) (GenericMLP method) (GenericMMLP method) (GenericModulationMLP method) (GenericMultiScaleFourierNet method) (GenericPeriodicNet method) (GenericResNet method) (GradPotentialSymplecticLayer method) (GSymplecticLayer method) (Hat method) (Heaviside method) (Id method) (InvertibleNet method) (IsotropicRadial method) (KernelxSpace method) (LASymplecticLayer method) (LinearSymplecticLayer method) (MassLoss method) (MultiMLP method) (MultiquadraticKernel method) (NeuralFlow method) (NNxSpace method), [1] (NNxSpaceModulation method) (NNxSpaceSplit method), [1] (NNxtSpace method) (NNxvSpace method) (ODESplittedLayer method) (PeriodicEmbedding method) (PeriodicGradPotentialSymplecticLayer method) (PeriodicGSymplecticLayer method) (PeriodicGSymplecticNet method) (PirateNet method) (PirateNetBlock method) (PSymplecticLayer method) (Rational method) (RegularizedHat method) (ReLUk method) (SeparatedNNxSpace method), [1] (SeparatedSpectralxSpace method) (Sigmoid method) (SiLU method) (Sine method) (SirenLayer method) (SirenNet method) (Softmax method) (Softplus method) (SpectralxSpace method) (Swish method) (SymplecticEulerFlowSep method) (Tanh method)

FourierMLP (class in scimba_torch.neural_nets.coordinates_based_nets.features) FourierResNet (class in scimba_torch.neural_nets.coordinates_based_nets.features) freq (Cosine attribute) (Sine attribute) from_dim (Mapping attribute), [1] full_bc_domain() (BallND method), [1] (CartesianProduct method) (Cube3D method), [1] (Cylinder3D method), [1] (Disk2D method), [1] (Disk3D method), [1] (HypercubeND method), [1] (Polygon2D method), [1] (Segment1D method), [1] (Square2D method), [1] (Torus3D method), [1] (TorusFrom2DVolume method), [1] (VolumetricDomain method) full_data (FlowTrainer attribute) func (GenericLoss attribute) FUNC_TYPE (in module scimba_torch.utils.typing_protocols) FuncFuncArgsCallable (class in scimba_torch.utils.typing_protocols) functional_operator() (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongFormImplicit method) (Laplacian2DDirichletStrongForm method) (LinearOrder2PDE method) (RegEikonalPDE method) (Transport1D method) (Transport1DImplicit method) functional_operator_bc() (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongFormImplicit method) (Laplacian2DDirichletRitzForm method) (Laplacian2DDirichletStrongForm method) (LinearOrder2PDE method) (RegEikonalPDE method) (Transport1D method) (Transport1DImplicit method) functional_operator_ic() (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongFormImplicit method) (Transport1D method) (Transport1DImplicit method) functional_operator_id() (in module scimba_torch.numerical_solvers.preconditioner_solvers) FunctionalOperator (class in scimba_torch.numerical_solvers.functional_operator) FuncTypeCallable (class in scimba_torch.utils.typing_protocols)

G

g (LinearOrder2PDE attribute) gauss_legendre() (in module scimba_torch.flows.integrators_ode) GaussianKernel (class in scimba_torch.approximation_space.kernelx_space) GELU (class in scimba_torch.neural_nets.coordinates_based_nets.activation) GenericFeatureNet (class in scimba_torch.neural_nets.coordinates_based_nets.features) GenericFourierNet (class in scimba_torch.neural_nets.coordinates_based_nets.features) GenericLoss (class in scimba_torch.optimizers.losses) GenericLosses (class in scimba_torch.optimizers.losses) GenericMLP (class in scimba_torch.neural_nets.coordinates_based_nets.mlp) GenericMMLP (class in scimba_torch.neural_nets.coordinates_based_nets.mlp) GenericModulationMLP (class in scimba_torch.neural_nets.coordinates_based_nets.mlp) GenericMultiScaleFourierNet (class in scimba_torch.neural_nets.coordinates_based_nets.features) GenericPeriodicNet (class in scimba_torch.neural_nets.coordinates_based_nets.features) GenericResNet (class in scimba_torch.neural_nets.coordinates_based_nets.res_net) get_all_bc_domains() (VolumetricDomain method) get_args_for_operator() (MatrixPreconditionerSolver method) get_args_for_operator_bc() (MatrixPreconditionerSolver method) get_args_for_operator_ic() (MatrixPreconditionerSolver method) get_coeff() (GenericLoss method) (GenericLosses method) get_coeff_history() (GenericLoss method) (GenericLosses method) get_components() (LabelTensor method) (MultiLabelTensor method) get_dof() (AbstractApproxSpace method) (AbstractNonlinearProjector method) (CollocationProjector method) (ContinuousFlowSpace method) (DeepRitzElliptic method) (DiscreteFlowSpace method) (FlowTrainer method) (KernelxSpace method) (NNxSpace method), [1] (NNxSpaceModulation method) (NNxSpaceSplit method), [1] (NNxtSpace method) (NNxvSpace method) (PinnsElliptic method) (SeparatedNNxSpace method), [1] (SeparatedSpectralxSpace method) (SpectralxSpace method) (TemporalPinns method)

get_extended_bounds() (VolumetricDomain method) get_extended_bounds_postmap() (VolumetricDomain method) get_filepath_for_save() (in module scimba_torch.utils.paths) get_foot() (Characteristic method) get_foot_kinetic_pde() (Characteristic method) get_formatted_current_theta() (MatrixPreconditionerSolver method) get_full_loss() (GenericLosses method) get_gradient() (AbstractNonlinearProjector method) get_history() (GenericLosses method) get_list_of_bc_labels() (DomainSampler method) (RectangleMethod method) get_list_of_labels() (DomainSampler method) (RectangleMethod method) get_loss() (GenericLoss method) (GenericLosses method) get_loss_grad_loss() (AbstractNonlinearProjector method) get_loss_history() (GenericLoss method) get_opt_gradients() (OptimizerData method) get_preconditioning_matrix() (MatrixPreconditionerProjector method) (MatrixPreconditionerSolver method) get_preconditioning_matrix_bc() (MatrixPreconditionerProjector method) (MatrixPreconditionerSolver method) get_preconditioning_matrix_dl() (MatrixPreconditionerSolver method) get_preconditioning_matrix_ic() (MatrixPreconditionerSolver method) get_sdf() (ArcCircle2D method), [1] (Circle2D method), [1] (Segment2D method), [1] (Sphere3D method), [1] (SurfacicDomain method) get_static_terminal_width() (in module scimba_torch.utils.environment) get_vectorizing_scheme() (MatrixPreconditionerSolver method) get_vectorizing_scheme_bc() (MatrixPreconditionerSolver method) get_vectorizing_scheme_ic() (MatrixPreconditionerSolver method) get_verbosity() (in module scimba_torch.utils.verbosity) get_weighted_loss() (GenericLoss method) grad() (AbstractApproxSpace method) (EllipticPDE method) (LinearOrder2PDE method) (TemporalPDE method) GradPotentialSymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.separated_symplectic_layers) GSymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.coupling_symplectic_layers) GSymplecticNet (class in scimba_torch.neural_nets.structure_preserving_nets.symplectic_nets)

H

h_func() (ExplicitEulerHamiltonianFlow method) handle_dirichlet() (Characteristic method) has_bc_domain (VolumetricDomain property) Hat (class in scimba_torch.neural_nets.coordinates_based_nets.activation) HeatEquation1DDirichletStrongForm (class in scimba_torch.physical_models.temporal_pde.heat_equation) HeatEquation1DStrongForm (class in scimba_torch.physical_models.temporal_pde.heat_equation) HeatEquation2DStrongForm (class in scimba_torch.physical_models.temporal_pde.heat_equation) HeatEquation2DStrongFormImplicit (class in scimba_torch.physical_models.temporal_pde.heat_equation)

Heaviside (class in scimba_torch.neural_nets.coordinates_based_nets.activation) hidden_layers (DiscontinuousMLP attribute) (GenericMLP attribute) (GenericMMLP attribute) (GenericResNet attribute) hidden_layers_mult (GenericMMLP attribute) HypercubeND (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_nd) HyperSurfaceSampler (class in scimba_torch.geometry.monte_carlo_hypersurface)

I

Id (class in scimba_torch.neural_nets.coordinates_based_nets.activation) identity() (in module scimba_torch.flows.deep_flows) (Mapping static method), [1] IdPreconditioner (class in scimba_torch.numerical_solvers.abstract_preconditioner) in_size (DiscreteFlowSpace attribute) (NNxSpace attribute), [1] (NNxSpaceSplit attribute), [1] (NNxtSpace attribute) (NNxvSpace attribute) (PirateNet attribute) (ScimbaModule attribute) (SeparatedNNxSpace attribute), [1] (SirenLayer attribute) inference() (DiscreteFlowSpace method) init_() (SirenLayer method) init_loss() (GenericLoss method) (GenericLosses method) init_losses() (GenericLosses method) init_ones() (in module scimba_torch.physical_models.temporal_pde.advection_diffusion_equation) initial_condition() (AdvectionReactionDiffusion method) (CollisionalKineticPDE method) (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongForm method) (HeatEquation2DStrongFormImplicit method) (KineticPDE method) (RadiativeTransfer method) (TemporalPDE method) (Transport1D method) (Transport1DImplicit method) (Vlasov method) initialization() (ExplicitTimeDiscreteScheme method) inner_rk_step() (NeuralGalerkin method) inner_step() (AbstractScimbaOptimizer method) (ScimbaAdam method) (ScimbaLBFGS method) (ScimbaMomentum method) (ScimbaSGD method) (ScimbaSSBFGS method) (ScimbaSSBroyden method)

inner_update_lstsq() (NeuralGalerkin method) inner_update_matrix() (NeuralGalerkin method) integrator (AbstractApproxSpace attribute) (NNxSpace attribute), [1] (NNxSpaceModulation attribute) (NNxSpaceSplit attribute), [1] (NNxtSpace attribute) (NNxvSpace attribute) (SeparatedNNxSpace attribute), [1] (SpectralxSpace attribute) inv (Mapping attribute), [1] inv_identity() (Mapping static method), [1] invert() (Mapping static method), [1] InvertibleLayer (class in scimba_torch.neural_nets.structure_preserving_nets.invertible_nn) InvertibleNet (class in scimba_torch.neural_nets.structure_preserving_nets.invertible_nn) is_first (SirenLayer attribute) is_google_colab() (in module scimba_torch.utils.environment) is_inside() (CartesianProduct method) (TorusFrom2DVolume method), [1] (VolumetricDomain method) is_inside_postmap() (VolumetricDomain method) is_inside_postmap_np() (VolumetricDomain method) is_invertible (Mapping attribute), [1] is_jupyter_notebook() (in module scimba_torch.utils.environment) is_main_domain (VolumetricDomain attribute) is_mapped (VolumetricDomain attribute) is_on_boundary() (TorusFrom2DVolume method), [1] (VolumetricDomain method) is_on_boundary_postmap() (VolumetricDomain method) is_outside() (CartesianProduct method) (TorusFrom2DVolume method), [1] (VolumetricDomain method) is_outside_np() (VolumetricDomain method) is_outside_postmap() (VolumetricDomain method) is_outside_postmap_np() (VolumetricDomain method) is_static_width_environment() (in module scimba_torch.utils.environment) is_valid_parametric_point_np() (SurfacicDomain method) IsotropicRadial (class in scimba_torch.neural_nets.coordinates_based_nets.activation)

J

jac (Mapping attribute), [1] jac_inv (Mapping attribute), [1] jacobian() (AbstractApproxSpace method) (ContinuousFlowSpace method) (DiscreteFlowSpace method) (KernelxSpace method) (NNxSpace method), [1] (NNxSpaceModulation method) (NNxSpaceSplit method), [1] (NNxtSpace method) (NNxvSpace method) (SeparatedNNxSpace method), [1] (SeparatedSpectralxSpace method) (SpectralxSpace method)

K

k (Heaviside attribute) k_func() (SymplecticEulerFlowSep method)

KernelxSpace (class in scimba_torch.approximation_space.kernelx_space) KineticPDE (class in scimba_torch.physical_models.kinetic_pde.abstract_kinetic_pde)

L

labels (LabelTensor attribute) (MultiLabelTensor attribute) LabelTensor (class in scimba_torch.utils.scimba_tensors) Laplacian1DDirichletStrongForm (class in scimba_torch.physical_models.elliptic_pde.laplacians) Laplacian1DNeumannStrongForm (class in scimba_torch.physical_models.elliptic_pde.laplacians) Laplacian2DDirichletRitzForm (class in scimba_torch.physical_models.elliptic_pde.laplacians) Laplacian2DDirichletStrongForm (class in scimba_torch.physical_models.elliptic_pde.laplacians) LASymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.coupling_symplectic_layers) LASymplecticNet (class in scimba_torch.neural_nets.structure_preserving_nets.symplectic_nets) layer (SirenLayer attribute) layers (SirenNet attribute) learn_regularized_sdf() (in module scimba_torch.geometry.regularized_sdf_projectors) learnable (Swish attribute) linear_a (ActivationSymplecticLayer attribute) linear_mu (GradPotentialSymplecticLayer attribute) (LinearSymplecticLayer attribute) (PeriodicGradPotentialSymplecticLayer attribute) linear_q (GradPotentialSymplecticLayer attribute) (LinearSymplecticLayer attribute) linear_q1 (PeriodicGradPotentialSymplecticLayer attribute) linearform() (DivAGradUPDE method) (Laplacian2DDirichletRitzForm method) (RitzFormEllipticPDE method) (WeakFormEllipticPDE method) linearlayer (DiscontinuousLayer attribute) LinearOrder2PDE (class in scimba_torch.physical_models.elliptic_pde.linear_order_2) LinearProjector (class in scimba_torch.numerical_solvers.collocation_projector) LinearSymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.separated_symplectic_layers) list_bc_domains (VolumetricDomain attribute) list_holes (VolumetricDomain attribute) list_sampler (TensorizedSampler attribute) list_subdomains (VolumetricDomain attribute)

load() (AbstractNonlinearProjector method) (AbstractScimbaOptimizer method) load_from_best_approx() (AbstractApproxSpace method) load_from_dict() (AbstractApproxSpace method) (AbstractNonlinearProjector method) (GenericLosses method) (OptimizerData method) log_abs_det_jacobian() (ActivationSymplecticLayer method) (AffineFlowLayer method) (ConstantFlowLayer method) (CouplingLayer method) (GradPotentialSymplecticLayer method) (GSymplecticLayer method) (GSymplecticNet method) (InvertibleLayer method) (InvertibleNet method) (LASymplecticLayer method) (LASymplecticNet method) (LinearSymplecticLayer method) (ODESplittedLayer method) (PeriodicGradPotentialSymplecticLayer method) (PeriodicGSymplecticLayer method) (PeriodicGSymplecticNet method) (PSymplecticLayer method) (PSymplecticNet method) logarithmic_grid_line_search() (in module scimba_torch.optimizers.line_search) loss (GenericLoss attribute) (GenericLosses attribute) loss_history (GenericLoss attribute) (GenericLosses attribute) losses (AbstractNonlinearProjector attribute) losses_dict (GenericLosses attribute) lower_bound (UniformVelocitySamplerOnCuboid attribute)

M

M_aniso (KernelxSpace attribute) make_diffusion_directions() (Characteristic method) make_tuple() (AbstractNonlinearProjector method) map (Mapping attribute), [1] Mapping (class in scimba_torch.utils) (class in scimba_torch.utils.mapping) mapping (SurfacicDomain attribute) (VolumetricDomain attribute) MassLoss (class in scimba_torch.optimizers.losses) matrix_zero() (in module scimba_torch.physical_models.elliptic_pde.advection_diffusion) matrix_zeros() (in module scimba_torch.physical_models.temporal_pde.advection_diffusion_equation) MatrixPreconditionerDeepRitz (class in scimba_torch.numerical_solvers.preconditioner_deep_ritz) MatrixPreconditionerPinn (class in scimba_torch.numerical_solvers.preconditioner_pinns) MatrixPreconditionerProjector (class in scimba_torch.numerical_solvers.preconditioner_projector) MatrixPreconditionerSolver (class in scimba_torch.numerical_solvers.preconditioner_solvers) metric_matrix() (CollocationProjector method) (DeepRitzElliptic method) (EnergyNaturalGradientPreconditionerProjector method) mlps (MultiMLP attribute) model (DiscretePINN attribute) module scimba_torch.approximation_space scimba_torch.approximation_space.abstract_space scimba_torch.approximation_space.kernelx_space scimba_torch.approximation_space.nn_space scimba_torch.approximation_space.radial_space scimba_torch.approximation_space.spectral_space scimba_torch.domain scimba_torch.domain.mesh_based_domain scimba_torch.domain.mesh_based_domain.cuboid scimba_torch.domain.meshless_domain scimba_torch.domain.meshless_domain.base scimba_torch.domain.meshless_domain.domain_1d scimba_torch.domain.meshless_domain.domain_2d scimba_torch.domain.meshless_domain.domain_3d scimba_torch.domain.meshless_domain.domain_nd scimba_torch.domain.sdf scimba_torch.flows scimba_torch.flows.create_solution scimba_torch.flows.deep_flows scimba_torch.flows.discretization_based_flows scimba_torch.flows.flow_trainer scimba_torch.flows.integrators_ode scimba_torch.geometry scimba_torch.geometry.monte_carlo_hypersurface scimba_torch.geometry.parametric_hypersurface scimba_torch.geometry.regularized_eikonal_pde scimba_torch.geometry.regularized_sdf_projectors scimba_torch.geometry.utils scimba_torch.integration scimba_torch.integration.mesh_based_quadrature scimba_torch.integration.monte_carlo scimba_torch.integration.monte_carlo_parameters scimba_torch.integration.monte_carlo_time scimba_torch.neural_nets scimba_torch.neural_nets.coordinates_based_nets scimba_torch.neural_nets.coordinates_based_nets.activation scimba_torch.neural_nets.coordinates_based_nets.discontinuous_mlp scimba_torch.neural_nets.coordinates_based_nets.features scimba_torch.neural_nets.coordinates_based_nets.mlp scimba_torch.neural_nets.coordinates_based_nets.pirate_net scimba_torch.neural_nets.coordinates_based_nets.res_net scimba_torch.neural_nets.coordinates_based_nets.scimba_module scimba_torch.neural_nets.coordinates_based_nets.siren scimba_torch.neural_nets.embeddings scimba_torch.neural_nets.embeddings.periodic_embedding scimba_torch.neural_nets.structure_preserving_nets scimba_torch.neural_nets.structure_preserving_nets.affine_ode_layers scimba_torch.neural_nets.structure_preserving_nets.coupling_layers scimba_torch.neural_nets.structure_preserving_nets.coupling_symplectic_layers scimba_torch.neural_nets.structure_preserving_nets.invertible_nn scimba_torch.neural_nets.structure_preserving_nets.nilpotent_symplectic_layer scimba_torch.neural_nets.structure_preserving_nets.nonlinear_ode_layers scimba_torch.neural_nets.structure_preserving_nets.ode_splitted_layer scimba_torch.neural_nets.structure_preserving_nets.separated_symplectic_layers scimba_torch.neural_nets.structure_preserving_nets.split_layer scimba_torch.neural_nets.structure_preserving_nets.symplectic_nets scimba_torch.numerical_solvers scimba_torch.numerical_solvers.abstract_preconditioner scimba_torch.numerical_solvers.abstract_projector scimba_torch.numerical_solvers.collocation_projector scimba_torch.numerical_solvers.elliptic_pde scimba_torch.numerical_solvers.elliptic_pde.deep_ritz scimba_torch.numerical_solvers.elliptic_pde.pinns scimba_torch.numerical_solvers.functional_operator scimba_torch.numerical_solvers.pinn_preconditioners scimba_torch.numerical_solvers.pinn_preconditioners.anagram_ng scimba_torch.numerical_solvers.pinn_preconditioners.energy_ng scimba_torch.numerical_solvers.pinn_preconditioners.nystrom_ng scimba_torch.numerical_solvers.pinn_preconditioners.sketchy_ng scimba_torch.numerical_solvers.preconditioner_deep_ritz scimba_torch.numerical_solvers.preconditioner_pinns scimba_torch.numerical_solvers.preconditioner_projector scimba_torch.numerical_solvers.preconditioner_solvers scimba_torch.numerical_solvers.temporal_pde scimba_torch.numerical_solvers.temporal_pde.discrete_pinn scimba_torch.numerical_solvers.temporal_pde.neural_galerkin scimba_torch.numerical_solvers.temporal_pde.neural_semilagrangian scimba_torch.numerical_solvers.temporal_pde.pinns scimba_torch.numerical_solvers.temporal_pde.time_discrete scimba_torch.optimizers scimba_torch.optimizers.line_search scimba_torch.optimizers.losses scimba_torch.optimizers.optimizers_data scimba_torch.optimizers.scimba_optimizers scimba_torch.optimizers.ssbroyden scimba_torch.physical_models scimba_torch.physical_models.elliptic_pde scimba_torch.physical_models.elliptic_pde.abstract_elliptic_pde scimba_torch.physical_models.elliptic_pde.advection_diffusion scimba_torch.physical_models.elliptic_pde.laplacians scimba_torch.physical_models.elliptic_pde.linear_order_2 scimba_torch.physical_models.kinetic_pde scimba_torch.physical_models.kinetic_pde.abstract_kinetic_pde scimba_torch.physical_models.kinetic_pde.radiative_transfer scimba_torch.physical_models.kinetic_pde.vlasov scimba_torch.physical_models.temporal_pde scimba_torch.physical_models.temporal_pde.abstract_temporal_pde scimba_torch.physical_models.temporal_pde.advection_diffusion_equation scimba_torch.physical_models.temporal_pde.heat_equation scimba_torch.physical_models.temporal_pde.transport_equation scimba_torch.plots scimba_torch.plots.plot_regularized_sdf_projector scimba_torch.plots.plot_time_discrete_scheme scimba_torch.plots.plots_nd scimba_torch.utils scimba_torch.utils.environment scimba_torch.utils.mapping scimba_torch.utils.paths scimba_torch.utils.scimba_tensors scimba_torch.utils.typing_protocols scimba_torch.utils.verbosity

MultiLabelTensor (class in scimba_torch.utils.scimba_tensors) MultiMLP (class in scimba_torch.neural_nets.coordinates_based_nets.mlp) MultiquadraticKernel (class in scimba_torch.approximation_space.kernelx_space) MultiScaleFourierMLP (class in scimba_torch.neural_nets.coordinates_based_nets.features) MultiScaleFourierResNet (class in scimba_torch.neural_nets.coordinates_based_nets.features)

N

n_sampler (TensorizedSampler attribute) NaturalGradientDeepRitzElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.deep_ritz) NaturalGradientFlowTrainer (class in scimba_torch.flows.flow_trainer) NaturalGradientPinnsElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.pinns) NaturalGradientProjector (class in scimba_torch.numerical_solvers.collocation_projector) NaturalGradientTemporalPinns (class in scimba_torch.numerical_solvers.temporal_pde.pinns) nb_blocks (PirateNet attribute) nb_epoch_preconditioner_computing (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) ndof (AbstractApproxSpace attribute) (DiscreteFlowSpace attribute) (KernelxSpace attribute) (NNxSpace attribute), [1] (NNxSpaceModulation attribute) (NNxSpaceSplit attribute), [1] (NNxtSpace attribute) (NNxvSpace attribute) (SeparatedNNxSpace attribute), [1] net (ContinuousFlowSpace attribute) network (DiscreteFlowSpace attribute) (NNxSpace attribute), [1] (NNxSpaceModulation attribute) (NNxtSpace attribute) (SeparatedNNxSpace attribute), [1]

network_cat (NNxSpaceSplit attribute), [1] network_mu (NNxSpaceSplit attribute), [1] network_x (NNxSpaceSplit attribute), [1] NeuralFlow (class in scimba_torch.flows.discretization_based_flows) NeuralGalerkin (class in scimba_torch.numerical_solvers.temporal_pde.neural_galerkin) NeuralSemiLagrangian (class in scimba_torch.numerical_solvers.temporal_pde.neural_semilagrangian) next_optimizer (OptimizerData attribute) NNxSpace (class in scimba_torch.approximation_space.nn_space) (class in scimba_torch.approximation_space.radial_space) NNxSpaceModulation (class in scimba_torch.approximation_space.nn_space) NNxSpaceSplit (class in scimba_torch.approximation_space.nn_space) (class in scimba_torch.approximation_space.radial_space) NNxtSpace (class in scimba_torch.approximation_space.nn_space) NNxvSpace (class in scimba_torch.approximation_space.nn_space) no_grad() (LabelTensor method) NoScheduler (class in scimba_torch.optimizers.scimba_optimizers) numerical_foot() (Characteristic method) NumpyUniform (class in scimba_torch.integration.monte_carlo) NystromNaturalGradientPinnsElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.pinns) NystromNaturalGradientPreconditioner (class in scimba_torch.numerical_solvers.pinn_preconditioners) (class in scimba_torch.numerical_solvers.pinn_preconditioners.nystrom_ng) NystromNaturalGradientTemporalPinns (class in scimba_torch.numerical_solvers.temporal_pde.pinns)

O

ODESplittedLayer (class in scimba_torch.neural_nets.structure_preserving_nets.ode_splitted_layer) ones() (in module scimba_torch.physical_models.elliptic_pde.laplacians) operator() (AdvectionReactionDiffusion1DDirichletStrongForm method) (AdvectionReactionDiffusion2DDirichletStrongForm method) (Laplacian1DDirichletStrongForm method) (Laplacian1DNeumannStrongForm method) (Laplacian2DDirichletStrongForm method) (LinearOrder2PDE method) (RegEikonalPDE method) (StrongFormEllipticPDE method) (TemporalPDE method) optimizer (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) optimizer_step() (AbstractScimbaOptimizer method) OptimizerData (class in scimba_torch.optimizers.optimizers_data) optimizers (OptimizerData attribute)

out_size (DiscreteFlowSpace attribute) (NNxSpace attribute), [1] (NNxSpaceModulation attribute) (NNxSpaceSplit attribute), [1] (NNxtSpace attribute) (NNxvSpace attribute) (PirateNet attribute) (ScimbaModule attribute) (SeparatedNNxSpace attribute), [1] (SirenLayer attribute) output_layer (DiscontinuousMLP attribute) (GenericMLP attribute) (GenericMMLP attribute) (GenericResNet attribute) (PirateNet attribute) (ScimbaModule attribute)

P

p0 (Rational attribute) p1 (Rational attribute) p2 (Rational attribute) p3 (Rational attribute) param_dim (ContinuousFlowSpace attribute) (DiscreteFlowSpace attribute) parameters() (GenericFeatureNet method) (ScimbaModule method) ParametricHyperSurface (class in scimba_torch.geometry.parametric_hypersurface) ParametricSampler (class in scimba_torch.integration.monte_carlo_parameters) percent_in (VolumetricSampler attribute) PeriodicEmbedding (class in scimba_torch.neural_nets.embeddings.periodic_embedding) PeriodicGradPotentialSymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.separated_symplectic_layers) PeriodicGSymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.coupling_symplectic_layers) PeriodicGSymplecticNet (class in scimba_torch.neural_nets.structure_preserving_nets.symplectic_nets) PeriodicMLP (class in scimba_torch.neural_nets.coordinates_based_nets.features) PeriodicResNet (class in scimba_torch.neural_nets.coordinates_based_nets.features) phase (PeriodicEmbedding attribute) phi() (PolygonalApproxSignedDistance method) PinnsElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.pinns) PirateNet (class in scimba_torch.neural_nets.coordinates_based_nets.pirate_net) PirateNetBlock (class in scimba_torch.neural_nets.coordinates_based_nets.pirate_net) plot() (GenericLosses method) (in module scimba_torch.numerical_solvers.collocation_projector) plot_abstract_approx_space() (in module scimba_torch.plots.plots_nd) plot_abstract_approx_spaces() (in module scimba_torch.plots.plots_nd) plot_regularized_sdf_projector() (in module scimba_torch.plots.plot_regularized_sdf_projector)

plot_time_discrete_scheme() (in module scimba_torch.plots.plot_time_discrete_scheme) Point1D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_1d) poly_coeffs (PSymplecticLayer attribute) Polygon2D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_2d) PolygonalApproxSignedDistance (class in scimba_torch.domain.sdf) post_processing (DiscreteFlowSpace attribute) (NNxSpace attribute) (NNxSpaceModulation attribute) (NNxSpaceSplit attribute) pre_processing (NNxSpace attribute) (NNxSpaceModulation attribute) pre_processing_mu (NNxSpaceModulation attribute) (NNxSpaceSplit attribute) pre_processing_x (NNxSpaceModulation attribute) (NNxSpaceSplit attribute) PreconditionedPinnsElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.pinns) PreconditionedTemporalPinns (class in scimba_torch.numerical_solvers.temporal_pde.pinns) preconditioner (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) principal_weights (GenericLosses attribute) print_torch_setting() (in module scimba_torch.utils.verbosity) projection_data (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) projector (DiscretePINN attribute) PSymplecticLayer (class in scimba_torch.neural_nets.structure_preserving_nets.nilpotent_symplectic_layer) PSymplecticNet (class in scimba_torch.neural_nets.structure_preserving_nets.symplectic_nets)

Q

q0 (Rational attribute) q1 (Rational attribute) q2 (Rational attribute)

quadraticform() (DivAGradUPDE method) (Laplacian2DDirichletRitzForm method) (RitzFormEllipticPDE method)

R

RadiativeTransfer (class in scimba_torch.physical_models.kinetic_pde.radiative_transfer) Rational (class in scimba_torch.neural_nets.coordinates_based_nets.activation) RbfSinus (class in scimba_torch.neural_nets.coordinates_based_nets.activation) re_init() (EnhancedFeatureNet method) re_init_features() (GenericFeatureNet method) (GenericMultiScaleFourierNet method) read_points_normals_from_file() (in module scimba_torch.geometry.utils) RectangleMethod (class in scimba_torch.integration.mesh_based_quadrature) RegEikonalPDE (class in scimba_torch.geometry.regularized_eikonal_pde) RegularizedHat (class in scimba_torch.neural_nets.coordinates_based_nets.activation) RegularizedSdfAnagram (class in scimba_torch.geometry.regularized_sdf_projectors) RegularizedSdfEnergyNaturalGradient (class in scimba_torch.geometry.regularized_sdf_projectors) RegularizedSdfPinnsElliptic (class in scimba_torch.geometry.regularized_sdf_projectors) RegularizedSdfProjector (class in scimba_torch.geometry.regularized_sdf_projectors) ReLUk (class in scimba_torch.neural_nets.coordinates_based_nets.activation) repeat() (LabelTensor method) rescale_samples() (ParametricSampler method) residual() (NeuralGalerkin method) restrict_to_labels() (MultiLabelTensor method) rhs() (AdvectionReactionDiffusion method) (AdvectionReactionDiffusion1DDirichletStrongForm method) (AdvectionReactionDiffusion2DDirichletStrongForm method) (CollisionalKineticPDE method) (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongForm method) (HeatEquation2DStrongFormImplicit method) (KineticPDE method) (Laplacian1DDirichletStrongForm method) (Laplacian1DNeumannStrongForm method) (Laplacian2DDirichletStrongForm method) (LinearOrder2PDE method) (RadiativeTransfer method) (RegEikonalPDE method) (StrongFormEllipticPDE method) (TemporalPDE method) (Transport1D method) (Transport1DImplicit method) (Vlasov method)

ridge (AbstractNonlinearProjector attribute) RitzFormEllipticPDE (class in scimba_torch.physical_models.elliptic_pde.abstract_elliptic_pde) Rk2Flow (class in scimba_torch.flows.discretization_based_flows) rk4() (in module scimba_torch.flows.integrators_ode) rk4_symplectic() (in module scimba_torch.flows.integrators_ode) Rk4Flow (class in scimba_torch.flows.discretization_based_flows) rollout (DiscreteFlowSpace attribute) rot_2d() (Mapping static method), [1] rot_3d() (Mapping static method), [1]

S

s (FactorizedLinear attribute) sample() (DomainSampler method) (NumpyUniform method) (ParametricHyperSurface method) (RectangleMethod method) (SurfacicSampler method) (TensorizedSampler method) (UniformTimeSampler method) (UniformVelocitySampler method) (UniformVelocitySamplerOnCuboid method) (VolumetricSampler method) sample_all_vars() (AbstractNonlinearProjector method) (CollocationProjector method) (DeepRitzElliptic method) (FlowTrainer method) (PinnsElliptic method) (TemporalPinns method) (TimeDiscreteCollocationProjector method) sample_from_pre_sampled_points() (ParametricSampler method) sample_new_points() (DomainSampler method) (ParametricSampler method) (SobolParametricSampler method) (UniformParametricSampler method) sample_postmap() (VolumetricSampler method) sample_pre_sampled_points() (DomainSampler method) sampling() (NeuralGalerkin method) save() (AbstractNonlinearProjector method) scalar_ones() (in module scimba_torch.physical_models.elliptic_pde.advection_diffusion) scalar_zero() (in module scimba_torch.physical_models.elliptic_pde.advection_diffusion) (in module scimba_torch.physical_models.kinetic_pde.vlasov) scalar_zeros() (in module scimba_torch.physical_models.temporal_pde.advection_diffusion_equation) scaling (GradPotentialSymplecticLayer attribute) (PeriodicGradPotentialSymplecticLayer attribute) scheduler (AbstractScimbaOptimizer attribute) scheduler_exists (AbstractScimbaOptimizer attribute) scheme (DiscretePINN attribute) scimba_torch.approximation_space module scimba_torch.approximation_space.abstract_space module scimba_torch.approximation_space.kernelx_space module scimba_torch.approximation_space.nn_space module scimba_torch.approximation_space.radial_space module scimba_torch.approximation_space.spectral_space module scimba_torch.domain module scimba_torch.domain.mesh_based_domain module scimba_torch.domain.mesh_based_domain.cuboid module scimba_torch.domain.meshless_domain module scimba_torch.domain.meshless_domain.base module scimba_torch.domain.meshless_domain.domain_1d module scimba_torch.domain.meshless_domain.domain_2d module scimba_torch.domain.meshless_domain.domain_3d module scimba_torch.domain.meshless_domain.domain_nd module scimba_torch.domain.sdf module scimba_torch.flows module scimba_torch.flows.create_solution module scimba_torch.flows.deep_flows module scimba_torch.flows.discretization_based_flows module scimba_torch.flows.flow_trainer module scimba_torch.flows.integrators_ode module scimba_torch.geometry module scimba_torch.geometry.monte_carlo_hypersurface module scimba_torch.geometry.parametric_hypersurface module scimba_torch.geometry.regularized_eikonal_pde module scimba_torch.geometry.regularized_sdf_projectors module scimba_torch.geometry.utils module scimba_torch.integration module scimba_torch.integration.mesh_based_quadrature module scimba_torch.integration.monte_carlo module scimba_torch.integration.monte_carlo_parameters module scimba_torch.integration.monte_carlo_time module scimba_torch.neural_nets module scimba_torch.neural_nets.coordinates_based_nets module scimba_torch.neural_nets.coordinates_based_nets.activation module scimba_torch.neural_nets.coordinates_based_nets.discontinuous_mlp module scimba_torch.neural_nets.coordinates_based_nets.features module scimba_torch.neural_nets.coordinates_based_nets.mlp module scimba_torch.neural_nets.coordinates_based_nets.pirate_net module scimba_torch.neural_nets.coordinates_based_nets.res_net module scimba_torch.neural_nets.coordinates_based_nets.scimba_module module scimba_torch.neural_nets.coordinates_based_nets.siren module scimba_torch.neural_nets.embeddings module scimba_torch.neural_nets.embeddings.periodic_embedding module scimba_torch.neural_nets.structure_preserving_nets module scimba_torch.neural_nets.structure_preserving_nets.affine_ode_layers module scimba_torch.neural_nets.structure_preserving_nets.coupling_layers module scimba_torch.neural_nets.structure_preserving_nets.coupling_symplectic_layers module scimba_torch.neural_nets.structure_preserving_nets.invertible_nn module scimba_torch.neural_nets.structure_preserving_nets.nilpotent_symplectic_layer module scimba_torch.neural_nets.structure_preserving_nets.nonlinear_ode_layers module scimba_torch.neural_nets.structure_preserving_nets.ode_splitted_layer module scimba_torch.neural_nets.structure_preserving_nets.separated_symplectic_layers module scimba_torch.neural_nets.structure_preserving_nets.split_layer module scimba_torch.neural_nets.structure_preserving_nets.symplectic_nets module scimba_torch.numerical_solvers module scimba_torch.numerical_solvers.abstract_preconditioner module scimba_torch.numerical_solvers.abstract_projector module scimba_torch.numerical_solvers.collocation_projector module scimba_torch.numerical_solvers.elliptic_pde module scimba_torch.numerical_solvers.elliptic_pde.deep_ritz module scimba_torch.numerical_solvers.elliptic_pde.pinns module scimba_torch.numerical_solvers.functional_operator module scimba_torch.numerical_solvers.pinn_preconditioners module scimba_torch.numerical_solvers.pinn_preconditioners.anagram_ng module scimba_torch.numerical_solvers.pinn_preconditioners.energy_ng module scimba_torch.numerical_solvers.pinn_preconditioners.nystrom_ng module scimba_torch.numerical_solvers.pinn_preconditioners.sketchy_ng module scimba_torch.numerical_solvers.preconditioner_deep_ritz module scimba_torch.numerical_solvers.preconditioner_pinns module scimba_torch.numerical_solvers.preconditioner_projector module scimba_torch.numerical_solvers.preconditioner_solvers module scimba_torch.numerical_solvers.temporal_pde module

scimba_torch.numerical_solvers.temporal_pde.discrete_pinn module scimba_torch.numerical_solvers.temporal_pde.neural_galerkin module scimba_torch.numerical_solvers.temporal_pde.neural_semilagrangian module scimba_torch.numerical_solvers.temporal_pde.pinns module scimba_torch.numerical_solvers.temporal_pde.time_discrete module scimba_torch.optimizers module scimba_torch.optimizers.line_search module scimba_torch.optimizers.losses module scimba_torch.optimizers.optimizers_data module scimba_torch.optimizers.scimba_optimizers module scimba_torch.optimizers.ssbroyden module scimba_torch.physical_models module scimba_torch.physical_models.elliptic_pde module scimba_torch.physical_models.elliptic_pde.abstract_elliptic_pde module scimba_torch.physical_models.elliptic_pde.advection_diffusion module scimba_torch.physical_models.elliptic_pde.laplacians module scimba_torch.physical_models.elliptic_pde.linear_order_2 module scimba_torch.physical_models.kinetic_pde module scimba_torch.physical_models.kinetic_pde.abstract_kinetic_pde module scimba_torch.physical_models.kinetic_pde.radiative_transfer module scimba_torch.physical_models.kinetic_pde.vlasov module scimba_torch.physical_models.temporal_pde module scimba_torch.physical_models.temporal_pde.abstract_temporal_pde module scimba_torch.physical_models.temporal_pde.advection_diffusion_equation module scimba_torch.physical_models.temporal_pde.heat_equation module scimba_torch.physical_models.temporal_pde.transport_equation module scimba_torch.plots module scimba_torch.plots.plot_regularized_sdf_projector module scimba_torch.plots.plot_time_discrete_scheme module scimba_torch.plots.plots_nd module scimba_torch.utils module scimba_torch.utils.environment module scimba_torch.utils.mapping module scimba_torch.utils.paths module scimba_torch.utils.scimba_tensors module scimba_torch.utils.typing_protocols module scimba_torch.utils.verbosity module ScimbaAdam (class in scimba_torch.optimizers.scimba_optimizers) ScimbaCustomOptomizer (class in scimba_torch.optimizers.scimba_optimizers) ScimbaLBFGS (class in scimba_torch.optimizers.scimba_optimizers) ScimbaModule (class in scimba_torch.neural_nets.coordinates_based_nets.scimba_module) ScimbaMomentum (class in scimba_torch.optimizers.scimba_optimizers) ScimbaSGD (class in scimba_torch.optimizers.scimba_optimizers) ScimbaSSBFGS (class in scimba_torch.optimizers.scimba_optimizers) ScimbaSSBroyden (class in scimba_torch.optimizers.scimba_optimizers) SecondOrderTemporalPDE (class in scimba_torch.physical_models.temporal_pde.abstract_temporal_pde) segment() (Mapping static method), [1] Segment1D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_1d) Segment2D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_2d) SeparatedNNxSpace (class in scimba_torch.approximation_space.nn_space) (class in scimba_torch.approximation_space.radial_space) SeparatedSpectralxSpace (class in scimba_torch.approximation_space.spectral_space) set_dof() (AbstractApproxSpace method) (ContinuousFlowSpace method) (DiscreteFlowSpace method) (KernelxSpace method) (NNxSpace method), [1] (NNxSpaceModulation method) (NNxSpaceSplit method), [1] (NNxtSpace method) (NNxvSpace method) (SeparatedNNxSpace method), [1] (SeparatedSpectralxSpace method) (SpectralxSpace method) set_gradient() (AbstractNonlinearProjector method) set_history() (GenericLoss method) set_lr() (OptimizerData method) set_mapping() (VolumetricDomain method) set_new_bounds() (ParametricSampler method) set_parameters() (GenericFeatureNet method) (ScimbaModule method) set_rhs() (CollocationProjector method) set_threshold() (VolumetricDomain method) set_verbosity() (in module scimba_torch.utils.verbosity) shape (MultiLabelTensor attribute) shape() (LabelTensor method) Sigmoid (class in scimba_torch.neural_nets.coordinates_based_nets.activation) SignedDistance (class in scimba_torch.domain.sdf) SiLU (class in scimba_torch.neural_nets.coordinates_based_nets.activation) Sine (class in scimba_torch.neural_nets.coordinates_based_nets.activation) SirenLayer (class in scimba_torch.neural_nets.coordinates_based_nets.siren) SirenNet (class in scimba_torch.neural_nets.coordinates_based_nets.siren) size (ContinuousFlowSpace attribute) (MultiLabelTensor attribute) SketchyNaturalGradientPinnsElliptic (class in scimba_torch.numerical_solvers.elliptic_pde.pinns) SketchyNaturalGradientPreconditioner (class in scimba_torch.numerical_solvers.pinn_preconditioners) (class in scimba_torch.numerical_solvers.pinn_preconditioners.sketchy_ng) SketchyNaturalGradientTemporalPinns (class in scimba_torch.numerical_solvers.temporal_pde.pinns) SobolParametricSampler (class in scimba_torch.integration.monte_carlo_parameters) Softmax (class in scimba_torch.neural_nets.coordinates_based_nets.activation) Softplus (class in scimba_torch.neural_nets.coordinates_based_nets.activation) solution_to_training_format() (in module scimba_torch.flows.create_solution) solve() (AbstractNonlinearProjector method) (ExplicitTimeDiscreteScheme method) solve_linearstep() (AbstractNonlinearProjector method) solve_nnstep() (AbstractNonlinearProjector method) space (FlowTrainer attribute) (LinearOrder2PDE attribute) space_operator() (AdvectionReactionDiffusion method) (CollisionalKineticPDE method) (HeatEquation1DDirichletStrongForm method) (HeatEquation1DStrongForm method) (HeatEquation2DStrongForm method) (HeatEquation2DStrongFormImplicit method) (KineticPDE method) (RadiativeTransfer method) (TemporalPDE method) (Transport1D method) (Transport1DImplicit method) (Vlasov method) spatial_dim (LinearOrder2PDE attribute) spatial_domain (NNxSpace attribute), [1] (NNxSpaceModulation attribute) (NNxSpaceSplit attribute), [1] (NNxtSpace attribute) (NNxvSpace attribute) (SeparatedNNxSpace attribute), [1] SpectralxSpace (class in scimba_torch.approximation_space.spectral_space) sphere() (Mapping static method), [1] Sphere3D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d) split() (SplittingLayer method) split_label_tensors() (FunctionalOperator method) split_label_tensors_to_dict_of_tensors_with_indexes() (in module scimba_torch.numerical_solvers.functional_operator) split_latent_space (NNxSpaceSplit attribute) SplittingLayer (class in scimba_torch.neural_nets.structure_preserving_nets.split_layer) square() (Mapping static method), [1] Square2D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_2d) Square3D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d) SSBroyden (class in scimba_torch.optimizers.ssbroyden) step() (OptimizerData method) (ScimbaCustomOptomizer method) (ScimbaMomentum method) (SSBroyden method) StrongFormEllipticPDE (class in scimba_torch.physical_models.elliptic_pde.abstract_elliptic_pde) surface_o_mapping (SurfacicDomain attribute) surface_torus_3d() (Mapping static method), [1] SurfaceTorus3D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d) SurfaceTorusFrom2DSurface (class in scimba_torch.domain.meshless_domain.domain_3d) SurfacicDomain (class in scimba_torch.domain.meshless_domain.base) SurfacicSampler (class in scimba_torch.integration.monte_carlo) Swish (class in scimba_torch.neural_nets.coordinates_based_nets.activation) SymplecticEulerFlowSep (class in scimba_torch.flows.discretization_based_flows)

T

Tanh (class in scimba_torch.neural_nets.coordinates_based_nets.activation) TemporalPDE (class in scimba_torch.physical_models.temporal_pde.abstract_temporal_pde) TemporalPinns (class in scimba_torch.numerical_solvers.temporal_pde.pinns) TensorizedSampler (class in scimba_torch.integration.monte_carlo) test_activate_next_optimizer() (OptimizerData method) test_and_activate_next_optimizer() (OptimizerData method) time_operator() (FirstOrderTemporalPDE method) (KineticPDE method) (SecondOrderTemporalPDE method) (TemporalPDE method) (Transport1D method) TimeDiscreteAnagramProjector (class in scimba_torch.numerical_solvers.temporal_pde.time_discrete) TimeDiscreteCollocationProjector (class in scimba_torch.numerical_solvers.temporal_pde.time_discrete) TimeDiscreteImplicitNaturalGradientProjector (class in scimba_torch.numerical_solvers.temporal_pde.time_discrete) TimeDiscreteNaturalGradientProjector (class in scimba_torch.numerical_solvers.temporal_pde.time_discrete) to_dim (Mapping attribute), [1] to_volumetric_domain() (Cuboid method)

Torus3D (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d) TorusFrom2DVolume (class in scimba_torch.domain.meshless_domain) (class in scimba_torch.domain.meshless_domain.domain_3d) translate() (Mapping static method), [1] Transport1D (class in scimba_torch.physical_models.temporal_pde.transport_equation) Transport1DImplicit (class in scimba_torch.physical_models.temporal_pde.transport_equation) try_to_load() (GenericLosses method) TYPE_FUNC_ARGS (in module scimba_torch.utils.typing_protocols) TYPE_FUNC_FUNC_ARGS (in module scimba_torch.utils.typing_protocols) type_linesearch (AbstractNonlinearProjector attribute) (NaturalGradientFlowTrainer attribute) type_projection (AbstractNonlinearProjector attribute) type_space (AbstractApproxSpace attribute) (DiscreteFlowSpace attribute) (NNxSpace attribute) (NNxSpaceModulation attribute)

U

u_func() (SymplecticEulerFlowSep method) uniform_mesh() (Cuboid method) UniformParametricSampler (class in scimba_torch.integration.monte_carlo_parameters) UniformTimeSampler (class in scimba_torch.integration.monte_carlo_time) UniformVelocitySampler (class in scimba_torch.integration.monte_carlo_parameters) UniformVelocitySamplerOnCuboid (class in scimba_torch.integration.monte_carlo_parameters) unsplit() (SplittingLayer method) unsqueeze() (LabelTensor method) update() (DiscretePINN method) (DiscretePINNImplicit method) (ExplicitTimeDiscreteScheme method) (NeuralGalerkin method) (NeuralSemiLagrangian method)

update_best_approx() (AbstractApproxSpace method) update_best_optimizer() (AbstractScimbaOptimizer method) (OptimizerData method) update_coeff() (GenericLoss method) (GenericLosses method) update_histories() (GenericLosses method) update_history() (GenericLoss method) update_implicit() (DiscretePINNImplicit method) update_loss() (GenericLoss method) (GenericLosses method) update_parameter_bounds() (CollocationProjector method)

V

v (FactorizedLinear attribute) VarArgAnyCallable (class in scimba_torch.utils.typing_protocols) VarArgCallable (class in scimba_torch.utils.typing_protocols) vect_x_to_xi() (PolygonalApproxSignedDistance method) vector_ones() (in module scimba_torch.physical_models.elliptic_pde.advection_diffusion) (in module scimba_torch.physical_models.temporal_pde.advection_diffusion_equation) vectorial_product() (PolygonalApproxSignedDistance method) vectorize_along_physical_variables() (MatrixPreconditionerDeepRitz method) (MatrixPreconditionerSolver method)

vectorize_along_physical_variables_bc() (MatrixPreconditionerSolver method) vectorize_along_physical_variables_ic() (MatrixPreconditionerSolver method) vectorize_dict_of_func() (in module scimba_torch.numerical_solvers.functional_operator) velocity_domain (NNxvSpace attribute) verlet_explicit() (in module scimba_torch.flows.integrators_ode) verlet_implicit() (in module scimba_torch.flows.integrators_ode) VerletSymplecticEulerFlow (class in scimba_torch.flows.discretization_based_flows) Vlasov (class in scimba_torch.physical_models.kinetic_pde.vlasov) VolumetricDomain (class in scimba_torch.domain.meshless_domain.base) VolumetricSampler (class in scimba_torch.integration.monte_carlo)

W

w (MultiLabelTensor attribute) W (PSymplecticLayer attribute) W_f (PirateNetBlock attribute) W_g (PirateNetBlock attribute) W_h (PirateNetBlock attribute)

W_mu (PSymplecticLayer attribute) Wavelet (class in scimba_torch.neural_nets.coordinates_based_nets.activation) WeakFormEllipticPDE (class in scimba_torch.physical_models.elliptic_pde.abstract_elliptic_pde) weight (PeriodicEmbedding attribute) weighted_loss (GenericLoss attribute) write_points_normals_to_file() (in module scimba_torch.geometry.utils)

X

x (LabelTensor attribute)

x_size (NNxSpaceModulation attribute)

Y

y (FlowTrainer attribute)

y_size (NNxSpaceModulation attribute)

Z

zero_grad() (OptimizerData method) zeros() (in module scimba_torch.physical_models.elliptic_pde.laplacians) zeros_bc_rhs() (in module scimba_torch.physical_models.temporal_pde.heat_equation) (in module scimba_torch.physical_models.temporal_pde.transport_equation)

zeros_rhs() (in module scimba_torch.physical_models.temporal_pde.heat_equation) (in module scimba_torch.physical_models.temporal_pde.transport_equation)