Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Y | Z A A2GradExpConf (class in physicsnemo.sym.hydra.optimizer) A2GradIncConf (class in physicsnemo.sym.hydra.optimizer) A2GradUniConf (class in physicsnemo.sym.hydra.optimizer) acc() (in module physicsnemo.metrics.climate.acc) AccSGDConf (class in physicsnemo.sym.hydra.optimizer) activation_fn (physicsnemo.sym.hydra.arch.ConvFullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FNOConf attribute) (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.FullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FusedFourierNetConf attribute) (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.FusedMLPConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) (physicsnemo.sym.hydra.arch.SRResConf attribute) AdaBeliefConf (class in physicsnemo.sym.hydra.optimizer) AdaBoundConf (class in physicsnemo.sym.hydra.optimizer) AdadeltaConf (class in physicsnemo.sym.hydra.optimizer) AdafactorConf (class in physicsnemo.sym.hydra.optimizer) AdagradConf (class in physicsnemo.sym.hydra.optimizer) AdahessianConf (class in physicsnemo.sym.hydra.optimizer) AdaHessianMixin (class in physicsnemo.sym.trainer) AdamaxConf (class in physicsnemo.sym.hydra.optimizer) AdamConf (class in physicsnemo.sym.hydra.optimizer) AdamMixin (class in physicsnemo.sym.trainer) AdaModConf (class in physicsnemo.sym.hydra.optimizer) AdamPConf (class in physicsnemo.sym.hydra.optimizer) AdamWConf (class in physicsnemo.sym.hydra.optimizer) adaptive_activations (physicsnemo.sym.hydra.arch.ConvFullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.FullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) add_edge_features() (in module physicsnemo.utils.graphcast.graph_utils) (physicsnemo.datapipes.gnn.ahmed_body_dataset.AhmedBodyDataset method) (physicsnemo.datapipes.gnn.stokes_dataset.StokesDataset method) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) add_hydra_run_path() (in module physicsnemo.sym.hydra.utils) add_node_features() (in module physicsnemo.utils.graphcast.graph_utils) add_pino_gradients() (physicsnemo.sym.models.fno.FNOArch method) ADF (class in physicsnemo.sym.geometry.adf) AdvectionDiffusion (class in physicsnemo.sym.eq.pdes.advection_diffusion) AFNO (class in physicsnemo.models.afno.afno) AFNO2DLayer (class in physicsnemo.models.afno.afno) AFNOArch (class in physicsnemo.sym.models.afno) AFNOConf (class in physicsnemo.sym.hydra.arch) AFNOMlp (class in physicsnemo.models.afno.afno) AggMoConf (class in physicsnemo.sym.hydra.optimizer) AggregationModel (class in physicsnemo.models.domino.model) Aggregator (class in physicsnemo.sym.loss.aggregator) AggregatorGradNormConf (class in physicsnemo.sym.hydra.loss) AggregatorHomoscedasticConf (class in physicsnemo.sym.hydra.loss) AggregatorLRAnnealingConf (class in physicsnemo.sym.hydra.loss) AggregatorRelobraloConf (class in physicsnemo.sym.hydra.loss) AggregatorResNormConf (class in physicsnemo.sym.hydra.loss) AggregatorSoftAdaptConf (class in physicsnemo.sym.hydra.loss) AggregatorSumConf (class in physicsnemo.sym.hydra.loss) AhmedBodyDataset (class in physicsnemo.datapipes.gnn.ahmed_body_dataset) alert() (in module physicsnemo.launch.logging.wandb) all_gather_v() (in module physicsnemo.distributed.autograd) all_gather_v_bwd_wrapper() (in module physicsnemo.distributed.utils) all_gather_v_wrapper() (in module physicsnemo.distributed.utils) AllGatherVAutograd (class in physicsnemo.distributed.autograd) alpha (physicsnemo.sym.hydra.loss.AggregatorGradNormConf attribute) (physicsnemo.sym.hydra.loss.AggregatorLRAnnealingConf attribute) (physicsnemo.sym.hydra.loss.AggregatorRelobraloConf attribute) (physicsnemo.sym.hydra.loss.AggregatorResNormConf attribute) (physicsnemo.sym.hydra.optimizer.ASGDConf attribute) (physicsnemo.sym.hydra.optimizer.RangerConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) (physicsnemo.sym.hydra.optimizer.RMSpropConf attribute) alpha_bar() (physicsnemo.models.diffusion.preconditioning.iDDPMPrecond method) amp_mode (physicsnemo.models.diffusion.dhariwal_unet.DhariwalUNet property) (physicsnemo.models.diffusion.preconditioning.EDMPrecondSuperResolution property) (physicsnemo.models.diffusion.song_unet.SongUNet property) (physicsnemo.models.diffusion.song_unet.SongUNetPosEmbd property) (physicsnemo.models.diffusion.song_unet.SongUNetPosLtEmbd property) (physicsnemo.models.diffusion.unet.UNet property) amsbound (physicsnemo.sym.hydra.optimizer.AdaBoundConf attribute) amsgrad (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) (physicsnemo.sym.hydra.optimizer.AdamConf attribute) (physicsnemo.sym.hydra.optimizer.AdamWConf attribute) (physicsnemo.sym.hydra.optimizer.NovoGradConf attribute) (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) (physicsnemo.sym.hydra.optimizer.SWATSConf attribute) ApolloConf (class in physicsnemo.sym.hydra.optimizer) apply() (physicsnemo.utils.patching.BasePatching2D method) (physicsnemo.utils.patching.GridPatching2D method) (physicsnemo.utils.patching.RandomPatching2D method) approx_area() (physicsnemo.sym.geometry.curve.Curve method) arch (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) arch_type (physicsnemo.sym.hydra.arch.AFNOConf attribute) (physicsnemo.sym.hydra.arch.ConvFullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.DeepOConf attribute) (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) (physicsnemo.sym.hydra.arch.FNOConf attribute) (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.FullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FusedFourierNetConf attribute) (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.FusedMLPConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModelConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) (physicsnemo.sym.hydra.arch.SirenConf attribute) (physicsnemo.sym.hydra.arch.SRResConf attribute) area_weighted_shuffle_array() (in module physicsnemo.utils.domino.utils) array_type() (in module physicsnemo.utils.domino.utils) ASGDConf (class in physicsnemo.sym.hydra.optimizer) assert_shape() (in module physicsnemo.utils.diffusion.utils) assign_atmosphere_weights() (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) assign_surface_weights() (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) assign_variable_weights() (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) azimuthal_angle() (in module physicsnemo.utils.graphcast.graph_utils) B backward() (physicsnemo.distributed.autograd.AllGatherVAutograd static method) (physicsnemo.distributed.autograd.GatherVAutograd static method) (physicsnemo.distributed.autograd.IndexedAllToAllVAutograd static method) (physicsnemo.distributed.autograd.ScatterVAutograd static method) (physicsnemo.distributed.fft.DistributedIRFFT2 static method) (physicsnemo.distributed.fft.DistributedRFFT2 static method) (physicsnemo.distributed.shard_tensor.ShardTensor method) (physicsnemo.sym.loss.loss.LossL2 static method) (physicsnemo.utils.graphcast.loss.CustomCellAreaWeightedLossAutogradFunction static method) base_resolution (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) BasePatching2D (class in physicsnemo.utils.patching) batch_norm (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) batch_size (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) beta (physicsnemo.sym.hydra.loss.AggregatorRelobraloConf attribute) (physicsnemo.sym.hydra.optimizer.A2GradExpConf attribute) (physicsnemo.sym.hydra.optimizer.A2GradIncConf attribute) (physicsnemo.sym.hydra.optimizer.A2GradUniConf attribute) (physicsnemo.sym.hydra.optimizer.ApolloConf attribute) beta1 (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) beta3 (physicsnemo.sym.hydra.optimizer.AdaModConf attribute) betas (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) (physicsnemo.sym.hydra.optimizer.AdaBoundConf attribute) (physicsnemo.sym.hydra.optimizer.AdahessianConf attribute) (physicsnemo.sym.hydra.optimizer.AdamaxConf attribute) (physicsnemo.sym.hydra.optimizer.AdamConf attribute) (physicsnemo.sym.hydra.optimizer.AdaModConf attribute) (physicsnemo.sym.hydra.optimizer.AdamPConf attribute) (physicsnemo.sym.hydra.optimizer.AdamWConf attribute) (physicsnemo.sym.hydra.optimizer.AggMoConf attribute) (physicsnemo.sym.hydra.optimizer.DiffGradConf attribute) (physicsnemo.sym.hydra.optimizer.LambConf attribute) (physicsnemo.sym.hydra.optimizer.NAdamConf attribute) (physicsnemo.sym.hydra.optimizer.NovoGradConf attribute) (physicsnemo.sym.hydra.optimizer.QHAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RangerConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) (physicsnemo.sym.hydra.optimizer.SparseAdamConf attribute) (physicsnemo.sym.hydra.optimizer.SWATSConf attribute) (physicsnemo.sym.hydra.optimizer.YogiConf attribute) BFGSConf (class in physicsnemo.sym.hydra.optimizer) BFGSMixin (class in physicsnemo.sym.trainer) BiStrideMeshGraphNet (class in physicsnemo.models.meshgraphnet.bsms_mgn) Block (class in physicsnemo.models.afno.afno) (class in physicsnemo.models.afno.modafno) bounding_box_dims (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) bounding_box_dims_surf (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) BoundingBox (class in physicsnemo.datapipes.cae.domino_datapipe) Bounds (class in physicsnemo.sym.geometry.parameterization) bounds (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) Box (class in physicsnemo.sym.geometry.primitives_3d) BQWarp (class in physicsnemo.models.domino.model) branch_dim (physicsnemo.sym.hydra.arch.DeepOConf attribute) broadcast_buffers (physicsnemo.distributed.manager.DistributedManager property) (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) build_fno() (physicsnemo.models.fno.fno.FNO1DEncoder method) (physicsnemo.models.fno.fno.FNO2DEncoder method) (physicsnemo.models.fno.fno.FNO3DEncoder method) (physicsnemo.models.fno.fno.FNO4DEncoder method) build_lift_network() (physicsnemo.models.fno.fno.FNO1DEncoder method) (physicsnemo.models.fno.fno.FNO2DEncoder method) (physicsnemo.models.fno.fno.FNO3DEncoder method) (physicsnemo.models.fno.fno.FNO4DEncoder method) C CachedDoMINODataset (class in physicsnemo.datapipes.cae.domino_datapipe) caching (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) calculate_center_of_mass() (in module physicsnemo.utils.domino.utils) calculate_linear_weights() (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) calculate_normal_positional_encoding() (in module physicsnemo.utils.domino.utils) calculate_pos_encoding() (in module physicsnemo.models.domino.model) (in module physicsnemo.utils.domino.utils) calculate_solution() (physicsnemo.models.domino.model.DoMINO method) calculate_solution_with_neighbors() (physicsnemo.models.domino.model.DoMINO method) call_func_by_name() (in module physicsnemo.utils.diffusion.utils) CausalLossNorm (class in physicsnemo.sym.loss.loss) cdf() (in module physicsnemo.metrics.general.histogram) cell_to_adj() (in module physicsnemo.utils.graphcast.graph_utils) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) CellAreaWeightedLossFunction (class in physicsnemo.utils.graphcast.loss) centered (physicsnemo.sym.hydra.optimizer.RMSpropConf attribute) Channel (class in physicsnemo.sym.geometry.primitives_3d) Channel2D (class in physicsnemo.sym.geometry.primitives_2d) channel_parallel_inputs (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) channel_parallel_outputs (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) check_ddp_consistency() (in module physicsnemo.utils.diffusion.utils) check_mlflow_logged_in() (in module physicsnemo.launch.logging.mlflow) check_ort_install() (in module physicsnemo.deploy.onnx.utils) Circle (class in physicsnemo.sym.geometry.primitives_2d) circle_adf() (physicsnemo.sym.geometry.adf.ADF static method) cleanup() (physicsnemo.distributed.manager.DistributedManager static method) ClimateDaliExternalSource (class in physicsnemo.datapipes.climate.climate) ClimateDatapipe (class in physicsnemo.datapipes.climate.climate) ClimateDataSourceSpec (class in physicsnemo.datapipes.climate.climate) ClimateHDF5DaliExternalSource (class in physicsnemo.datapipes.climate.climate) ClimateNetCDF4DaliExternalSource (class in physicsnemo.datapipes.climate.climate) clip_threshold (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) Collectives combine_dict() (in module physicsnemo.utils.domino.utils) compose() (in module physicsnemo.sym.hydra.utils) CompressibleIntegralContinuity (class in physicsnemo.sym.eq.pdes.navier_stokes) compute_connectivity_tensor() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) compute_gradients() (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientCalculator method) compute_scaling_factors (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) compute_stencil2d() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) compute_stencil3d() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) Cone (class in physicsnemo.sym.geometry.primitives_3d) constant() (in module physicsnemo.utils.diffusion.utils) Constraint (class in physicsnemo.sym.domain.constraint.constraint) construct_class_by_name() (in module physicsnemo.utils.diffusion.utils) conv_layer_size (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) (physicsnemo.sym.hydra.arch.SRResConf attribute) convert_config() (physicsnemo.sym.key.Key static method) convert_datetime_to_cftime() (in module physicsnemo.utils.diffusion.utils) convert_point_data_to_cell_data() (in module physicsnemo.utils.domino.utils) convert_to_tet_mesh() (in module physicsnemo.utils.domino.utils) ConvFullyConnectedConf (class in physicsnemo.sym.hydra.arch) ConvolutionalBlock3d (class in physicsnemo.models.srrn.super_res_net) coord_features (physicsnemo.sym.hydra.arch.FNOConf attribute) copy_files_and_create_dirs() (in module physicsnemo.utils.diffusion.utils) copy_params_and_buffers() (in module physicsnemo.utils.diffusion.utils) copy_to_parallel_region() (in module physicsnemo.distributed.mappings) CosineAnnealingLRConf (class in physicsnemo.sym.hydra.scheduler) CosineAnnealingWarmRestartsConf (class in physicsnemo.sym.hydra.scheduler) create_ddp_group_tag() (in module physicsnemo.launch.logging.utils) create_directory() (in module physicsnemo.utils.domino.utils) create_g2m_graph() (physicsnemo.utils.graphcast.graph.Graph method) create_graph() (in module physicsnemo.utils.graphcast.graph_utils) (physicsnemo.datapipes.gnn.ahmed_body_dataset.AhmedBodyDataset method) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) create_grid() (in module physicsnemo.utils.domino.utils) create_heterograph() (in module physicsnemo.utils.graphcast.graph_utils) create_m2g_graph() (physicsnemo.utils.graphcast.graph.Graph method) create_mesh_graph() (physicsnemo.utils.graphcast.graph.Graph method) create_orthogonal_process_group() (physicsnemo.distributed.manager.DistributedManager static method) create_process_subgroup() (physicsnemo.distributed.manager.DistributedManager static method) crps() (in module physicsnemo.metrics.general.crps) csv_to_dict() (in module physicsnemo.sym.utils.io.csv_rw) cuda (physicsnemo.distributed.manager.DistributedManager property) cuda_graph_warmup (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) cuda_graphs (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) Curl (class in physicsnemo.sym.eq.pdes.basic) (class in physicsnemo.sym.eq.pdes.navier_stokes) Curve (class in physicsnemo.sym.geometry.curve) custom (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) custom_forward() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) CustomCellAreaWeightedLossAutogradFunction (class in physicsnemo.utils.graphcast.loss) CustomCellAreaWeightedLossFunction (class in physicsnemo.utils.graphcast.loss) CustomDerivativePrinter (class in physicsnemo.sym.utils.sympy.torch_printer) Cylinder (class in physicsnemo.sym.geometry.primitives_3d) D dampening (physicsnemo.sym.hydra.optimizer.PIDConf attribute) (physicsnemo.sym.hydra.optimizer.SGDConf attribute) (physicsnemo.sym.hydra.optimizer.SGDPConf attribute) (physicsnemo.sym.hydra.optimizer.SGDWConf attribute) Darcy2D (class in physicsnemo.datapipes.benchmarks.darcy) Data Parallel data_path (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) datefmt (physicsnemo.sym.hydra.hydra.DebugFormat attribute) (physicsnemo.sym.hydra.hydra.SimpleFormat attribute) ddp_sync() (in module physicsnemo.utils.diffusion.utils) debug (physicsnemo.sym.hydra.config.DebugPhysicsNeMoConfig attribute) (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) DebugFormat (class in physicsnemo.sym.hydra.hydra) DebugLogging (class in physicsnemo.sym.hydra.hydra) DebugPhysicsNeMoConfig (class in physicsnemo.sym.hydra.config) decay_rate (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) (physicsnemo.sym.hydra.scheduler.TFExponentialLRConf attribute) decay_steps (physicsnemo.sym.hydra.scheduler.TFExponentialLRConf attribute) DecayedIntegralLossNorm (class in physicsnemo.sym.loss.loss) DecayedLossNorm (class in physicsnemo.sym.loss.loss) DecayedPointwiseLossNorm (class in physicsnemo.sym.loss.loss) decode() (physicsnemo.models.mesh_reduced.mesh_reduced.Mesh_Reduced method) decoder_forward() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) decouple_weight_decay (physicsnemo.sym.hydra.optimizer.QHAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) DeepOConf (class in physicsnemo.sym.hydra.arch) DeepONet_Data_Validator (class in physicsnemo.sym.domain.validator.discrete) DeepONet_Physics_Validator (class in physicsnemo.sym.domain.validator.discrete) DeepONetArch (class in physicsnemo.sym.models.deeponet) DeepONetConstraint (class in physicsnemo.sym.domain.constraint.continuous) DeepONetConstraint_Data (class in physicsnemo.sym.domain.constraint.discrete) DeepONetConstraint_Physics (class in physicsnemo.sym.domain.constraint.discrete) DeepONetValidatorPlotter (class in physicsnemo.sym.utils.io.plotter) DeepSpeed DefaultLogging (class in physicsnemo.sym.hydra.hydra) DefaultPhysicsNeMoConfig (class in physicsnemo.sym.hydra.config) defaults (physicsnemo.sym.hydra.config.DebugPhysicsNeMoConfig attribute) (physicsnemo.sym.hydra.config.DefaultPhysicsNeMoConfig attribute) (physicsnemo.sym.hydra.config.ExperimentalPhysicsNeMoConfig attribute) DefaultStopCriterion (class in physicsnemo.sym.hydra.training) DefaultTraining (class in physicsnemo.sym.hydra.training) deg2rad() (in module physicsnemo.utils.graphcast.graph_utils) delta (physicsnemo.sym.hydra.optimizer.AdamPConf attribute) (physicsnemo.sym.hydra.optimizer.SGDPConf attribute) denormalize() (physicsnemo.datapipes.gnn.ahmed_body_dataset.AhmedBodyDataset method) (physicsnemo.datapipes.gnn.drivaernet_dataset.DrivAerNetDataset method) (physicsnemo.datapipes.gnn.stokes_dataset.StokesDataset static method) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) depth (physicsnemo.sym.hydra.arch.AFNOConf attribute) (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) derivative (physicsnemo.sym.hydra.optimizer.PIDConf attribute) derivatives (physicsnemo.sym.node.Node property) detach_keys (physicsnemo.sym.hydra.arch.ModelConf attribute) deterministic (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) deterministic_sampler() (in module physicsnemo.utils.diffusion.deterministic_sampler) device (physicsnemo.distributed.manager.DistributedManager property) (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) DeviceMesh DGMArch (class in physicsnemo.sym.models.dgm) DhariwalUNet (class in physicsnemo.models.diffusion.dhariwal_unet) dict_to_csv() (in module physicsnemo.sym.utils.io.csv_rw) dict_to_device() (in module physicsnemo.utils.domino.utils) DiffGradConf (class in physicsnemo.sym.hydra.optimizer) Diffusion (class in physicsnemo.sym.eq.pdes.diffusion) diffusion_step() (in module physicsnemo.utils.corrdiff.utils) DiffusionInterface (class in physicsnemo.sym.eq.pdes.diffusion) dimension (physicsnemo.sym.hydra.arch.FNOConf attribute) (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) dimensions_compatible() (physicsnemo.datapipes.climate.climate.ClimateDataSourceSpec method) dims (physicsnemo.sym.geometry.curve.Curve property) (physicsnemo.sym.geometry.geometry.Geometry property) (physicsnemo.sym.geometry.parameterization.Bounds property) disable_existing_loggers (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) DiscreteGeometry (class in physicsnemo.sym.geometry.discrete_geometry) distributed (physicsnemo.distributed.manager.DistributedManager property) distributed_transpose() (in module physicsnemo.distributed.utils) DistributedAFNOConf (class in physicsnemo.sym.hydra.arch) DistributedDataParallel DistributedIRFFT2 (class in physicsnemo.distributed.fft) DistributedManager (class in physicsnemo.distributed.manager) DistributedRFFT2 (class in physicsnemo.distributed.fft) DLWP (class in physicsnemo.models.dlwp.dlwp) Domain Parallel DoMINO (class in physicsnemo.models.domino.model) domino_collate_fn() (in module physicsnemo.datapipes.cae.domino_datapipe) DoMINODataConfig (class in physicsnemo.datapipes.cae.domino_datapipe) DoMINODataPipe (class in physicsnemo.datapipes.cae.domino_datapipe) DrivAerNetDataset (class in physicsnemo.datapipes.gnn.drivaernet_dataset) E EasyDict (class in physicsnemo.utils.diffusion.utils) edges_to_adjacency() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) EDMPrecond (class in physicsnemo.models.diffusion.preconditioning) EDMPrecondMetaData (class in physicsnemo.models.diffusion.preconditioning) EDMPrecondSR (class in physicsnemo.models.diffusion.preconditioning) EDMPrecondSRMetaData (class in physicsnemo.models.diffusion.preconditioning) EDMPrecondSuperResolution (class in physicsnemo.models.diffusion.preconditioning) EDMPrecondSuperResolutionMetaData (class in physicsnemo.models.diffusion.preconditioning) efi() (in module physicsnemo.metrics.climate.efi) efi_gaussian() (in module physicsnemo.metrics.climate.efi) ElliCylinder (class in physicsnemo.sym.geometry.primitives_3d) Ellipse (class in physicsnemo.sym.geometry.primitives_2d) embed_dim (physicsnemo.sym.hydra.arch.AFNOConf attribute) (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) encode() (physicsnemo.models.mesh_reduced.mesh_reduced.Mesh_Reduced method) encoder_forward() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) end_step (physicsnemo.sym.hydra.profiler.NvtxProfiler attribute) (physicsnemo.sym.hydra.profiler.ProfilerConf attribute) (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) EnsembleMetrics (class in physicsnemo.metrics.general.ensemble_metrics) entropy_from_counts() (in module physicsnemo.metrics.general.entropy) eps (physicsnemo.sym.hydra.loss.AggregatorLRAnnealingConf attribute) (physicsnemo.sym.hydra.loss.AggregatorRelobraloConf attribute) (physicsnemo.sym.hydra.loss.AggregatorSoftAdaptConf attribute) (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) (physicsnemo.sym.hydra.optimizer.AdaBoundConf attribute) (physicsnemo.sym.hydra.optimizer.AdadeltaConf attribute) (physicsnemo.sym.hydra.optimizer.AdagradConf attribute) (physicsnemo.sym.hydra.optimizer.AdahessianConf attribute) (physicsnemo.sym.hydra.optimizer.AdamaxConf attribute) (physicsnemo.sym.hydra.optimizer.AdamConf attribute) (physicsnemo.sym.hydra.optimizer.AdaModConf attribute) (physicsnemo.sym.hydra.optimizer.AdamPConf attribute) (physicsnemo.sym.hydra.optimizer.AdamWConf attribute) (physicsnemo.sym.hydra.optimizer.ApolloConf attribute) (physicsnemo.sym.hydra.optimizer.DiffGradConf attribute) (physicsnemo.sym.hydra.optimizer.LambConf attribute) (physicsnemo.sym.hydra.optimizer.MADGRADConf attribute) (physicsnemo.sym.hydra.optimizer.NAdamConf attribute) (physicsnemo.sym.hydra.optimizer.NovoGradConf attribute) (physicsnemo.sym.hydra.optimizer.QHAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RangerConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) (physicsnemo.sym.hydra.optimizer.RMSpropConf attribute) (physicsnemo.sym.hydra.optimizer.SparseAdamConf attribute) (physicsnemo.sym.hydra.optimizer.SWATSConf attribute) (physicsnemo.sym.hydra.optimizer.YogiConf attribute) eps2 (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) epsilon (physicsnemo.sym.hydra.optimizer.ShampooConf attribute) ERA5DaliExternalSource (class in physicsnemo.datapipes.climate.era5_hdf5) ERA5HDF5Datapipe (class in physicsnemo.datapipes.climate.era5_hdf5) error() (physicsnemo.launch.logging.console.PythonLogger method) eta_min (physicsnemo.sym.hydra.scheduler.CosineAnnealingLRConf attribute) (physicsnemo.sym.hydra.scheduler.CosineAnnealingWarmRestartsConf attribute) etas (physicsnemo.sym.hydra.optimizer.RpropConf attribute) ExperimentalPhysicsNeMoConfig (class in physicsnemo.sym.hydra.config) ExponentialLRConf (class in physicsnemo.sym.hydra.scheduler) export_to_onnx_stream() (in module physicsnemo.deploy.onnx.utils) extract_surface_triangles() (in module physicsnemo.utils.domino.utils) F Fengwu (class in physicsnemo.models.fengwu.fengwu) file_logging() (physicsnemo.launch.logging.console.PythonLogger method) FileInfo (class in physicsnemo.datapipes.gnn.ahmed_body_dataset) filename (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) filter_type (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) final_lr (physicsnemo.sym.hydra.optimizer.AdaBoundConf attribute) finalize() (physicsnemo.metrics.general.ensemble_metrics.EnsembleMetrics method) (physicsnemo.metrics.general.ensemble_metrics.Mean method) (physicsnemo.metrics.general.ensemble_metrics.Variance method) (physicsnemo.metrics.general.histogram.Histogram method) find_rank() (in module physicsnemo.metrics.general.calibration) find_unused_parameters (physicsnemo.distributed.manager.DistributedManager property) (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) finest_resolution (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) first_omega (physicsnemo.sym.hydra.arch.SirenConf attribute) fixed_decay (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) FluxContinuity (class in physicsnemo.sym.eq.pdes.navier_stokes) FNO (class in physicsnemo.models.fno.fno) FNO1DEncoder (class in physicsnemo.models.fno.fno) FNO2DEncoder (class in physicsnemo.models.fno.fno) FNO3DEncoder (class in physicsnemo.models.fno.fno) FNO4DEncoder (class in physicsnemo.models.fno.fno) fno_modes (physicsnemo.sym.hydra.arch.FNOConf attribute) FNOArch (class in physicsnemo.sym.models.fno) FNOConf (class in physicsnemo.sym.hydra.arch) format (physicsnemo.sym.hydra.hydra.DebugFormat attribute) (physicsnemo.sym.hydra.hydra.SimpleFormat attribute) format_time() (in module physicsnemo.utils.diffusion.utils) format_time_brief() (in module physicsnemo.utils.diffusion.utils) formatters (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) forward() (physicsnemo.distributed.autograd.AllGatherVAutograd static method) (physicsnemo.distributed.autograd.GatherVAutograd static method) (physicsnemo.distributed.autograd.IndexedAllToAllVAutograd static method) (physicsnemo.distributed.autograd.ScatterVAutograd static method) (physicsnemo.distributed.fft.DistributedIRFFT2 static method) (physicsnemo.distributed.fft.DistributedRFFT2 static method) (physicsnemo.models.afno.afno.AFNO method) (physicsnemo.models.afno.afno.AFNO2DLayer method) (physicsnemo.models.afno.afno.AFNOMlp method) (physicsnemo.models.afno.afno.Block method) (physicsnemo.models.afno.afno.PatchEmbed method) (physicsnemo.models.afno.modafno.Block method) (physicsnemo.models.afno.modafno.ModAFNO method) (physicsnemo.models.afno.modafno.ModAFNO2DLayer method) (physicsnemo.models.afno.modafno.ModAFNOMlp method) (physicsnemo.models.afno.modafno.ScaleShiftMlp method) (physicsnemo.models.diffusion.preconditioning.EDMPrecond method) (physicsnemo.models.diffusion.preconditioning.EDMPrecondSR method) (physicsnemo.models.diffusion.preconditioning.EDMPrecondSuperResolution method) (physicsnemo.models.diffusion.preconditioning.iDDPMPrecond method) (physicsnemo.models.diffusion.preconditioning.VEPrecond method) (physicsnemo.models.diffusion.preconditioning.VEPrecond_dfsr method) (physicsnemo.models.diffusion.preconditioning.VEPrecond_dfsr_cond method) (physicsnemo.models.diffusion.preconditioning.VPPrecond method) (physicsnemo.models.dlwp.dlwp.DLWP method) (physicsnemo.models.dlwp_healpix.HEALPixRecUNet.HEALPixRecUNet method) (physicsnemo.models.domino.model.AggregationModel method) (physicsnemo.models.domino.model.BQWarp method) (physicsnemo.models.domino.model.DoMINO method) (physicsnemo.models.domino.model.GeoConvOut method) (physicsnemo.models.domino.model.GeometryRep method) (physicsnemo.models.domino.model.GeoProcessor method) (physicsnemo.models.domino.model.LocalPointConv method) (physicsnemo.models.domino.model.NNBasisFunctions method) (physicsnemo.models.domino.model.ParameterModel method) (physicsnemo.models.domino.model.PositionEncoder method) (physicsnemo.models.fengwu.fengwu.Fengwu method) (physicsnemo.models.fno.fno.FNO method) (physicsnemo.models.fno.fno.FNO1DEncoder method) (physicsnemo.models.fno.fno.FNO2DEncoder method) (physicsnemo.models.fno.fno.FNO3DEncoder method) (physicsnemo.models.fno.fno.FNO4DEncoder method) (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) (physicsnemo.models.meshgraphnet.bsms_mgn.BiStrideMeshGraphNet method) (physicsnemo.models.meshgraphnet.meshgraphnet.MeshGraphNet method) (physicsnemo.models.meshgraphnet.meshgraphnet.MeshGraphNetProcessor method) (physicsnemo.models.mlp.fully_connected.FullyConnected method) (physicsnemo.models.pangu.pangu.Pangu method) (physicsnemo.models.pix2pix.pix2pix.Pix2Pix method) (physicsnemo.models.pix2pix.pix2pix.ResnetBlock method) (physicsnemo.models.rnn.rnn_one2many.One2ManyRNN method) (physicsnemo.models.rnn.rnn_seq2seq.Seq2SeqRNN method) (physicsnemo.models.srrn.super_res_net.ConvolutionalBlock3d method) (physicsnemo.models.srrn.super_res_net.PixelShuffle3d method) (physicsnemo.models.srrn.super_res_net.ResidualConvBlock3d method) (physicsnemo.models.srrn.super_res_net.SRResNet method) (physicsnemo.models.srrn.super_res_net.SubPixel_ConvolutionalBlock3d method) (physicsnemo.models.swinvrnn.swinvrnn.SwinRNN method) (physicsnemo.sym.eq.phy_informer.PhysicsInformer method) (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientsAutoDiff method) (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientsFiniteDifference method) (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientsLeastSquares method) (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientsMeshlessFiniteDifference method) (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientsSpectral method) (physicsnemo.sym.geometry.adf.ADF method) (physicsnemo.sym.graph.Graph method) (physicsnemo.sym.loss.aggregator.GradNorm method) (physicsnemo.sym.loss.aggregator.HomoscedasticUncertainty method) (physicsnemo.sym.loss.aggregator.LRAnnealing method) (physicsnemo.sym.loss.aggregator.NTK method) (physicsnemo.sym.loss.aggregator.Relobralo method) (physicsnemo.sym.loss.aggregator.ResNorm method) (physicsnemo.sym.loss.aggregator.SoftAdapt method) (physicsnemo.sym.loss.aggregator.Sum method) (physicsnemo.sym.loss.loss.CausalLossNorm method) (physicsnemo.sym.loss.loss.DecayedIntegralLossNorm method) (physicsnemo.sym.loss.loss.DecayedPointwiseLossNorm method) (physicsnemo.sym.loss.loss.IntegralLossNorm method) (physicsnemo.sym.loss.loss.Loss method) (physicsnemo.sym.loss.loss.LossL2 static method) (physicsnemo.sym.loss.loss.PointwiseLossNorm method) (physicsnemo.sym.models.afno.AFNOArch method) (physicsnemo.sym.models.deeponet.DeepONetArch method) (physicsnemo.sym.models.dgm.DGMArch method) (physicsnemo.sym.models.fno.FNOArch method) (physicsnemo.sym.models.fourier_net.FourierNetArch method) (physicsnemo.sym.models.fully_connected.FullyConnectedArch method) (physicsnemo.sym.models.hash_encoding_net.MultiresolutionHashNetArch method) (physicsnemo.sym.models.highway_fourier_net.HighwayFourierNetArch method) (physicsnemo.sym.models.modified_fourier_net.ModifiedFourierNetArch method) (physicsnemo.sym.models.moving_time_window.MovingTimeWindowArch method) (physicsnemo.sym.models.multiplicative_filter_net.MultiplicativeFilterNetArch method) (physicsnemo.sym.models.multiscale_fourier_net.MultiscaleFourierNetArch method) (physicsnemo.sym.models.pix2pix.Pix2PixArch method) (physicsnemo.sym.models.radial_basis.RadialBasisArch method) (physicsnemo.sym.models.siren.SirenArch method) (physicsnemo.sym.models.super_res_net.SRResNetArch method) (physicsnemo.sym.utils.sympy.torch_printer.SympyToTorch method) (physicsnemo.utils.graphcast.loss.CellAreaWeightedLossFunction method) (physicsnemo.utils.graphcast.loss.CustomCellAreaWeightedLossAutogradFunction static method) (physicsnemo.utils.graphcast.loss.CustomCellAreaWeightedLossFunction method) (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) forward_features() (physicsnemo.models.afno.afno.AFNO method) (physicsnemo.models.afno.modafno.ModAFNO method) fourier_encode() (in module physicsnemo.models.domino.model) fourier_encode_vectorized() (in module physicsnemo.models.domino.model) FourierConf (class in physicsnemo.sym.hydra.arch) FourierNetArch (class in physicsnemo.sym.models.fourier_net) freq (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) frequencies (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) frequencies_params (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) from_dtensor() (physicsnemo.distributed.shard_tensor.ShardTensor class method) from_local() (physicsnemo.distributed.shard_tensor.ShardTensor static method) from_numpy() (physicsnemo.sym.domain.constraint.continuous.DeepONetConstraint class method) (physicsnemo.sym.domain.constraint.continuous.PointwiseConstraint class method) from_stl() (physicsnemo.sym.geometry.tessellation.Tessellation class method) from_sympy() (physicsnemo.sym.node.Node class method) FSDP full_tensor() (physicsnemo.distributed.shard_tensor.ShardTensor method) FullyConnected (class in physicsnemo.models.mlp.fully_connected) FullyConnectedArch (class in physicsnemo.sym.models.fully_connected) FullyConnectedConf (class in physicsnemo.sym.hydra.arch) fuse() (physicsnemo.utils.patching.BasePatching2D method) (physicsnemo.utils.patching.GridPatching2D method) FusedFourierNetConf (class in physicsnemo.sym.hydra.arch) FusedGridEncodingNetConf (class in physicsnemo.sym.hydra.arch) FusedMLPConf (class in physicsnemo.sym.hydra.arch) G gabor_alpha (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) gabor_beta (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) gamma (physicsnemo.sym.hydra.optimizer.AdaBoundConf attribute) (physicsnemo.sym.hydra.scheduler.ExponentialLRConf attribute) gather_from_parallel_region() (in module physicsnemo.distributed.mappings) gather_v() (in module physicsnemo.distributed.autograd) gather_v_wrapper() (in module physicsnemo.distributed.utils) GatherVAutograd (class in physicsnemo.distributed.autograd) generate_batch() (physicsnemo.datapipes.benchmarks.darcy.Darcy2D method) (physicsnemo.datapipes.benchmarks.kelvin_helmholtz.KelvinHelmholtz2D method) generate_data() (physicsnemo.datapipes.climate.synthetic.SyntheticWeatherDataset method) geo_encoding_local() (physicsnemo.models.domino.model.DoMINO method) GeoConvOut (class in physicsnemo.models.domino.model) geom_points_sample (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) Geometry (class in physicsnemo.sym.geometry.geometry) GeometryDatapipe (class in physicsnemo.sym.geometry.geometry_dataloader) GeometryRep (class in physicsnemo.models.domino.model) GeometrySource (class in physicsnemo.sym.geometry.geometry_dataloader) GeoProcessor (class in physicsnemo.models.domino.model) geospatial_rotation() (in module physicsnemo.utils.graphcast.graph_utils) get() (physicsnemo.utils.filesystem.Package method) (physicsnemo.utils.graphcast.data_utils.StaticData method) get_activation() (in module physicsnemo.models.domino.model) get_available_backend() (physicsnemo.distributed.manager.DistributedManager static method) get_channel_dict() (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) get_checkpoint_dir() (in module physicsnemo.launch.utils.checkpoint) get_constants() (physicsnemo.datapipes.healpix.timeseries_dataset.TimeSeriesDataset method) get_dataloader() (physicsnemo.sym.domain.constraint.constraint.Constraint static method) get_dtype_and_ctype() (in module physicsnemo.utils.diffusion.utils) get_face_centroids() (in module physicsnemo.utils.graphcast.graph_utils) get_fields() (in module physicsnemo.utils.domino.utils) get_fields_from_cell() (in module physicsnemo.utils.domino.utils) get_filenames() (in module physicsnemo.utils.domino.utils) get_geop() (physicsnemo.utils.graphcast.data_utils.StaticData method) get_gradient_module() (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientCalculator method) get_lat_lon() (physicsnemo.utils.graphcast.data_utils.StaticData method) get_lat_lon_partition_separators() (in module physicsnemo.models.graphcast.graph_cast_net) get_lsm() (physicsnemo.utils.graphcast.data_utils.StaticData method) get_memory_format() (in module physicsnemo.distributed.utils) get_mesh_group() (physicsnemo.distributed.manager.DistributedManager method) get_module_dir_by_obj_name() (in module physicsnemo.utils.diffusion.utils) get_module_from_obj_name() (in module physicsnemo.utils.diffusion.utils) get_node_to_elem() (in module physicsnemo.utils.domino.utils) get_obj_by_name() (in module physicsnemo.utils.diffusion.utils) get_obj_from_module() (in module physicsnemo.utils.diffusion.utils) get_overlap_count() (physicsnemo.utils.patching.GridPatching2D static method) get_patch_indices() (physicsnemo.utils.patching.RandomPatching2D method) get_surface_data() (in module physicsnemo.utils.domino.utils) get_time_diff_std() (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) get_time_from_range() (in module physicsnemo.utils.corrdiff.utils) get_top_level_function_name() (in module physicsnemo.utils.diffusion.utils) get_vertices() (in module physicsnemo.utils.domino.utils) get_volume_data() (in module physicsnemo.utils.domino.utils) global_index() (physicsnemo.utils.patching.BasePatching2D method) global_mean() (in module physicsnemo.metrics.climate.reduction) global_mesh (physicsnemo.distributed.manager.DistributedManager property) global_var() (in module physicsnemo.metrics.climate.reduction) gpu_output (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) gpu_preprocessing (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) grad_agg_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) grad_averaging (physicsnemo.sym.hydra.optimizer.NovoGradConf attribute) grad_clip_max_norm (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) grad_transformer (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) GradientCalculator (class in physicsnemo.sym.eq.spatial_grads.spatial_grads) GradientsAutoDiff (class in physicsnemo.sym.eq.spatial_grads.spatial_grads) GradientsFiniteDifference (class in physicsnemo.sym.eq.spatial_grads.spatial_grads) GradientsLeastSquares (class in physicsnemo.sym.eq.spatial_grads.spatial_grads) GradientsMeshlessFiniteDifference (class in physicsnemo.sym.eq.spatial_grads.spatial_grads) GradientsSpectral (class in physicsnemo.sym.eq.spatial_grads.spatial_grads) GradNorm (class in physicsnemo.sym.loss.aggregator) GradNormal (class in physicsnemo.sym.eq.pdes.basic) (class in physicsnemo.sym.eq.pdes.navier_stokes) Graph (class in physicsnemo.sym.graph) (class in physicsnemo.utils.graphcast.graph) GraphCastLossFunction (class in physicsnemo.utils.graphcast.loss) GraphCastNet (class in physicsnemo.models.graphcast.graph_cast_net) grid_resolution (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) grid_to_points() (physicsnemo.models.fno.fno.FNO1DEncoder method) (physicsnemo.models.fno.fno.FNO2DEncoder method) (physicsnemo.models.fno.fno.FNO3DEncoder method) (physicsnemo.models.fno.fno.FNO4DEncoder method) grid_to_vtk() (in module physicsnemo.sym.utils.io.vtk) GridPatching2D (class in physicsnemo.utils.patching) GridValidator (class in physicsnemo.sym.domain.validator.discrete) GridValidatorPlotter (class in physicsnemo.sym.utils.io.plotter) group() (physicsnemo.distributed.manager.DistributedManager method) group_name() (physicsnemo.distributed.manager.DistributedManager method) group_names (physicsnemo.distributed.manager.DistributedManager property) group_rank() (physicsnemo.distributed.manager.DistributedManager method) group_size() (physicsnemo.distributed.manager.DistributedManager method) H handlers (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) HEALPixRecUNet (class in physicsnemo.models.dlwp_healpix.HEALPixRecUNet) HelmholtzEquation (class in physicsnemo.sym.eq.pdes.wave_equation) hessian_power (physicsnemo.sym.hydra.optimizer.AdahessianConf attribute) HighwayFourierConf (class in physicsnemo.sym.hydra.arch) HighwayFourierNetArch (class in physicsnemo.sym.models.highway_fourier_net) Histogram (class in physicsnemo.metrics.general.histogram) histogram() (in module physicsnemo.metrics.general.histogram) history_size (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) HomoscedasticUncertainty (class in physicsnemo.sym.loss.aggregator) hydra (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) I iDDPMPrecond (class in physicsnemo.models.diffusion.preconditioning) iDDPMPrecondMetaData (class in physicsnemo.models.diffusion.preconditioning) image_batching() (in module physicsnemo.utils.patching) image_fuse() (in module physicsnemo.utils.patching) img_shape (physicsnemo.sym.hydra.arch.AFNOConf attribute) (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) IncompressibleNavierStokes (class in physicsnemo.sym.eq.pdes.navier_stokes) indexed_all_to_all_v() (in module physicsnemo.distributed.autograd) indexed_all_to_all_v_wrapper() (in module physicsnemo.distributed.utils) indexed_all_to_all_v_wrapper_bwd() (in module physicsnemo.distributed.utils) IndexedAllToAllVAutograd (class in physicsnemo.distributed.autograd) indexing (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) Inferencer (class in physicsnemo.sym.domain.inferencer.inferencer) InferencerPlotter (class in physicsnemo.sym.utils.io.plotter) infinite_line_adf() (physicsnemo.sym.geometry.adf.ADF static method) InfiniteDataLoader (class in physicsnemo.sym.domain.constraint.constraint) InfiniteSampler (class in physicsnemo.utils.diffusion.utils) info() (physicsnemo.launch.logging.console.PythonLogger method) init_lr (physicsnemo.sym.hydra.optimizer.ApolloConf attribute) initial_accumulator (physicsnemo.sym.hydra.optimizer.YogiConf attribute) initial_accumulator_value (physicsnemo.sym.hydra.optimizer.AdagradConf attribute) initialization_network_dir (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) initialize() (physicsnemo.distributed.manager.DistributedManager static method) (physicsnemo.launch.logging.launch.LaunchLogger static method) initialize_batch() (physicsnemo.datapipes.benchmarks.darcy.Darcy2D method) (physicsnemo.datapipes.benchmarks.kelvin_helmholtz.KelvinHelmholtz2D method) initialize_env() (physicsnemo.distributed.manager.DistributedManager static method) initialize_mesh() (physicsnemo.distributed.manager.DistributedManager method) initialize_mlflow() (in module physicsnemo.launch.logging.mlflow) initialize_open_mpi() (physicsnemo.distributed.manager.DistributedManager static method) initialize_slurm() (physicsnemo.distributed.manager.DistributedManager static method) initialize_wandb() (in module physicsnemo.launch.logging.wandb) input_keys (physicsnemo.sym.hydra.arch.ModelConf attribute) input_scale (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) inputs (physicsnemo.sym.node.Node property) insolation() (in module physicsnemo.utils.insolation) integral (physicsnemo.sym.hydra.optimizer.PIDConf attribute) IntegralBoundaryConstraint (class in physicsnemo.sym.domain.constraint.continuous) IntegralConstraint (class in physicsnemo.sym.domain.constraint.continuous) IntegralLossNorm (class in physicsnemo.sym.loss.loss) integration_steps (physicsnemo.models.dlwp_healpix.HEALPixRecUNet.HEALPixRecUNet property) Inter-Node interpolation (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) Intra-Node is_initialized() (physicsnemo.distributed.manager.DistributedManager class method) is_top_level_function() (in module physicsnemo.utils.diffusion.utils) is_wandb_initialized() (in module physicsnemo.launch.logging.wandb) IsoTriangularPrism (class in physicsnemo.sym.geometry.primitives_3d) J jit (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) jit_arch_mode (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) jit_autograd_nodes (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) jit_use_nvfuser (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) K k (physicsnemo.sym.hydra.optimizer.RangerConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) kappa (physicsnemo.sym.hydra.optimizer.AccSGDConf attribute) kcrps() (in module physicsnemo.metrics.general.crps) KelvinHelmholtz2D (class in physicsnemo.datapipes.benchmarks.kelvin_helmholtz) Key (class in physicsnemo.sym.key) knn_interpolate() (physicsnemo.models.mesh_reduced.mesh_reduced.Mesh_Reduced method) L LambConf (class in physicsnemo.sym.hydra.optimizer) lambd (physicsnemo.sym.hydra.optimizer.ASGDConf attribute) large_kernel_size (physicsnemo.sym.hydra.arch.SRResConf attribute) last_epoch (physicsnemo.sym.hydra.scheduler.CosineAnnealingLRConf attribute) (physicsnemo.sym.hydra.scheduler.CosineAnnealingWarmRestartsConf attribute) latlon2xyz() (in module physicsnemo.utils.graphcast.graph_utils) LaunchLogger (class in physicsnemo.launch.logging.launch) layer_size (physicsnemo.sym.hydra.arch.ConvFullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.FullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FusedFourierNetConf attribute) (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.FusedMLPConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) (physicsnemo.sym.hydra.arch.SirenConf attribute) level (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) Line (class in physicsnemo.sym.geometry.primitives_2d) line() (in module physicsnemo.sym.utils.sympy.functions) Line1D (class in physicsnemo.sym.geometry.primitives_1d) line_search_fn (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) line_segment_adf() (physicsnemo.sym.geometry.adf.ADF static method) LinearElasticity (class in physicsnemo.sym.eq.pdes.linear_elasticity) LinearElasticityPlaneStress (class in physicsnemo.sym.eq.pdes.linear_elasticity) lips (physicsnemo.sym.hydra.optimizer.A2GradExpConf attribute) (physicsnemo.sym.hydra.optimizer.A2GradIncConf attribute) (physicsnemo.sym.hydra.optimizer.A2GradUniConf attribute) list_dir_recursively_with_ignore() (in module physicsnemo.utils.diffusion.utils) load_checkpoint() (in module physicsnemo.launch.utils.checkpoint) load_json() (in module physicsnemo.datapipes.gnn.utils) load_statistics() (physicsnemo.datapipes.cae.mesh_datapipe.MeshDatapipe method) (physicsnemo.datapipes.climate.era5_hdf5.ERA5HDF5Datapipe method) Local Rank local_rank (physicsnemo.distributed.manager.DistributedManager property) LocalPointConv (class in physicsnemo.models.domino.model) log() (physicsnemo.launch.logging.console.PythonLogger method) log2_hashmap_size (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) log_epoch() (physicsnemo.launch.logging.launch.LaunchLogger method) log_figure() (physicsnemo.launch.logging.launch.LaunchLogger method) log_minibatch() (physicsnemo.launch.logging.launch.LaunchLogger method) Loss (class in physicsnemo.sym.loss.loss) loss (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) LossConf (class in physicsnemo.sym.hydra.loss) LossL2 (class in physicsnemo.sym.loss.loss) lr (physicsnemo.sym.hydra.optimizer.A2GradExpConf attribute) (physicsnemo.sym.hydra.optimizer.A2GradIncConf attribute) (physicsnemo.sym.hydra.optimizer.A2GradUniConf attribute) (physicsnemo.sym.hydra.optimizer.AccSGDConf attribute) (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) (physicsnemo.sym.hydra.optimizer.AdaBoundConf attribute) (physicsnemo.sym.hydra.optimizer.AdadeltaConf attribute) (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) (physicsnemo.sym.hydra.optimizer.AdagradConf attribute) (physicsnemo.sym.hydra.optimizer.AdahessianConf attribute) (physicsnemo.sym.hydra.optimizer.AdamaxConf attribute) (physicsnemo.sym.hydra.optimizer.AdamConf attribute) (physicsnemo.sym.hydra.optimizer.AdaModConf attribute) (physicsnemo.sym.hydra.optimizer.AdamPConf attribute) (physicsnemo.sym.hydra.optimizer.AdamWConf attribute) (physicsnemo.sym.hydra.optimizer.AggMoConf attribute) (physicsnemo.sym.hydra.optimizer.ApolloConf attribute) (physicsnemo.sym.hydra.optimizer.ASGDConf attribute) (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) (physicsnemo.sym.hydra.optimizer.DiffGradConf attribute) (physicsnemo.sym.hydra.optimizer.LambConf attribute) (physicsnemo.sym.hydra.optimizer.MADGRADConf attribute) (physicsnemo.sym.hydra.optimizer.NAdamConf attribute) (physicsnemo.sym.hydra.optimizer.NovoGradConf attribute) (physicsnemo.sym.hydra.optimizer.PIDConf attribute) (physicsnemo.sym.hydra.optimizer.QHAdamConf attribute) (physicsnemo.sym.hydra.optimizer.QHMConf attribute) (physicsnemo.sym.hydra.optimizer.RAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RangerConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) (physicsnemo.sym.hydra.optimizer.RMSpropConf attribute) (physicsnemo.sym.hydra.optimizer.RpropConf attribute) (physicsnemo.sym.hydra.optimizer.SGDConf attribute) (physicsnemo.sym.hydra.optimizer.SGDPConf attribute) (physicsnemo.sym.hydra.optimizer.SGDWConf attribute) (physicsnemo.sym.hydra.optimizer.ShampooConf attribute) (physicsnemo.sym.hydra.optimizer.SparseAdamConf attribute) (physicsnemo.sym.hydra.optimizer.SWATSConf attribute) (physicsnemo.sym.hydra.optimizer.YogiConf attribute) lr_decay (physicsnemo.sym.hydra.optimizer.AdagradConf attribute) LRAnnealing (class in physicsnemo.sym.loss.aggregator) M MADGRADConf (class in physicsnemo.sym.hydra.optimizer) main() (in module physicsnemo.sym.hydra.utils) mark_module_as_shared() (in module physicsnemo.distributed.utils) max_edge_length() (in module physicsnemo.utils.graphcast.graph_utils) max_eval (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) max_iter (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) max_steps (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) MaxwellFreqReal (class in physicsnemo.sym.eq.pdes.electromagnetic) Mean (class in physicsnemo.metrics.general.ensemble_metrics) mean (physicsnemo.metrics.general.ensemble_metrics.Variance property) mean_std_sampling() (in module physicsnemo.utils.domino.utils) mesh() (physicsnemo.distributed.manager.DistributedManager method) mesh_dims (physicsnemo.distributed.manager.DistributedManager property) mesh_names() (physicsnemo.distributed.manager.DistributedManager method) Mesh_Reduced (class in physicsnemo.models.mesh_reduced.mesh_reduced) mesh_sizes() (physicsnemo.distributed.manager.DistributedManager method) MeshDaliExternalSource (class in physicsnemo.datapipes.cae.mesh_datapipe) MeshDatapipe (class in physicsnemo.datapipes.cae.mesh_datapipe) MeshGraphNet (class in physicsnemo.models.meshgraphnet.meshgraphnet) MeshGraphNetProcessor (class in physicsnemo.models.meshgraphnet.meshgraphnet) meshgrid() (physicsnemo.models.fno.fno.FNO1DEncoder method) (physicsnemo.models.fno.fno.FNO2DEncoder method) (physicsnemo.models.fno.fno.FNO3DEncoder method) (physicsnemo.models.fno.fno.FNO4DEncoder method) MetaData (class in physicsnemo.datapipes.benchmarks.darcy) (class in physicsnemo.datapipes.benchmarks.kelvin_helmholtz) (class in physicsnemo.datapipes.cae.mesh_datapipe) (class in physicsnemo.datapipes.climate.climate) (class in physicsnemo.datapipes.climate.era5_hdf5) (class in physicsnemo.datapipes.gnn.ahmed_body_dataset) (class in physicsnemo.datapipes.gnn.drivaernet_dataset) (class in physicsnemo.datapipes.healpix.timeseries_dataset) (class in physicsnemo.models.afno.afno) (class in physicsnemo.models.afno.modafno) (class in physicsnemo.models.dlwp.dlwp) (class in physicsnemo.models.dlwp_healpix.HEALPixRecUNet) (class in physicsnemo.models.fengwu.fengwu) (class in physicsnemo.models.fno.fno) (class in physicsnemo.models.graphcast.graph_cast_net) (class in physicsnemo.models.meshgraphnet.bsms_mgn) (class in physicsnemo.models.meshgraphnet.meshgraphnet) (class in physicsnemo.models.mlp.fully_connected) (class in physicsnemo.models.pangu.pangu) (class in physicsnemo.models.pix2pix.pix2pix) (class in physicsnemo.models.rnn.rnn_one2many) (class in physicsnemo.models.rnn.rnn_seq2seq) (class in physicsnemo.models.srrn.super_res_net) (class in physicsnemo.models.swinvrnn.swinvrnn) metric (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) min_delta (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) ModAFNO (class in physicsnemo.models.afno.modafno) ModAFNO2DLayer (class in physicsnemo.models.afno.modafno) ModAFNOMlp (class in physicsnemo.models.afno.modafno) mode (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) Model Parallel ModelConf (class in physicsnemo.sym.hydra.arch) models (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) ModifiedFourierConf (class in physicsnemo.sym.hydra.arch) ModifiedFourierNetArch (class in physicsnemo.sym.models.modified_fourier_net) module physicsnemo.datapipes physicsnemo.datapipes.benchmarks.darcy physicsnemo.datapipes.benchmarks.kelvin_helmholtz physicsnemo.datapipes.cae.domino_datapipe physicsnemo.datapipes.cae.mesh_datapipe physicsnemo.datapipes.climate.climate physicsnemo.datapipes.climate.era5_hdf5 physicsnemo.datapipes.climate.synthetic physicsnemo.datapipes.gnn.ahmed_body_dataset physicsnemo.datapipes.gnn.drivaernet_dataset physicsnemo.datapipes.gnn.stokes_dataset physicsnemo.datapipes.gnn.utils physicsnemo.datapipes.gnn.vortex_shedding_dataset physicsnemo.datapipes.healpix.timeseries_dataset physicsnemo.deploy physicsnemo.deploy.onnx.utils physicsnemo.distributed physicsnemo.distributed.autograd physicsnemo.distributed.fft physicsnemo.distributed.manager physicsnemo.distributed.mappings physicsnemo.distributed.utils physicsnemo.launch.logging physicsnemo.launch.logging.console physicsnemo.launch.logging.launch physicsnemo.launch.logging.mlflow physicsnemo.launch.logging.utils physicsnemo.launch.logging.wandb physicsnemo.launch.utils physicsnemo.launch.utils.checkpoint physicsnemo.metrics physicsnemo.metrics.climate.acc physicsnemo.metrics.climate.efi physicsnemo.metrics.climate.reduction physicsnemo.metrics.general.calibration physicsnemo.metrics.general.crps physicsnemo.metrics.general.ensemble_metrics physicsnemo.metrics.general.entropy physicsnemo.metrics.general.histogram physicsnemo.metrics.general.mse physicsnemo.metrics.general.reduction physicsnemo.metrics.general.wasserstein physicsnemo.models physicsnemo.models.afno.afno physicsnemo.models.afno.modafno physicsnemo.models.diffusion.preconditioning physicsnemo.models.dlwp.dlwp physicsnemo.models.dlwp_healpix.HEALPixRecUNet physicsnemo.models.domino.model physicsnemo.models.fengwu.fengwu physicsnemo.models.fno.fno physicsnemo.models.graphcast.graph_cast_net physicsnemo.models.mesh_reduced.mesh_reduced physicsnemo.models.meshgraphnet.bsms_mgn physicsnemo.models.meshgraphnet.meshgraphnet physicsnemo.models.mlp.fully_connected physicsnemo.models.pangu.pangu physicsnemo.models.pix2pix.pix2pix physicsnemo.models.rnn.rnn_one2many physicsnemo.models.rnn.rnn_seq2seq physicsnemo.models.srrn.super_res_net physicsnemo.models.swinvrnn.swinvrnn physicsnemo.sym.constants physicsnemo.sym.domain.constraint.constraint physicsnemo.sym.domain.constraint.continuous physicsnemo.sym.domain.constraint.discrete physicsnemo.sym.domain.inferencer.inferencer physicsnemo.sym.domain.inferencer.pointwise physicsnemo.sym.domain.inferencer.voxel physicsnemo.sym.domain.inferencer.vtkpointwise physicsnemo.sym.domain.monitor.monitor physicsnemo.sym.domain.monitor.pointwise physicsnemo.sym.domain.validator.continuous physicsnemo.sym.domain.validator.discrete physicsnemo.sym.domain.validator.validator physicsnemo.sym.eq.pdes.advection_diffusion physicsnemo.sym.eq.pdes.basic physicsnemo.sym.eq.pdes.diffusion physicsnemo.sym.eq.pdes.electromagnetic physicsnemo.sym.eq.pdes.linear_elasticity physicsnemo.sym.eq.pdes.navier_stokes physicsnemo.sym.eq.pdes.signed_distance_function physicsnemo.sym.eq.pdes.turbulence_zero_eq physicsnemo.sym.eq.pdes.wave_equation physicsnemo.sym.eq.phy_informer physicsnemo.sym.eq.spatial_grads.spatial_grads physicsnemo.sym.geometry.adf physicsnemo.sym.geometry.curve physicsnemo.sym.geometry.discrete_geometry physicsnemo.sym.geometry.geometry physicsnemo.sym.geometry.geometry_dataloader physicsnemo.sym.geometry.parameterization physicsnemo.sym.geometry.primitives_1d physicsnemo.sym.geometry.primitives_2d physicsnemo.sym.geometry.primitives_3d physicsnemo.sym.geometry.tessellation physicsnemo.sym.graph physicsnemo.sym.hydra.arch physicsnemo.sym.hydra.config physicsnemo.sym.hydra.hydra physicsnemo.sym.hydra.loss physicsnemo.sym.hydra.optimizer physicsnemo.sym.hydra.profiler physicsnemo.sym.hydra.scheduler physicsnemo.sym.hydra.training physicsnemo.sym.hydra.utils physicsnemo.sym.key physicsnemo.sym.loss.aggregator physicsnemo.sym.loss.loss physicsnemo.sym.models.afno physicsnemo.sym.models.deeponet physicsnemo.sym.models.dgm physicsnemo.sym.models.fno physicsnemo.sym.models.fourier_net physicsnemo.sym.models.fully_connected physicsnemo.sym.models.hash_encoding_net physicsnemo.sym.models.highway_fourier_net physicsnemo.sym.models.modified_fourier_net physicsnemo.sym.models.moving_time_window physicsnemo.sym.models.multiplicative_filter_net physicsnemo.sym.models.multiscale_fourier_net physicsnemo.sym.models.pix2pix physicsnemo.sym.models.radial_basis physicsnemo.sym.models.siren physicsnemo.sym.models.super_res_net physicsnemo.sym.node physicsnemo.sym.trainer physicsnemo.sym.utils.io.csv_rw physicsnemo.sym.utils.io.plotter physicsnemo.sym.utils.io.vtk physicsnemo.sym.utils.sympy.functions physicsnemo.sym.utils.sympy.numpy_printer physicsnemo.sym.utils.sympy.torch_printer physicsnemo.sym.utils.vpinn.integral physicsnemo.sym.utils.vpinn.test_functions physicsnemo.utils physicsnemo.utils.capture physicsnemo.utils.corrdiff.utils physicsnemo.utils.diffusion.deterministic_sampler physicsnemo.utils.diffusion.stochastic_sampler physicsnemo.utils.diffusion.utils physicsnemo.utils.domino.utils physicsnemo.utils.filesystem physicsnemo.utils.graphcast.data_utils physicsnemo.utils.graphcast.graph physicsnemo.utils.graphcast.graph_utils physicsnemo.utils.graphcast.loss physicsnemo.utils.insolation physicsnemo.utils.neighbors physicsnemo.utils.neighbors.radius_search physicsnemo.utils.patching physicsnemo.utils.profiling physicsnemo.utils.sdf physicsnemo.utils.zenith_angle momentum (physicsnemo.sym.hydra.optimizer.MADGRADConf attribute) (physicsnemo.sym.hydra.optimizer.PIDConf attribute) (physicsnemo.sym.hydra.optimizer.QHMConf attribute) (physicsnemo.sym.hydra.optimizer.RMSpropConf attribute) (physicsnemo.sym.hydra.optimizer.SGDConf attribute) (physicsnemo.sym.hydra.optimizer.SGDPConf attribute) (physicsnemo.sym.hydra.optimizer.SGDWConf attribute) (physicsnemo.sym.hydra.optimizer.ShampooConf attribute) momentum_decay (physicsnemo.sym.hydra.optimizer.NAdamConf attribute) Monitor (class in physicsnemo.sym.domain.monitor.monitor) monitor_grad_clip (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) MovingTimeWindowArch (class in physicsnemo.sym.models.moving_time_window) mse() (in module physicsnemo.metrics.general.mse) MultiplicativeFilterConf (class in physicsnemo.sym.hydra.arch) MultiplicativeFilterNetArch (class in physicsnemo.sym.models.multiplicative_filter_net) MultiresolutionHashNetArch (class in physicsnemo.sym.models.hash_encoding_net) MultiresolutionHashNetConf (class in physicsnemo.sym.hydra.arch) MultiscaleFourierConf (class in physicsnemo.sym.hydra.arch) MultiscaleFourierNetArch (class in physicsnemo.sym.models.multiscale_fourier_net) N n_blocks (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) n_downsampling (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) n_features_per_level (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) n_frequencies (physicsnemo.sym.hydra.arch.FusedFourierNetConf attribute) n_levels (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) n_resid_blocks (physicsnemo.sym.hydra.arch.SRResConf attribute) N_sma_threshhold (physicsnemo.sym.hydra.optimizer.RangerConf attribute) n_sma_threshhold (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) NAdamConf (class in physicsnemo.sym.hydra.optimizer) name (physicsnemo.sym.hydra.profiler.NvtxProfiler attribute) (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) named_params_and_buffers() (in module physicsnemo.utils.diffusion.utils) NavierStokes (class in physicsnemo.sym.eq.pdes.navier_stokes) nd_interpolator() (in module physicsnemo.utils.domino.utils) nesterov (physicsnemo.sym.hydra.optimizer.SGDConf attribute) (physicsnemo.sym.hydra.optimizer.SGDPConf attribute) (physicsnemo.sym.hydra.optimizer.SGDWConf attribute) (physicsnemo.sym.hydra.optimizer.SWATSConf attribute) NetCDFWriter (class in physicsnemo.utils.corrdiff.utils) network_dir (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) NNBasisFunctions (class in physicsnemo.models.domino.model) Node (class in physicsnemo.sym.node) normal_cdf() (in module physicsnemo.metrics.general.histogram) normal_pdf() (in module physicsnemo.metrics.general.histogram) NormalDotVec (class in physicsnemo.sym.eq.pdes.basic) normalization (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) (physicsnemo.sym.hydra.arch.SirenConf attribute) normalize() (in module physicsnemo.utils.domino.utils) normalize_coordinates (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) normalize_edge() (physicsnemo.datapipes.gnn.ahmed_body_dataset.AhmedBodyDataset method) (physicsnemo.datapipes.gnn.stokes_dataset.StokesDataset method) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) normalize_node() (physicsnemo.datapipes.gnn.ahmed_body_dataset.AhmedBodyDataset method) (physicsnemo.datapipes.gnn.stokes_dataset.StokesDataset method) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) normalized_entropy() (in module physicsnemo.metrics.climate.efi) NovoGradConf (class in physicsnemo.sym.hydra.optimizer) np_lambdify() (in module physicsnemo.sym.utils.sympy.numpy_printer) nr_features_per_level (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) nr_fno_layers (physicsnemo.sym.hydra.arch.FNOConf attribute) nr_layers (physicsnemo.sym.hydra.arch.ConvFullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.FullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FusedFourierNetConf attribute) (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.FusedMLPConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) (physicsnemo.sym.hydra.arch.SirenConf attribute) nr_levels (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) NTK (class in physicsnemo.sym.loss.aggregator) ntk (physicsnemo.sym.hydra.training.DefaultTraining attribute) NTKConf (class in physicsnemo.sym.hydra.loss) nu (physicsnemo.sym.hydra.optimizer.QHMConf attribute) num_blocks (physicsnemo.sym.hydra.arch.AFNOConf attribute) (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) num_surface_neighbors (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) nus (physicsnemo.sym.hydra.optimizer.QHAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) NvtxProfiler (class in physicsnemo.sym.hydra.profiler) O offsets() (physicsnemo.distributed.shard_tensor.ShardTensor method) omega (physicsnemo.sym.hydra.arch.SirenConf attribute) One2ManyRNN (class in physicsnemo.models.rnn.rnn_one2many) optimizer (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) OptimizerConf (class in physicsnemo.sym.hydra.optimizer) OrderedParameterization (class in physicsnemo.sym.geometry.parameterization) output_keys (physicsnemo.sym.hydra.arch.ModelConf attribute) outputs (physicsnemo.sym.node.Node property) P Package (class in physicsnemo.utils.filesystem) pad() (in module physicsnemo.utils.domino.utils) pad_helper() (in module physicsnemo.distributed.utils) pad_inp() (in module physicsnemo.utils.domino.utils) padding (physicsnemo.sym.hydra.arch.FNOConf attribute) padding_type (physicsnemo.sym.hydra.arch.FNOConf attribute) (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) Pangu (class in physicsnemo.models.pangu.pangu) parabola() (in module physicsnemo.sym.utils.sympy.functions) parabola2D() (in module physicsnemo.sym.utils.sympy.functions) Parameter (class in physicsnemo.sym.geometry.parameterization) Parameterization (class in physicsnemo.sym.geometry.parameterization) ParameterModel (class in physicsnemo.models.domino.model) params_and_buffers() (in module physicsnemo.utils.diffusion.utils) parse_dataset_files() (physicsnemo.datapipes.cae.mesh_datapipe.MeshDatapipe method) (physicsnemo.datapipes.climate.climate.ClimateDataSourceSpec method) (physicsnemo.datapipes.climate.era5_hdf5.ERA5HDF5Datapipe method) (physicsnemo.sym.geometry.geometry_dataloader.GeometryDatapipe method) parse_int_list() (in module physicsnemo.utils.diffusion.utils) parse_variable() (physicsnemo.utils.graphcast.loss.GraphCastLossFunction method) patch_indices (physicsnemo.utils.patching.RandomPatching2D attribute) patch_num (physicsnemo.utils.patching.GridPatching2D attribute) (physicsnemo.utils.patching.RandomPatching2D property) patch_size (physicsnemo.sym.hydra.arch.AFNOConf attribute) (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) PatchEmbed (class in physicsnemo.models.afno.afno) patches_enabled() (physicsnemo.distributed.shard_tensor.ShardTensor class method) patience (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) pde (physicsnemo.sym.hydra.config.ExperimentalPhysicsNeMoConfig attribute) PEC (class in physicsnemo.sym.eq.pdes.electromagnetic) per_level_scale (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) phase (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) PhysicsInformer (class in physicsnemo.sym.eq.phy_informer) physicsnemo.datapipes module physicsnemo.datapipes.benchmarks.darcy module physicsnemo.datapipes.benchmarks.kelvin_helmholtz module physicsnemo.datapipes.cae.domino_datapipe module physicsnemo.datapipes.cae.mesh_datapipe module physicsnemo.datapipes.climate.climate module physicsnemo.datapipes.climate.era5_hdf5 module physicsnemo.datapipes.climate.synthetic module physicsnemo.datapipes.gnn.ahmed_body_dataset module physicsnemo.datapipes.gnn.drivaernet_dataset module physicsnemo.datapipes.gnn.stokes_dataset module physicsnemo.datapipes.gnn.utils module physicsnemo.datapipes.gnn.vortex_shedding_dataset module physicsnemo.datapipes.healpix.timeseries_dataset module physicsnemo.deploy module physicsnemo.deploy.onnx.utils module physicsnemo.distributed module physicsnemo.distributed.autograd module physicsnemo.distributed.fft module physicsnemo.distributed.manager module physicsnemo.distributed.mappings module physicsnemo.distributed.utils module physicsnemo.launch.logging module physicsnemo.launch.logging.console module physicsnemo.launch.logging.launch module physicsnemo.launch.logging.mlflow module physicsnemo.launch.logging.utils module physicsnemo.launch.logging.wandb module physicsnemo.launch.utils module physicsnemo.launch.utils.checkpoint module physicsnemo.metrics module physicsnemo.metrics.climate.acc module physicsnemo.metrics.climate.efi module physicsnemo.metrics.climate.reduction module physicsnemo.metrics.general.calibration module physicsnemo.metrics.general.crps module physicsnemo.metrics.general.ensemble_metrics module physicsnemo.metrics.general.entropy module physicsnemo.metrics.general.histogram module physicsnemo.metrics.general.mse module physicsnemo.metrics.general.reduction module physicsnemo.metrics.general.wasserstein module physicsnemo.models module physicsnemo.models.afno.afno module physicsnemo.models.afno.modafno module physicsnemo.models.diffusion.preconditioning module physicsnemo.models.dlwp.dlwp module physicsnemo.models.dlwp_healpix.HEALPixRecUNet module physicsnemo.models.domino.model module physicsnemo.models.fengwu.fengwu module physicsnemo.models.fno.fno module physicsnemo.models.graphcast.graph_cast_net module physicsnemo.models.mesh_reduced.mesh_reduced module physicsnemo.models.meshgraphnet.bsms_mgn module physicsnemo.models.meshgraphnet.meshgraphnet module physicsnemo.models.mlp.fully_connected module physicsnemo.models.pangu.pangu module physicsnemo.models.pix2pix.pix2pix module physicsnemo.models.rnn.rnn_one2many module physicsnemo.models.rnn.rnn_seq2seq module physicsnemo.models.srrn.super_res_net module physicsnemo.models.swinvrnn.swinvrnn module physicsnemo.sym.constants module physicsnemo.sym.domain.constraint.constraint module physicsnemo.sym.domain.constraint.continuous module physicsnemo.sym.domain.constraint.discrete module physicsnemo.sym.domain.inferencer.inferencer module physicsnemo.sym.domain.inferencer.pointwise module physicsnemo.sym.domain.inferencer.voxel module physicsnemo.sym.domain.inferencer.vtkpointwise module physicsnemo.sym.domain.monitor.monitor module physicsnemo.sym.domain.monitor.pointwise module physicsnemo.sym.domain.validator.continuous module physicsnemo.sym.domain.validator.discrete module physicsnemo.sym.domain.validator.validator module physicsnemo.sym.eq.pdes.advection_diffusion module physicsnemo.sym.eq.pdes.basic module physicsnemo.sym.eq.pdes.diffusion module physicsnemo.sym.eq.pdes.electromagnetic module physicsnemo.sym.eq.pdes.linear_elasticity module physicsnemo.sym.eq.pdes.navier_stokes module physicsnemo.sym.eq.pdes.signed_distance_function module physicsnemo.sym.eq.pdes.turbulence_zero_eq module physicsnemo.sym.eq.pdes.wave_equation module physicsnemo.sym.eq.phy_informer module physicsnemo.sym.eq.spatial_grads.spatial_grads module physicsnemo.sym.geometry.adf module physicsnemo.sym.geometry.curve module physicsnemo.sym.geometry.discrete_geometry module physicsnemo.sym.geometry.geometry module physicsnemo.sym.geometry.geometry_dataloader module physicsnemo.sym.geometry.parameterization module physicsnemo.sym.geometry.primitives_1d module physicsnemo.sym.geometry.primitives_2d module physicsnemo.sym.geometry.primitives_3d module physicsnemo.sym.geometry.tessellation module physicsnemo.sym.graph module physicsnemo.sym.hydra.arch module physicsnemo.sym.hydra.config module physicsnemo.sym.hydra.hydra module physicsnemo.sym.hydra.loss module physicsnemo.sym.hydra.optimizer module physicsnemo.sym.hydra.profiler module physicsnemo.sym.hydra.scheduler module physicsnemo.sym.hydra.training module physicsnemo.sym.hydra.utils module physicsnemo.sym.key module physicsnemo.sym.loss.aggregator module physicsnemo.sym.loss.loss module physicsnemo.sym.models.afno module physicsnemo.sym.models.deeponet module physicsnemo.sym.models.dgm module physicsnemo.sym.models.fno module physicsnemo.sym.models.fourier_net module physicsnemo.sym.models.fully_connected module physicsnemo.sym.models.hash_encoding_net module physicsnemo.sym.models.highway_fourier_net module physicsnemo.sym.models.modified_fourier_net module physicsnemo.sym.models.moving_time_window module physicsnemo.sym.models.multiplicative_filter_net module physicsnemo.sym.models.multiscale_fourier_net module physicsnemo.sym.models.pix2pix module physicsnemo.sym.models.radial_basis module physicsnemo.sym.models.siren module physicsnemo.sym.models.super_res_net module physicsnemo.sym.node module physicsnemo.sym.trainer module physicsnemo.sym.utils.io.csv_rw module physicsnemo.sym.utils.io.plotter module physicsnemo.sym.utils.io.vtk module physicsnemo.sym.utils.sympy.functions module physicsnemo.sym.utils.sympy.numpy_printer module physicsnemo.sym.utils.sympy.torch_printer module physicsnemo.sym.utils.vpinn.integral module physicsnemo.sym.utils.vpinn.test_functions module physicsnemo.utils module physicsnemo.utils.capture module physicsnemo.utils.corrdiff.utils module physicsnemo.utils.diffusion.deterministic_sampler module physicsnemo.utils.diffusion.stochastic_sampler module physicsnemo.utils.diffusion.utils module physicsnemo.utils.domino.utils module physicsnemo.utils.filesystem module physicsnemo.utils.graphcast.data_utils module physicsnemo.utils.graphcast.graph module physicsnemo.utils.graphcast.graph_utils module physicsnemo.utils.graphcast.loss module physicsnemo.utils.insolation module physicsnemo.utils.neighbors module physicsnemo.utils.neighbors.radius_search module physicsnemo.utils.patching module physicsnemo.utils.profiling module physicsnemo.utils.sdf module physicsnemo.utils.zenith_angle module PhysicsNeMoConfig (class in physicsnemo.sym.hydra.config) PhysicsNeMoUndefinedGroupError PhysicsNeMoUninitializedDistributedManagerWarning PIDConf (class in physicsnemo.sym.hydra.optimizer) Pipeline Parallel Pix2Pix (class in physicsnemo.models.pix2pix.pix2pix) Pix2PixArch (class in physicsnemo.sym.models.pix2pix) Pix2PixConf (class in physicsnemo.sym.hydra.arch) PixelShuffle3d (class in physicsnemo.models.srrn.super_res_net) Plane (class in physicsnemo.sym.geometry.primitives_3d) Point1D (class in physicsnemo.sym.geometry.primitives_1d) points_to_grid() (physicsnemo.models.fno.fno.FNO1DEncoder method) (physicsnemo.models.fno.fno.FNO2DEncoder method) (physicsnemo.models.fno.fno.FNO3DEncoder method) (physicsnemo.models.fno.fno.FNO4DEncoder method) PointVTKInferencer (class in physicsnemo.sym.domain.inferencer.vtkpointwise) PointVTKValidator (class in physicsnemo.sym.domain.validator.continuous) PointwiseBoundaryConstraint (class in physicsnemo.sym.domain.constraint.continuous) PointwiseConstraint (class in physicsnemo.sym.domain.constraint.continuous) PointwiseInferencer (class in physicsnemo.sym.domain.inferencer.pointwise) PointwiseInteriorConstraint (class in physicsnemo.sym.domain.constraint.continuous) PointwiseLossNorm (class in physicsnemo.sym.loss.loss) PointwiseMonitor (class in physicsnemo.sym.domain.monitor.pointwise) PointwiseValidator (class in physicsnemo.sym.domain.validator.continuous) polar_angle() (in module physicsnemo.utils.graphcast.graph_utils) Polygon (class in physicsnemo.sym.geometry.primitives_2d) position_encoder() (physicsnemo.models.domino.model.DoMINO method) positional_embedding_indexing() (physicsnemo.models.diffusion.song_unet.SongUNetPosEmbd method) (physicsnemo.models.diffusion.song_unet.SongUNetPosLtEmbd method) positional_embedding_selector() (physicsnemo.models.diffusion.song_unet.SongUNetPosEmbd method) (physicsnemo.models.diffusion.song_unet.SongUNetPosLtEmbd method) positional_encoding (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) PositionEncoder (class in physicsnemo.models.domino.model) prepare_input() (physicsnemo.models.fengwu.fengwu.Fengwu method) (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) (physicsnemo.models.pangu.pangu.Pangu method) prepare_output() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) print_module_summary() (in module physicsnemo.utils.diffusion.utils) print_stats_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) profile (physicsnemo.sym.hydra.profiler.NvtxProfiler attribute) (physicsnemo.sym.hydra.profiler.ProfilerConf attribute) (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) profile_mode (physicsnemo.models.diffusion.dhariwal_unet.DhariwalUNet property) (physicsnemo.models.diffusion.preconditioning.EDMPrecondSuperResolution property) (physicsnemo.models.diffusion.song_unet.SongUNet property) (physicsnemo.models.diffusion.song_unet.SongUNetPosEmbd property) (physicsnemo.models.diffusion.song_unet.SongUNetPosLtEmbd property) (physicsnemo.models.diffusion.unet.UNet property) profiled_function() (in module physicsnemo.utils.diffusion.utils) profiler (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) ProfilerConf (class in physicsnemo.sym.hydra.profiler) project_fourier_features (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) PythonLogger (class in physicsnemo.launch.logging.console) Q QHAdamConf (class in physicsnemo.sym.hydra.optimizer) QHMConf (class in physicsnemo.sym.hydra.optimizer) Quad_Rect (class in physicsnemo.sym.utils.vpinn.integral) R r_equivalence() (physicsnemo.sym.geometry.adf.ADF static method) rad2deg() (in module physicsnemo.utils.graphcast.graph_utils) RAdamConf (class in physicsnemo.sym.hydra.optimizer) RadialBasisArch (class in physicsnemo.sym.models.radial_basis) radius_search() (in module physicsnemo.utils.neighbors) randint() (physicsnemo.utils.diffusion.utils.StackedRandomGenerator method) randn() (physicsnemo.utils.diffusion.utils.StackedRandomGenerator method) randn_like() (physicsnemo.utils.diffusion.utils.StackedRandomGenerator method) RandomPatching2D (class in physicsnemo.utils.patching) randt() (physicsnemo.utils.diffusion.utils.StackedRandomGenerator method) RangerConf (class in physicsnemo.sym.hydra.optimizer) RangerQHConf (class in physicsnemo.sym.hydra.optimizer) RangerVAConf (class in physicsnemo.sym.hydra.optimizer) Rank rank (physicsnemo.distributed.manager.DistributedManager property) rank_probability_score() (in module physicsnemo.metrics.general.calibration) RankZeroLoggingWrapper (class in physicsnemo.launch.logging.console) read_vtp_file() (in module physicsnemo.datapipes.gnn.utils) rec_constraint_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) rec_inference_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) rec_monitor_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) rec_results_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) rec_validation_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) Rectangle (class in physicsnemo.sym.geometry.primitives_2d) rectify (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) redistribute() (physicsnemo.distributed.shard_tensor.ShardTensor method) reduce_from_parallel_region() (in module physicsnemo.distributed.mappings) reduce_loss() (in module physicsnemo.distributed.utils) ref_key (physicsnemo.sym.hydra.loss.AggregatorLRAnnealingConf attribute) register_arch_configs() (in module physicsnemo.sym.hydra.arch) register_dispatch_handler() (physicsnemo.distributed.shard_tensor.ShardTensor class method) register_function_handler() (physicsnemo.distributed.shard_tensor.ShardTensor class method) register_hydra_configs() (in module physicsnemo.sym.hydra.hydra) register_loss_configs() (in module physicsnemo.sym.hydra.loss) register_named_function_handler() (physicsnemo.distributed.shard_tensor.ShardTensor class method) register_optimizer_configs() (in module physicsnemo.sym.hydra.optimizer) register_physicsnemo_configs() (in module physicsnemo.sym.hydra.config) register_profiler_configs() (in module physicsnemo.sym.hydra.profiler) register_scheduler_configs() (in module physicsnemo.sym.hydra.scheduler) register_training_configs() (in module physicsnemo.sym.hydra.training) regression_step() (in module physicsnemo.utils.corrdiff.utils) relative_entropy_from_counts() (in module physicsnemo.metrics.general.entropy) relative_step (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) Relobralo (class in physicsnemo.sym.loss.aggregator) repeat (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) repeat() (physicsnemo.sym.geometry.geometry.Geometry method) required_inputs (physicsnemo.sym.eq.phy_informer.PhysicsInformer property) resample_surfaces (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) resampling_points (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) reset() (physicsnemo.models.dlwp_healpix.HEALPixRecUNet.HEALPixRecUNet method) reset_patch_indices() (physicsnemo.utils.patching.RandomPatching2D method) ResidualConvBlock3d (class in physicsnemo.models.srrn.super_res_net) ResnetBlock (class in physicsnemo.models.pix2pix.pix2pix) ResNorm (class in physicsnemo.sym.loss.aggregator) rho (physicsnemo.sym.hydra.optimizer.AdadeltaConf attribute) rmse() (in module physicsnemo.metrics.general.mse) RMSpropConf (class in physicsnemo.sym.hydra.optimizer) root (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) rotate() (physicsnemo.sym.geometry.curve.Curve method) (physicsnemo.sym.geometry.geometry.Geometry method) round_sigma() (physicsnemo.models.diffusion.preconditioning.EDMPrecond static method) (physicsnemo.models.diffusion.preconditioning.EDMPrecondSuperResolution static method) (physicsnemo.models.diffusion.preconditioning.iDDPMPrecond method) (physicsnemo.models.diffusion.preconditioning.VEPrecond method) (physicsnemo.models.diffusion.preconditioning.VPPrecond method) RpropConf (class in physicsnemo.sym.hydra.optimizer) run_freq (physicsnemo.sym.hydra.loss.NTKConf attribute) run_function() (physicsnemo.models.meshgraphnet.meshgraphnet.MeshGraphNetProcessor method) run_mode (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) S sample() (physicsnemo.sym.geometry.parameterization.Bounds method) (physicsnemo.sym.geometry.parameterization.OrderedParameterization method) (physicsnemo.sym.geometry.parameterization.Parameterization method) sample_boundary() (physicsnemo.sym.geometry.geometry.Geometry method) sample_in_bbox (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) sample_interior() (physicsnemo.sym.geometry.geometry.Geometry method) (physicsnemo.sym.geometry.primitives_2d.Line method) sample_sphere() (physicsnemo.models.domino.model.DoMINO method) sample_sphere_shell() (physicsnemo.models.domino.model.DoMINO method) sampling (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) save_checkpoint() (in module physicsnemo.launch.utils.checkpoint) save_filetypes (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) save_json() (in module physicsnemo.datapipes.gnn.utils) save_name (physicsnemo.sym.hydra.loss.NTKConf attribute) save_network_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) scale() (physicsnemo.sym.geometry.curve.Curve method) (physicsnemo.sym.geometry.geometry.Geometry method) scale_parameter (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) scale_sdf() (in module physicsnemo.models.domino.model) ScaleShiftMlp (class in physicsnemo.models.afno.modafno) scaling (physicsnemo.sym.hydra.arch.ModelConf attribute) scaling_factor (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) (physicsnemo.sym.hydra.arch.SRResConf attribute) scaling_type (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) scatter_tensor() (in module physicsnemo.distributed.shard_tensor) scatter_to_parallel_region() (in module physicsnemo.distributed.mappings) scatter_v() (in module physicsnemo.distributed.autograd) scatter_v_wrapper() (in module physicsnemo.distributed.utils) ScatterVAutograd (class in physicsnemo.distributed.autograd) scheduler (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) SchedulerConf (class in physicsnemo.sym.hydra.scheduler) ScreenedPoissonDistance (class in physicsnemo.sym.eq.pdes.signed_distance_function) Seq2SeqRNN (class in physicsnemo.models.rnn.rnn_seq2seq) set_checkpoint_decoder() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_checkpoint_encoder() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_checkpoint_model() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_checkpoint_offload_ctx() (physicsnemo.models.meshgraphnet.meshgraphnet.MeshGraphNetProcessor method) set_checkpoint_processor() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_checkpoint_segments() (physicsnemo.models.meshgraphnet.meshgraphnet.MeshGraphNetProcessor method) set_patch_num() (physicsnemo.utils.patching.RandomPatching2D method) setup() (physicsnemo.distributed.manager.DistributedManager static method) setup_deriv_scaler() (physicsnemo.sym.graph.Graph method) SGDConf (class in physicsnemo.sym.hydra.optimizer) SGDPConf (class in physicsnemo.sym.hydra.optimizer) SGDWConf (class in physicsnemo.sym.hydra.optimizer) ShampooConf (class in physicsnemo.sym.hydra.optimizer) ShardTensor (class in physicsnemo.distributed.shard_tensor) shuffle_array() (in module physicsnemo.utils.domino.utils) shuffle_array_without_sampling() (in module physicsnemo.utils.domino.utils) sigma() (physicsnemo.models.diffusion.preconditioning.VPPrecond method) sigma_inv() (physicsnemo.models.diffusion.preconditioning.VPPrecond method) SimpleFormat (class in physicsnemo.sym.hydra.hydra) SirenArch (class in physicsnemo.sym.models.siren) SirenConf (class in physicsnemo.sym.hydra.arch) skip_connections (physicsnemo.sym.hydra.arch.ConvFullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.FullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) small_const (physicsnemo.sym.hydra.optimizer.AccSGDConf attribute) small_kernel_size (physicsnemo.sym.hydra.arch.SRResConf attribute) smooth (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) SoftAdapt (class in physicsnemo.sym.loss.aggregator) solution_weighted_shuffle_array() (in module physicsnemo.utils.domino.utils) SommerfeldBC (class in physicsnemo.sym.eq.pdes.electromagnetic) SongUNet (class in physicsnemo.models.diffusion.song_unet) SongUNetPosEmbd (class in physicsnemo.models.diffusion.song_unet) SongUNetPosLtEmbd (class in physicsnemo.models.diffusion.song_unet) SparseAdamConf (class in physicsnemo.sym.hydra.optimizer) Sphere (class in physicsnemo.sym.geometry.primitives_3d) SRResConf (class in physicsnemo.sym.hydra.arch) SRResNet (class in physicsnemo.models.srrn.super_res_net) SRResNetArch (class in physicsnemo.sym.models.super_res_net) StackedRandomGenerator (class in physicsnemo.utils.diffusion.utils) standardize() (in module physicsnemo.utils.domino.utils) start_step (physicsnemo.sym.hydra.profiler.NvtxProfiler attribute) (physicsnemo.sym.hydra.profiler.ProfilerConf attribute) (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) StaticCaptureEvaluateNoGrad (class in physicsnemo.utils.capture) StaticCaptureTraining (class in physicsnemo.utils.capture) StaticData (class in physicsnemo.utils.graphcast.data_utils) step_sizes (physicsnemo.sym.hydra.optimizer.RpropConf attribute) stochastic_sampler() (in module physicsnemo.utils.diffusion.stochastic_sampler) StokesDataset (class in physicsnemo.datapipes.gnn.stokes_dataset) stop_criterion (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) StopCriterionConf (class in physicsnemo.sym.hydra.training) strict (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) Strong Scaling SubPixel_ConvolutionalBlock3d (class in physicsnemo.models.srrn.super_res_net) success() (physicsnemo.launch.logging.console.PythonLogger method) Sum (class in physicsnemo.sym.loss.aggregator) summary_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) summary_histograms (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) SupervisedGridConstraint (class in physicsnemo.sym.domain.constraint.discrete) supports_func_arch (physicsnemo.sym.models.deeponet.DeepONetArch property) suppress_tracer_warnings() (in module physicsnemo.utils.diffusion.utils) surface_factors (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) surface_points_sample (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) surface_sampling_algorithm (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) surface_variables (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) SWATSConf (class in physicsnemo.sym.hydra.optimizer) SwinRNN (class in physicsnemo.models.swinvrnn.swinvrnn) SympyCurve (class in physicsnemo.sym.geometry.curve) SympyToTorch (class in physicsnemo.sym.utils.sympy.torch_printer) SyntheticWeatherDataLoader (class in physicsnemo.datapipes.climate.synthetic) SyntheticWeatherDataset (class in physicsnemo.datapipes.climate.synthetic) T t0 (physicsnemo.sym.hydra.optimizer.ASGDConf attribute) T_0 (physicsnemo.sym.hydra.scheduler.CosineAnnealingWarmRestartsConf attribute) T_max (physicsnemo.sym.hydra.scheduler.CosineAnnealingLRConf attribute) T_mult (physicsnemo.sym.hydra.scheduler.CosineAnnealingWarmRestartsConf attribute) tau (physicsnemo.sym.hydra.loss.AggregatorRelobraloConf attribute) Tensor Parallel TensorBoardProfiler (class in physicsnemo.sym.hydra.profiler) Tessellation (class in physicsnemo.sym.geometry.tessellation) test_function (physicsnemo.sym.hydra.training.VariationalTraining attribute) Tetrahedron (class in physicsnemo.sym.geometry.primitives_3d) TFExponentialLRConf (class in physicsnemo.sym.hydra.scheduler) time_range() (in module physicsnemo.utils.diffusion.utils) TimeSeriesDataset (class in physicsnemo.datapipes.healpix.timeseries_dataset) to() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) to_absolute_path() (in module physicsnemo.sym.hydra.utils) to_local() (physicsnemo.distributed.shard_tensor.ShardTensor method) to_yaml() (in module physicsnemo.sym.hydra.utils) toggle_mlflow() (physicsnemo.launch.logging.launch.LaunchLogger class method) toggle_wandb() (physicsnemo.launch.logging.launch.LaunchLogger class method) tolerance_change (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) tolerance_grad (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) torch_lambdify() (in module physicsnemo.sym.utils.sympy.torch_printer) Torus (class in physicsnemo.sym.geometry.primitives_3d) Trainer (class in physicsnemo.sym.trainer) training (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) TrainingConf (class in physicsnemo.sym.hydra.training) transfinite_interpolation() (physicsnemo.sym.geometry.adf.ADF static method) transform_fourier_features (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) transformer (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) translate() (physicsnemo.sym.geometry.curve.Curve method) (physicsnemo.sym.geometry.geometry.Geometry method) Triangle (class in physicsnemo.sym.geometry.primitives_2d) TriangularPrism (class in physicsnemo.sym.geometry.primitives_3d) trimmed_circle_adf() (physicsnemo.sym.geometry.adf.ADF static method) truncate_helper() (in module physicsnemo.distributed.utils) trunk_dim (physicsnemo.sym.hydra.arch.DeepOConf attribute) tuple_product() (in module physicsnemo.utils.diffusion.utils) U UNet (class in physicsnemo.models.diffusion.unet) unique_axis0_fast() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) unmark_module_as_shared() (in module physicsnemo.distributed.utils) unnormalize() (in module physicsnemo.utils.domino.utils) unstandardize() (in module physicsnemo.utils.domino.utils) update() (physicsnemo.metrics.general.ensemble_metrics.EnsembleMetrics method) (physicsnemo.metrics.general.ensemble_metrics.Mean method) (physicsnemo.metrics.general.ensemble_metrics.Variance method) (physicsnemo.metrics.general.histogram.Histogram method) update_freq (physicsnemo.sym.hydra.loss.AggregatorLRAnnealingConf attribute) (physicsnemo.sym.hydra.optimizer.ShampooConf attribute) use_fp16 (physicsnemo.models.diffusion.preconditioning.EDMPrecondSuperResolution property) (physicsnemo.models.diffusion.unet.UNet property) use_ntk (physicsnemo.sym.hydra.loss.NTKConf attribute) use_quadratures (physicsnemo.sym.hydra.training.VariationalTraining attribute) V Validator (class in physicsnemo.sym.domain.validator.validator) ValidatorPlotter (class in physicsnemo.sym.utils.io.plotter) var_to_polyvtk() (in module physicsnemo.sym.utils.io.vtk) Variance (class in physicsnemo.metrics.general.ensemble_metrics) VariationalConstraint (class in physicsnemo.sym.domain.constraint.continuous) VariationalDomainConstraint (class in physicsnemo.sym.domain.constraint.continuous) VariationalTraining (class in physicsnemo.sym.hydra.training) VectorizedBoxes (class in physicsnemo.sym.geometry.primitives_3d) VEPrecond (class in physicsnemo.models.diffusion.preconditioning) VEPrecond_dfsr (class in physicsnemo.models.diffusion.preconditioning) VEPrecond_dfsr_cond (class in physicsnemo.models.diffusion.preconditioning) VEPrecondMetaData (class in physicsnemo.models.diffusion.preconditioning) version (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) volume() (physicsnemo.sym.geometry.parameterization.Bounds method) volume_factors (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) volume_points_sample (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) volume_variables (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) voriticity_residual() (physicsnemo.models.diffusion.preconditioning.VEPrecond_dfsr_cond method) VortexSheddingDataset (class in physicsnemo.datapipes.gnn.vortex_shedding_dataset) VoxelInferencer (class in physicsnemo.sym.domain.inferencer.voxel) VPPrecond (class in physicsnemo.models.diffusion.preconditioning) VPPrecondMetaData (class in physicsnemo.models.diffusion.preconditioning) VTKFromFile (class in physicsnemo.sym.utils.io.vtk) VTKPolyData (class in physicsnemo.sym.utils.io.vtk) VTKRectilinearGrid (class in physicsnemo.sym.utils.io.vtk) VTKStructuredGrid (class in physicsnemo.sym.utils.io.vtk) VTKUniformGrid (class in physicsnemo.sym.utils.io.vtk) VTKUnstructuredGrid (class in physicsnemo.sym.utils.io.vtk) W warmup (physicsnemo.sym.hydra.optimizer.ApolloConf attribute) (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) warmup_init (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) warning() (physicsnemo.launch.logging.console.PythonLogger method) wasserstein_from_cdf() (in module physicsnemo.metrics.general.wasserstein) wasserstein_from_normal() (in module physicsnemo.metrics.general.wasserstein) wasserstein_from_samples() (in module physicsnemo.metrics.general.wasserstein) WaveEquation (class in physicsnemo.sym.eq.pdes.wave_equation) wd_ratio (physicsnemo.sym.hydra.optimizer.AdamPConf attribute) (physicsnemo.sym.hydra.optimizer.SGDPConf attribute) Weak Scaling weight_decay (physicsnemo.sym.hydra.optimizer.AccSGDConf attribute) (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) (physicsnemo.sym.hydra.optimizer.AdaBoundConf attribute) (physicsnemo.sym.hydra.optimizer.AdadeltaConf attribute) (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) (physicsnemo.sym.hydra.optimizer.AdagradConf attribute) (physicsnemo.sym.hydra.optimizer.AdahessianConf attribute) (physicsnemo.sym.hydra.optimizer.AdamaxConf attribute) (physicsnemo.sym.hydra.optimizer.AdamConf attribute) (physicsnemo.sym.hydra.optimizer.AdaModConf attribute) (physicsnemo.sym.hydra.optimizer.AdamPConf attribute) (physicsnemo.sym.hydra.optimizer.AdamWConf attribute) (physicsnemo.sym.hydra.optimizer.AggMoConf attribute) (physicsnemo.sym.hydra.optimizer.ApolloConf attribute) (physicsnemo.sym.hydra.optimizer.ASGDConf attribute) (physicsnemo.sym.hydra.optimizer.DiffGradConf attribute) (physicsnemo.sym.hydra.optimizer.LambConf attribute) (physicsnemo.sym.hydra.optimizer.MADGRADConf attribute) (physicsnemo.sym.hydra.optimizer.NAdamConf attribute) (physicsnemo.sym.hydra.optimizer.NovoGradConf attribute) (physicsnemo.sym.hydra.optimizer.PIDConf attribute) (physicsnemo.sym.hydra.optimizer.QHAdamConf attribute) (physicsnemo.sym.hydra.optimizer.QHMConf attribute) (physicsnemo.sym.hydra.optimizer.RAdamConf attribute) (physicsnemo.sym.hydra.optimizer.RangerConf attribute) (physicsnemo.sym.hydra.optimizer.RangerQHConf attribute) (physicsnemo.sym.hydra.optimizer.RangerVAConf attribute) (physicsnemo.sym.hydra.optimizer.RMSpropConf attribute) (physicsnemo.sym.hydra.optimizer.SGDConf attribute) (physicsnemo.sym.hydra.optimizer.SGDPConf attribute) (physicsnemo.sym.hydra.optimizer.SGDWConf attribute) (physicsnemo.sym.hydra.optimizer.ShampooConf attribute) (physicsnemo.sym.hydra.optimizer.SWATSConf attribute) (physicsnemo.sym.hydra.optimizer.YogiConf attribute) weight_decay_type (physicsnemo.sym.hydra.optimizer.QHMConf attribute) weight_decouple (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) weight_norm (physicsnemo.sym.hydra.arch.ConvFullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.FourierConf attribute) (physicsnemo.sym.hydra.arch.FullyConnectedConf attribute) (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) (physicsnemo.sym.hydra.arch.ModifiedFourierConf attribute) (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) (physicsnemo.sym.hydra.arch.MultiscaleFourierConf attribute) WeightedMean (class in physicsnemo.metrics.general.reduction) WeightedStatistic (class in physicsnemo.metrics.general.reduction) WeightedVariance (class in physicsnemo.metrics.general.reduction) weights (physicsnemo.sym.hydra.loss.LossConf attribute) world_size (physicsnemo.distributed.manager.DistributedManager property) write_input() (physicsnemo.utils.corrdiff.utils.NetCDFWriter method) write_prediction() (physicsnemo.utils.corrdiff.utils.NetCDFWriter method) write_time() (physicsnemo.utils.corrdiff.utils.NetCDFWriter method) write_to_vtp() (in module physicsnemo.utils.domino.utils) write_to_vtu() (in module physicsnemo.utils.domino.utils) write_truth() (physicsnemo.utils.corrdiff.utils.NetCDFWriter method) X xi (physicsnemo.sym.hydra.optimizer.AccSGDConf attribute) xyz2latlon() (in module physicsnemo.utils.graphcast.graph_utils) Y YogiConf (class in physicsnemo.sym.hydra.optimizer) Z ZeroEquation (class in physicsnemo.sym.eq.pdes.turbulence_zero_eq) zonal_mean() (in module physicsnemo.metrics.climate.reduction) zonal_var() (in module physicsnemo.metrics.climate.reduction)
