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 _ __getitem__() (physicsnemo.active_learning.protocols.DataPool method) __init__() (physicsnemo.optim.CombinedOptimizer method) __is_external_process__ (physicsnemo.active_learning.protocols.LabelStrategy attribute) __iter__() (physicsnemo.active_learning.protocols.DataPool method) __len__() (physicsnemo.active_learning.protocols.DataPool method) __protocol_name__ (physicsnemo.active_learning.protocols.ActiveLearningProtocol attribute), [1] __provides_fields__ (physicsnemo.active_learning.protocols.LabelStrategy attribute) _args (physicsnemo.active_learning.protocols.ActiveLearningProtocol attribute), [1] _registry (in module physicsnemo.active_learning) _setup_logger() (physicsnemo.active_learning.protocols.ActiveLearningProtocol method), [1] A A2GradExpConf (class in physicsnemo.sym.hydra.optimizer) A2GradIncConf (class in physicsnemo.sym.hydra.optimizer) A2GradUniConf (class in physicsnemo.sym.hydra.optimizer) AbstractQueue (class in physicsnemo.active_learning.protocols) acc() (in module physicsnemo.metrics.climate.acc) AccSGDConf (class in physicsnemo.sym.hydra.optimizer) activation() (physicsnemo.models.dlwp.dlwp.DLWP method) 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) active_learning_phase (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] active_learning_step() (physicsnemo.active_learning.driver.Driver method) (physicsnemo.active_learning.protocols.DriverProtocol method) active_learning_step_idx (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] (physicsnemo.active_learning.driver.Driver attribute) (physicsnemo.active_learning.driver.Driver property) ActiveLearningCheckpoint (class in physicsnemo.active_learning.driver) ActiveLearningPhase (class in physicsnemo.active_learning.protocols) ActiveLearningProtocol (class in physicsnemo.active_learning.protocols), [1] 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() (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_param_group() (physicsnemo.optim.CombinedOptimizer method) 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) (class in physicsnemo.nn.module.afno_layers) AFNOArch (class in physicsnemo.sym.models.afno) AFNOConf (class in physicsnemo.sym.hydra.arch) AFNOMlp (class in physicsnemo.models.afno.afno) (class in physicsnemo.nn.module.afno_layers) AFNOPatchEmbed (class in physicsnemo.nn.module.afno_layers) AggMoConf (class in physicsnemo.sym.hydra.optimizer) 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) 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.diffusion.preconditioners.iDDPMPrecond method) amp_mode (physicsnemo.diffusion.preconditioners.EDMPrecondSuperResolution property) (physicsnemo.models.diffusion_unets.CorrDiffRegressionUNet property) (physicsnemo.models.diffusion_unets.DhariwalUNet property) (physicsnemo.models.diffusion_unets.SongUNet property) amp_type (physicsnemo.active_learning.loop.DefaultTrainingLoop 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) append() (physicsnemo.active_learning.protocols.DataPool method), [1] (physicsnemo.active_learning.protocols.MetrologyStrategy method) apply_activation() (physicsnemo.nn.module.fully_connected_layers.ConvFCLayer 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) ASGDConf (class in physicsnemo.sym.hydra.optimizer) attach() (physicsnemo.active_learning.protocols.ActiveLearningProtocol method), [1], [2], [3] attach_strategies() (physicsnemo.active_learning.driver.Driver method) (physicsnemo.active_learning.protocols.DriverProtocol method) AttentionOp (class in physicsnemo.nn.module.attention_layers) 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.nn.module.attention_layers.AttentionOp static method) (physicsnemo.nn.module.fused_silu.FusedSiLU static method) (physicsnemo.nn.module.fused_silu.FusedSiLU_deriv_1 static method) (physicsnemo.nn.module.fused_silu.FusedSiLU_deriv_2 static method) (physicsnemo.nn.module.fused_silu.FusedSiLU_deriv_3 static method) (physicsnemo.sym.loss.loss.LossL2 static method) barrier() (physicsnemo.active_learning.driver.Driver method) base_resolution (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) batch_norm (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) batch_size (physicsnemo.active_learning.config.DriverConfig attribute), [1] (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) Block (class in physicsnemo.models.afno.afno) 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.nn.module.ball_query) branch_dim (physicsnemo.sym.hydra.arch.DeepOConf attribute) broadcast_buffers (physicsnemo.distributed.manager.DistributedManager property) (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) C CachedDoMINODataset (class in physicsnemo.datapipes.cae.domino_datapipe) caching (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) calc_latent_derivatives() (in module physicsnemo.nn.module.spectral_layers) CappedGELU (class in physicsnemo.nn.module.activations) CappedLeakyReLU (class in physicsnemo.nn.module.activations) CausalLossNorm (class in physicsnemo.sym.loss.loss) cdf() (in module physicsnemo.metrics.general.histogram) cell_to_adj() (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) 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_ort_install() (in module physicsnemo.deploy.onnx.utils) checkpoint_dir (physicsnemo.active_learning.protocols.ActiveLearningProtocol property), [1] checkpoint_interval (physicsnemo.active_learning.config.DriverConfig attribute), [1] checkpoint_on_labeling (physicsnemo.active_learning.config.DriverConfig attribute), [1] checkpoint_on_metrology (physicsnemo.active_learning.config.DriverConfig attribute), [1] checkpoint_on_query (physicsnemo.active_learning.config.DriverConfig attribute), [1] checkpoint_on_training (physicsnemo.active_learning.config.DriverConfig attribute), [1] 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) collate_fn (physicsnemo.active_learning.config.DriverConfig attribute), [1] Collectives CombinedOptimizer (class in physicsnemo.optim) compl_mul1d() (physicsnemo.nn.module.spectral_layers.SpectralConv1d method) compl_mul2d() (physicsnemo.nn.module.spectral_layers.SpectralConv2d method) compl_mul3d() (physicsnemo.nn.module.spectral_layers.SpectralConv3d method) compl_mul4d() (physicsnemo.nn.module.spectral_layers.SpectralConv4d method) compose() (in module physicsnemo.sym.hydra.utils) CompressibleIntegralContinuity (class in physicsnemo.sym.eq.pdes.navier_stokes) compute() (physicsnemo.active_learning.protocols.MetrologyStrategy method) 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_scaling_factors() (in module physicsnemo.datapipes.cae.domino_datapipe) compute_stencil2d() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) compute_stencil3d() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) compute_stl_scaling_and_surface_grids() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) compute_volume_scaling_and_grids() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) Cone (class in physicsnemo.sym.geometry.primitives_3d) config (physicsnemo.active_learning.driver.Driver attribute) configure_optimizer() (physicsnemo.active_learning.driver.Driver method) Constraint (class in physicsnemo.sym.domain.constraint.constraint) construct() (in module physicsnemo.active_learning) Conv1dFCLayer (class in physicsnemo.nn.module.fully_connected_layers) Conv2d (class in physicsnemo.nn.module.conv_layers) Conv2dFCLayer (class in physicsnemo.nn.module.fully_connected_layers) Conv3dFCLayer (class in physicsnemo.nn.module.fully_connected_layers) conv_layer_size (physicsnemo.sym.hydra.arch.Pix2PixConf attribute) (physicsnemo.sym.hydra.arch.SRResConf attribute) ConvBlock (class in physicsnemo.nn.module.conv_layers) convert_config() (physicsnemo.sym.key.Key static method) ConvFCLayer (class in physicsnemo.nn.module.fully_connected_layers) ConvFullyConnectedConf (class in physicsnemo.sym.hydra.arch) ConvGRULayer (class in physicsnemo.nn.module.conv_layers) ConvLayer (class in physicsnemo.nn.module.conv_layers) ConvNdFCLayer (class in physicsnemo.nn.module.fully_connected_layers) ConvNdKernel1Layer (class in physicsnemo.nn.module.fully_connected_layers) ConvolutionalBlock3d (class in physicsnemo.models.srrn.super_res_net) ConvResidualBlock (class in physicsnemo.nn.module.conv_layers) coord_features (physicsnemo.sym.hydra.arch.FNOConf attribute) copy_to_parallel_region() (in module physicsnemo.distributed.mappings) CorrDiffRegressionUNet (class in physicsnemo.models.diffusion_unets) CosineAnnealingLRConf (class in physicsnemo.sym.hydra.scheduler) CosineAnnealingWarmRestartsConf (class in physicsnemo.sym.hydra.scheduler) create_graph() (physicsnemo.datapipes.gnn.ahmed_body_dataset.AhmedBodyDataset method) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static 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) CubeEmbedding (class in physicsnemo.nn.module.conv_layers) 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) current_phase (physicsnemo.active_learning.driver.Driver attribute) Curve (class in physicsnemo.sym.geometry.curve) custom (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) 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_INTEGRATION (physicsnemo.active_learning.protocols.ActiveLearningPhase attribute) data_path (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) DataPool (class in physicsnemo.active_learning.protocols) datefmt (physicsnemo.sym.hydra.hydra.DebugFormat attribute) (physicsnemo.sym.hydra.hydra.SimpleFormat attribute) 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) DecoderLayer (class in physicsnemo.nn.module.transformer_layers) DecoderOnlyLayer (class in physicsnemo.nn.module.transformer_decoder) 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) DefaultTrainingLoop (class in physicsnemo.active_learning.loop) 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) device (physicsnemo.active_learning.config.DriverConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) (physicsnemo.distributed.manager.DistributedManager property) (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) DeviceMesh DGMArch (class in physicsnemo.sym.models.dgm) DGMLayer (class in physicsnemo.nn.module.dgm_layers) DhariwalUNet (class in physicsnemo.models.diffusion_unets) dict_to_csv() (in module physicsnemo.sym.utils.io.csv_rw) DiffGradConf (class in physicsnemo.sym.hydra.optimizer) Diffusion (class in physicsnemo.sym.eq.pdes.diffusion) 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) dist_manager (physicsnemo.active_learning.config.DriverConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) 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) DoMINODataConfig (class in physicsnemo.datapipes.cae.domino_datapipe) DoMINODataPipe (class in physicsnemo.datapipes.cae.domino_datapipe) DownSample2D (class in physicsnemo.nn.module.resample_layers) DownSample3D (class in physicsnemo.nn.module.resample_layers) downsample_geometry() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) DrivAerNetDataset (class in physicsnemo.datapipes.gnn.drivaernet_dataset) Driver (class in physicsnemo.active_learning.driver) driver_config (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] DriverConfig (class in physicsnemo.active_learning.config) DriverProtocol (class in physicsnemo.active_learning.protocols) drop_path() (in module physicsnemo.nn.functional) DropPath (class in physicsnemo.nn.module.drop) dtype (physicsnemo.active_learning.config.DriverConfig attribute), [1] E EarthAttention2D (class in physicsnemo.nn.module.attention_layers) EarthAttention3D (class in physicsnemo.nn.module.attention_layers) edges_to_adjacency() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) EDMPrecond (class in physicsnemo.diffusion.preconditioners) EDMPrecondSuperResolution (class in physicsnemo.diffusion.preconditioners) 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) empty() (physicsnemo.active_learning.protocols.AbstractQueue method) EncoderLayer (class in physicsnemo.nn.module.transformer_layers) 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) 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) extra_repr() (physicsnemo.nn.module.drop.DropPath method) (physicsnemo.nn.module.weight_fact.WeightFactLinear method) (physicsnemo.nn.module.weight_norm.WeightNormLinear method) F FCLayer (class in physicsnemo.nn.module.fully_connected_layers) Fengwu (class in physicsnemo.models.fengwu.fengwu) 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) fine_tuning_lr (physicsnemo.active_learning.config.DriverConfig attribute), [1] finest_resolution (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) first_omega (physicsnemo.sym.hydra.arch.SirenConf attribute) first_order_pino_grads() (in module physicsnemo.nn.module.spectral_layers) 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) (class in physicsnemo.nn.module.fno_layers) FNO2DEncoder (class in physicsnemo.models.fno.fno) (class in physicsnemo.nn.module.fno_layers) FNO3DEncoder (class in physicsnemo.models.fno.fno) (class in physicsnemo.nn.module.fno_layers) FNO4DEncoder (class in physicsnemo.models.fno.fno) (class in physicsnemo.nn.module.fno_layers) 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) formatters (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) forward() (physicsnemo.active_learning.protocols.LearnerProtocol method) (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.dlwp.dlwp.DLWP method) (physicsnemo.models.dlwp_healpix.HEALPixRecUNet.HEALPixRecUNet method) (physicsnemo.models.fengwu.fengwu.Fengwu method) (physicsnemo.models.pangu.pangu.Pangu method) (physicsnemo.models.swinvrnn.swinvrnn.SwinRNN method) (physicsnemo.nn.module.activations.CappedGELU method) (physicsnemo.nn.module.activations.CappedLeakyReLU method) (physicsnemo.nn.module.activations.Identity method) (physicsnemo.nn.module.activations.SquarePlus method) (physicsnemo.nn.module.activations.Stan method) (physicsnemo.nn.module.afno_layers.AFNO2DLayer method) (physicsnemo.nn.module.afno_layers.AFNOMlp method) (physicsnemo.nn.module.afno_layers.AFNOPatchEmbed method) (physicsnemo.nn.module.afno_layers.ModAFNO2DLayer method) (physicsnemo.nn.module.afno_layers.ModAFNOMlp method) (physicsnemo.nn.module.afno_layers.ScaleShiftMlp method) (physicsnemo.nn.module.attention_layers.AttentionOp static method) (physicsnemo.nn.module.attention_layers.EarthAttention2D method) (physicsnemo.nn.module.attention_layers.EarthAttention3D method) (physicsnemo.nn.module.attention_layers.UNetAttention method) (physicsnemo.nn.module.ball_query.BQWarp method) (physicsnemo.nn.module.conv_layers.Conv2d method) (physicsnemo.nn.module.conv_layers.ConvBlock method) (physicsnemo.nn.module.conv_layers.ConvGRULayer method) (physicsnemo.nn.module.conv_layers.ConvLayer method) (physicsnemo.nn.module.conv_layers.ConvResidualBlock method) (physicsnemo.nn.module.conv_layers.CubeEmbedding method) (physicsnemo.nn.module.conv_layers.TransposeConvLayer method) (physicsnemo.nn.module.dgm_layers.DGMLayer method) (physicsnemo.nn.module.drop.DropPath method) (physicsnemo.nn.module.embedding_layers.FourierEmbedding method) (physicsnemo.nn.module.embedding_layers.OneHotEmbedding method) (physicsnemo.nn.module.embedding_layers.PositionalEmbedding method) (physicsnemo.nn.module.embedding_layers.SinusoidalTimestepEmbedding method) (physicsnemo.nn.module.fft.OnnxIrfft static method) (physicsnemo.nn.module.fft.OnnxIrfft2 static method) (physicsnemo.nn.module.fft.OnnxRfft static method) (physicsnemo.nn.module.fft.OnnxRfft2 static method) (physicsnemo.nn.module.fno_layers.FNO1DEncoder method) (physicsnemo.nn.module.fno_layers.FNO2DEncoder method) (physicsnemo.nn.module.fno_layers.FNO3DEncoder method) (physicsnemo.nn.module.fno_layers.FNO4DEncoder method) (physicsnemo.nn.module.fourier_layers.FourierFilter method) (physicsnemo.nn.module.fourier_layers.FourierLayer method) (physicsnemo.nn.module.fourier_layers.FourierMLP method) (physicsnemo.nn.module.fourier_layers.GaborFilter method) (physicsnemo.nn.module.fully_connected_layers.Conv1dFCLayer method) (physicsnemo.nn.module.fully_connected_layers.Conv2dFCLayer method) (physicsnemo.nn.module.fully_connected_layers.Conv3dFCLayer method) (physicsnemo.nn.module.fully_connected_layers.ConvNdFCLayer method) (physicsnemo.nn.module.fully_connected_layers.ConvNdKernel1Layer method) (physicsnemo.nn.module.fully_connected_layers.FCLayer method) (physicsnemo.nn.module.fully_connected_layers.Linear method) (physicsnemo.nn.module.fused_silu.FusedSiLU static method) (physicsnemo.nn.module.fused_silu.FusedSiLU_deriv_1 static method) (physicsnemo.nn.module.fused_silu.FusedSiLU_deriv_2 static method) (physicsnemo.nn.module.fused_silu.FusedSiLU_deriv_3 static method) (physicsnemo.nn.module.group_norm.GroupNorm method) (physicsnemo.nn.module.gumbel_softmax.GumbelSoftmax method) (physicsnemo.nn.module.kan_layers.KolmogorovArnoldNetwork method) (physicsnemo.nn.module.mlp_layers.Mlp method) (physicsnemo.nn.module.physics_attention.PhysicsAttentionBase method) (physicsnemo.nn.module.resample_layers.DownSample2D method) (physicsnemo.nn.module.resample_layers.DownSample3D method) (physicsnemo.nn.module.resample_layers.UpSample2D method) (physicsnemo.nn.module.resample_layers.UpSample3D method) (physicsnemo.nn.module.siren_layers.SirenLayer method) (physicsnemo.nn.module.spectral_layers.SpectralConv1d method) (physicsnemo.nn.module.spectral_layers.SpectralConv2d method) (physicsnemo.nn.module.spectral_layers.SpectralConv3d method) (physicsnemo.nn.module.spectral_layers.SpectralConv4d method) (physicsnemo.nn.module.transformer_decoder.DecoderOnlyLayer method) (physicsnemo.nn.module.transformer_decoder.TransformerDecoder method) (physicsnemo.nn.module.transformer_layers.DecoderLayer method) (physicsnemo.nn.module.transformer_layers.EncoderLayer method) (physicsnemo.nn.module.transformer_layers.FuserLayer method) (physicsnemo.nn.module.transformer_layers.SwinTransformer method) (physicsnemo.nn.module.transformer_layers.Transformer2DBlock method) (physicsnemo.nn.module.transformer_layers.Transformer3DBlock method) (physicsnemo.nn.module.unet_layers.UNetBlock method) (physicsnemo.nn.module.weight_fact.WeightFactLinear method) (physicsnemo.nn.module.weight_norm.WeightNormLinear 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.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) fourier_derivatives() (in module physicsnemo.nn.module.spectral_layers) fourier_encode() (in module physicsnemo.nn.module.fourier_layers) FourierConf (class in physicsnemo.sym.hydra.arch) FourierEmbedding (class in physicsnemo.nn.module.embedding_layers) FourierFilter (class in physicsnemo.nn.module.fourier_layers) FourierLayer (class in physicsnemo.nn.module.fourier_layers) FourierMLP (class in physicsnemo.nn.module.fourier_layers) 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_dict() (physicsnemo.active_learning.config.OptimizerConfig class method) (physicsnemo.active_learning.config.StrategiesConfig class method) (physicsnemo.active_learning.config.TrainingConfig class method) from_json() (physicsnemo.active_learning.config.DriverConfig class 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) (physicsnemo.sym.geometry.tessellation_warp.Tessellation class method) from_sympy() (physicsnemo.sym.node.Node class method) FSDP FullyConnected (class in physicsnemo.models.mlp.fully_connected) FullyConnectedArch (class in physicsnemo.sym.models.fully_connected) FullyConnectedConf (class in physicsnemo.sym.hydra.arch) FusedFourierNetConf (class in physicsnemo.sym.hydra.arch) FusedGridEncodingNetConf (class in physicsnemo.sym.hydra.arch) FusedMLPConf (class in physicsnemo.sym.hydra.arch) FusedSiLU (class in physicsnemo.nn.module.fused_silu) FusedSiLU_deriv_1 (class in physicsnemo.nn.module.fused_silu) FusedSiLU_deriv_2 (class in physicsnemo.nn.module.fused_silu) FusedSiLU_deriv_3 (class in physicsnemo.nn.module.fused_silu) FuserLayer (class in physicsnemo.nn.module.transformer_layers) G gabor_alpha (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) gabor_beta (physicsnemo.sym.hydra.arch.MultiplicativeFilterConf attribute) GaborFilter (class in physicsnemo.nn.module.fourier_layers) 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) 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) GeometrySource (class in physicsnemo.sym.geometry.geometry_dataloader) get() (physicsnemo.active_learning.protocols.AbstractQueue method) get_activation() (in module physicsnemo.nn.module.activations) get_activation_function() (physicsnemo.nn.module.group_norm.GroupNorm method) get_available_backend() (physicsnemo.distributed.manager.DistributedManager static method) get_constants() (physicsnemo.datapipes.healpix.timeseries_dataset.TimeSeriesDataset method) get_dataloader() (physicsnemo.sym.domain.constraint.constraint.Constraint static method) get_gradient_module() (physicsnemo.sym.eq.spatial_grads.spatial_grads.GradientCalculator method) get_group_norm() (in module physicsnemo.nn.module.group_norm) get_layer_norm_class() (in module physicsnemo.nn.module.layer_norm) get_memory_format() (in module physicsnemo.distributed.utils) get_mesh_group() (physicsnemo.distributed.manager.DistributedManager 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) 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) (physicsnemo.nn.module.fno_layers.FNO1DEncoder method) (physicsnemo.nn.module.fno_layers.FNO2DEncoder method) (physicsnemo.nn.module.fno_layers.FNO3DEncoder method) (physicsnemo.nn.module.fno_layers.FNO4DEncoder method) grid_to_vtk() (in module physicsnemo.sym.utils.io.vtk) 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) GroupNorm (class in physicsnemo.nn.module.group_norm) gumbel_softmax() (in module physicsnemo.nn.module.gumbel_softmax) GumbelSoftmax (class in physicsnemo.nn.module.gumbel_softmax) H handlers (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) has_label_queue (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] has_query_queue (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] 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.diffusion.preconditioners) Identity (class in physicsnemo.nn.module.activations) ignore_missing_extra_state_key() (in module physicsnemo.nn.module.layer_norm) imag() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) 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) inference_fn (physicsnemo.active_learning.driver.Driver attribute) inference_step() (physicsnemo.active_learning.protocols.LearnerProtocol method) InferenceProtocol (class in physicsnemo.active_learning.protocols) 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) 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) initialise_parameters() (physicsnemo.nn.module.fully_connected_layers.ConvNdFCLayer method) initialization_network_dir (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) initialize() (physicsnemo.distributed.manager.DistributedManager 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_open_mpi() (physicsnemo.distributed.manager.DistributedManager static method) initialize_slurm() (physicsnemo.distributed.manager.DistributedManager static method) 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) interpolation() (in module physicsnemo.nn.functional) Intra-Node irfft() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) irfft2() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) is_attached (physicsnemo.active_learning.protocols.ActiveLearningProtocol property), [1] is_initialized() (physicsnemo.distributed.manager.DistributedManager class method) is_lr_scheduler_configured (physicsnemo.active_learning.driver.Driver property) is_optimizer_configured (physicsnemo.active_learning.driver.Driver property) is_registered() (in module physicsnemo.active_learning) 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() (in module physicsnemo.nn.functional) KolmogorovArnoldNetwork (class in physicsnemo.nn.module.kan_layers) L label() (physicsnemo.active_learning.protocols.LabelStrategy method) label_queue (physicsnemo.active_learning.driver.Driver attribute) (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] label_strategy (physicsnemo.active_learning.config.StrategiesConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] LABELING (physicsnemo.active_learning.protocols.ActiveLearningPhase attribute) LabelStrategy (class in physicsnemo.active_learning.protocols) LambConf (class in physicsnemo.sym.hydra.optimizer) lambd (physicsnemo.sym.hydra.optimizer.ASGDConf attribute) large_kernel_size (physicsnemo.sym.hydra.arch.SRResConf attribute) last_checkpoint (physicsnemo.active_learning.driver.Driver property) last_epoch (physicsnemo.sym.hydra.scheduler.CosineAnnealingLRConf attribute) (physicsnemo.sym.hydra.scheduler.CosineAnnealingWarmRestartsConf attribute) 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) LayerNorm (class in physicsnemo.nn.module.layer_norm) learner (physicsnemo.active_learning.driver.Driver attribute) (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] LearnerProtocol (class in physicsnemo.active_learning.protocols) 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) Linear (class in physicsnemo.nn.module.fully_connected_layers) 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) load_checkpoint() (physicsnemo.active_learning.driver.Driver class method) load_json() (in module physicsnemo.datapipes.gnn.utils) load_records() (physicsnemo.active_learning.protocols.MetrologyStrategy method) load_state_dict() (physicsnemo.optim.CombinedOptimizer method) load_statistics() (physicsnemo.datapipes.cae.mesh_datapipe.MeshDatapipe method) (physicsnemo.datapipes.climate.era5_hdf5.ERA5HDF5Datapipe method) load_training_checkpoint() (physicsnemo.active_learning.loop.DefaultTrainingLoop static method) Local Rank local_rank (physicsnemo.distributed.manager.DistributedManager property) log2_hashmap_size (physicsnemo.sym.hydra.arch.FusedGridEncodingNetConf attribute) (physicsnemo.sym.hydra.arch.MultiresolutionHashNetConf attribute) log_dir (physicsnemo.active_learning.driver.Driver property) logger (physicsnemo.active_learning.driver.Driver attribute) (physicsnemo.active_learning.protocols.ActiveLearningProtocol property), [1] 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) lr_scheduler (physicsnemo.active_learning.driver.Driver attribute) lr_scheduler_state (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] 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_active_learning_steps (physicsnemo.active_learning.config.DriverConfig attribute), [1] max_eval (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) max_fine_tuning_epochs (physicsnemo.active_learning.config.TrainingConfig attribute), [1] max_iter (physicsnemo.sym.hydra.optimizer.BFGSConf attribute) max_samples (physicsnemo.active_learning.protocols.QueryStrategy attribute), [1] max_steps (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) max_training_epochs (physicsnemo.active_learning.config.TrainingConfig attribute), [1] MaxwellFreqReal (class in physicsnemo.sym.eq.pdes.electromagnetic) Mean (class in physicsnemo.metrics.general.ensemble_metrics) mean (physicsnemo.metrics.general.ensemble_metrics.Variance property) mesh() (physicsnemo.distributed.manager.DistributedManager method) mesh_dims (physicsnemo.distributed.manager.DistributedManager property) mesh_names() (physicsnemo.distributed.manager.DistributedManager method) mesh_sizes() (physicsnemo.distributed.manager.DistributedManager method) MeshDaliExternalSource (class in physicsnemo.datapipes.cae.mesh_datapipe) MeshDatapipe (class in physicsnemo.datapipes.cae.mesh_datapipe) 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) metric (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) METROLOGY (physicsnemo.active_learning.protocols.ActiveLearningPhase attribute) metrology_strategies (physicsnemo.active_learning.config.StrategiesConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] MetrologyStrategy (class in physicsnemo.active_learning.protocols) min_delta (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) Mlp (class in physicsnemo.nn.module.mlp_layers) ModAFNO (class in physicsnemo.models.afno.modafno) ModAFNO2DLayer (class in physicsnemo.nn.module.afno_layers) ModAFNOMlp (class in physicsnemo.nn.module.afno_layers) mode (physicsnemo.sym.hydra.training.DefaultStopCriterion attribute) (physicsnemo.sym.hydra.training.StopCriterionConf attribute) Model Parallel model_checkpoint_frequency (physicsnemo.active_learning.config.DriverConfig attribute), [1] 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.active_learning.config physicsnemo.active_learning.driver physicsnemo.active_learning.loop physicsnemo.active_learning.protocols 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.diffusion.multi_diffusion physicsnemo.diffusion.samplers physicsnemo.diffusion.samplers.deterministic_sampler physicsnemo.diffusion.samplers.stochastic_sampler physicsnemo.diffusion.utils physicsnemo.distributed physicsnemo.distributed.autograd physicsnemo.distributed.fft physicsnemo.distributed.manager physicsnemo.distributed.mappings physicsnemo.distributed.utils 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.nn.module.activations physicsnemo.nn.module.afno_layers physicsnemo.nn.module.attention_layers physicsnemo.nn.module.ball_query physicsnemo.nn.module.conv_layers physicsnemo.nn.module.dgm_layers physicsnemo.nn.module.drop physicsnemo.nn.module.embedding_layers physicsnemo.nn.module.fft physicsnemo.nn.module.fno_layers physicsnemo.nn.module.fourier_layers physicsnemo.nn.module.fully_connected_layers physicsnemo.nn.module.fused_silu physicsnemo.nn.module.gnn_layers physicsnemo.nn.module.group_norm physicsnemo.nn.module.gumbel_softmax physicsnemo.nn.module.hpx physicsnemo.nn.module.kan_layers physicsnemo.nn.module.layer_norm physicsnemo.nn.module.mlp_layers physicsnemo.nn.module.physics_attention physicsnemo.nn.module.resample_layers physicsnemo.nn.module.siren_layers physicsnemo.nn.module.spectral_layers physicsnemo.nn.module.transformer_decoder physicsnemo.nn.module.transformer_layers physicsnemo.nn.module.unet_layers physicsnemo.nn.module.weight_fact physicsnemo.nn.module.weight_norm physicsnemo.optim 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.geometry.tessellation_warp 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.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.insolation physicsnemo.utils.profiling 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) 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) NavierStokes (class in physicsnemo.sym.eq.pdes.navier_stokes) 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) network_dir (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) 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_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_dataloader_workers (physicsnemo.active_learning.config.DriverConfig attribute), [1] 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 omega (physicsnemo.sym.hydra.arch.SirenConf attribute) One2ManyRNN (class in physicsnemo.models.rnn.rnn_one2many) OneHotEmbedding (class in physicsnemo.nn.module.embedding_layers) OnnxIrfft (class in physicsnemo.nn.module.fft) OnnxIrfft2 (class in physicsnemo.nn.module.fft) OnnxRfft (class in physicsnemo.nn.module.fft) OnnxRfft2 (class in physicsnemo.nn.module.fft) optimizer (physicsnemo.active_learning.driver.Driver attribute) (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) optimizer_cls (physicsnemo.active_learning.config.OptimizerConfig attribute), [1] optimizer_config (physicsnemo.active_learning.config.TrainingConfig attribute), [1] optimizer_kwargs (physicsnemo.active_learning.config.OptimizerConfig attribute), [1] optimizer_state (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] OptimizerConf (class in physicsnemo.sym.hydra.optimizer) OptimizerConfig (class in physicsnemo.active_learning.config) OrderedParameterization (class in physicsnemo.sym.geometry.parameterization) output_keys (physicsnemo.sym.hydra.arch.ModelConf attribute) outputs (physicsnemo.sym.node.Node property) P pad_helper() (in module physicsnemo.distributed.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) parameters (physicsnemo.active_learning.protocols.LearnerProtocol property) 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) patch_size (physicsnemo.sym.hydra.arch.AFNOConf attribute) (physicsnemo.sym.hydra.arch.DistributedAFNOConf attribute) PatchEmbed (in module physicsnemo.models.afno.afno) (in module physicsnemo.nn.module.afno_layers) 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) PhysicsAttentionBase (class in physicsnemo.nn.module.physics_attention) PhysicsAttentionIrregularMesh (class in physicsnemo.nn.module.physics_attention) PhysicsAttentionStructuredMesh2D (class in physicsnemo.nn.module.physics_attention) PhysicsAttentionStructuredMesh3D (class in physicsnemo.nn.module.physics_attention) PhysicsInformer (class in physicsnemo.sym.eq.phy_informer) physicsnemo.active_learning.config module physicsnemo.active_learning.driver module physicsnemo.active_learning.loop module physicsnemo.active_learning.protocols module 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.diffusion.multi_diffusion module physicsnemo.diffusion.samplers module physicsnemo.diffusion.samplers.deterministic_sampler module physicsnemo.diffusion.samplers.stochastic_sampler module physicsnemo.diffusion.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.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.nn.module.activations module physicsnemo.nn.module.afno_layers module physicsnemo.nn.module.attention_layers module physicsnemo.nn.module.ball_query module physicsnemo.nn.module.conv_layers module physicsnemo.nn.module.dgm_layers module physicsnemo.nn.module.drop module physicsnemo.nn.module.embedding_layers module physicsnemo.nn.module.fft module physicsnemo.nn.module.fno_layers module physicsnemo.nn.module.fourier_layers module physicsnemo.nn.module.fully_connected_layers module physicsnemo.nn.module.fused_silu module physicsnemo.nn.module.gnn_layers module physicsnemo.nn.module.group_norm module physicsnemo.nn.module.gumbel_softmax module physicsnemo.nn.module.hpx module physicsnemo.nn.module.kan_layers module physicsnemo.nn.module.layer_norm module physicsnemo.nn.module.mlp_layers module physicsnemo.nn.module.physics_attention module physicsnemo.nn.module.resample_layers module physicsnemo.nn.module.siren_layers module physicsnemo.nn.module.spectral_layers module physicsnemo.nn.module.transformer_decoder module physicsnemo.nn.module.transformer_layers module physicsnemo.nn.module.unet_layers module physicsnemo.nn.module.weight_fact module physicsnemo.nn.module.weight_norm module physicsnemo.optim 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.geometry.tessellation_warp 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.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.insolation module physicsnemo.utils.profiling module physicsnemo.utils.zenith_angle module physicsnemo_version (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] 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) Pix2PixUnet (class in physicsnemo.models.pix2pix.pix2pixunet) 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) (physicsnemo.nn.module.fno_layers.FNO1DEncoder method) (physicsnemo.nn.module.fno_layers.FNO2DEncoder method) (physicsnemo.nn.module.fno_layers.FNO3DEncoder method) (physicsnemo.nn.module.fno_layers.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) Polygon (class in physicsnemo.sym.geometry.primitives_2d) positional_embedding_indexing() (physicsnemo.models.diffusion_unets.SongUNetPosEmbd method) positional_embedding_selector() (physicsnemo.models.diffusion_unets.SongUNetPosEmbd method) PositionalEmbedding (class in physicsnemo.nn.module.embedding_layers) prepare_input() (physicsnemo.models.fengwu.fengwu.Fengwu method) (physicsnemo.models.pangu.pangu.Pangu method) print_stats_freq (physicsnemo.sym.hydra.training.DefaultTraining attribute) (physicsnemo.sym.hydra.training.TrainingConf attribute) process_volume() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) profile (physicsnemo.sym.hydra.profiler.NvtxProfiler attribute) (physicsnemo.sym.hydra.profiler.ProfilerConf attribute) (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) profile_mode (physicsnemo.diffusion.preconditioners.EDMPrecondSuperResolution property) (physicsnemo.models.diffusion_unets.CorrDiffRegressionUNet property) (physicsnemo.models.diffusion_unets.DhariwalUNet property) (physicsnemo.models.diffusion_unets.SongUNet property) profiler (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) ProfilerConf (class in physicsnemo.sym.hydra.profiler) project_fourier_features (physicsnemo.sym.hydra.arch.HighwayFourierConf attribute) project_input_onto_slices() (physicsnemo.nn.module.physics_attention.PhysicsAttentionBase method) (physicsnemo.nn.module.physics_attention.PhysicsAttentionIrregularMesh method) (physicsnemo.nn.module.physics_attention.PhysicsAttentionStructuredMesh2D method) (physicsnemo.nn.module.physics_attention.PhysicsAttentionStructuredMesh3D method) Properties() (in module physicsnemo.active_learning) put() (physicsnemo.active_learning.protocols.AbstractQueue method) Q QHAdamConf (class in physicsnemo.sym.hydra.optimizer) QHMConf (class in physicsnemo.sym.hydra.optimizer) Quad_Rect (class in physicsnemo.sym.utils.vpinn.integral) QUERY (physicsnemo.active_learning.protocols.ActiveLearningPhase attribute) query_queue (physicsnemo.active_learning.driver.Driver attribute) (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] query_strategies (physicsnemo.active_learning.config.StrategiesConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] QueryStrategy (class in physicsnemo.active_learning.protocols) queue_cls (physicsnemo.active_learning.config.StrategiesConfig attribute), [1] R r_equivalence() (physicsnemo.sym.geometry.adf.ADF static method) RAdamConf (class in physicsnemo.sym.hydra.optimizer) RadialBasisArch (class in physicsnemo.sym.models.radial_basis) radius_search() (in module physicsnemo.nn.functional) 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) read_vtp_file() (in module physicsnemo.datapipes.gnn.utils) real() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) 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) records (physicsnemo.active_learning.protocols.MetrologyStrategy attribute), [1] Rectangle (class in physicsnemo.sym.geometry.primitives_2d) rectify (physicsnemo.sym.hydra.optimizer.AdaBeliefConf attribute) 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() (in module physicsnemo.active_learning) register_arch_configs() (in module physicsnemo.sym.hydra.arch) register_hydra_configs() (in module physicsnemo.sym.hydra.hydra) register_loss_configs() (in module physicsnemo.sym.hydra.loss) 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) registry (in module physicsnemo.active_learning) 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) remove_extra_state_hook_for_torch() (in module physicsnemo.nn.module.layer_norm) repeat (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) repeat() (physicsnemo.sym.geometry.geometry.Geometry method) required_inputs (physicsnemo.sym.eq.phy_informer.PhysicsInformer property) reset() (physicsnemo.active_learning.protocols.MetrologyStrategy method) (physicsnemo.models.dlwp_healpix.HEALPixRecUNet.HEALPixRecUNet method) reset_optim_states (physicsnemo.active_learning.config.DriverConfig attribute), [1] reset_parameters() (physicsnemo.nn.module.fourier_layers.FourierFilter method) (physicsnemo.nn.module.fourier_layers.GaborFilter method) (physicsnemo.nn.module.fully_connected_layers.Conv1dFCLayer method) (physicsnemo.nn.module.fully_connected_layers.Conv2dFCLayer method) (physicsnemo.nn.module.fully_connected_layers.Conv3dFCLayer method) (physicsnemo.nn.module.fully_connected_layers.ConvNdFCLayer method) (physicsnemo.nn.module.fully_connected_layers.FCLayer method) (physicsnemo.nn.module.siren_layers.SirenLayer method) (physicsnemo.nn.module.spectral_layers.SpectralConv1d method) (physicsnemo.nn.module.spectral_layers.SpectralConv2d method) (physicsnemo.nn.module.spectral_layers.SpectralConv3d method) (physicsnemo.nn.module.spectral_layers.SpectralConv4d method) (physicsnemo.nn.module.weight_fact.WeightFactLinear method) (physicsnemo.nn.module.weight_norm.WeightNormLinear method) ResidualConvBlock3d (class in physicsnemo.models.srrn.super_res_net) ResnetBlock (class in physicsnemo.models.pix2pix.pix2pix) ResNorm (class in physicsnemo.sym.loss.aggregator) rfft() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) rfft2() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) 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) root_log_dir (physicsnemo.active_learning.config.DriverConfig attribute), [1] rotate() (physicsnemo.sym.geometry.curve.Curve method) (physicsnemo.sym.geometry.geometry.Geometry method) round_sigma() (physicsnemo.diffusion.preconditioners.EDMPrecond static method) (physicsnemo.diffusion.preconditioners.EDMPrecondSuperResolution static method) (physicsnemo.diffusion.preconditioners.iDDPMPrecond method) (physicsnemo.diffusion.preconditioners.VEPrecond method) (physicsnemo.diffusion.preconditioners.VPPrecond method) (physicsnemo.models.diffusion_unets.CorrDiffRegressionUNet method) RpropConf (class in physicsnemo.sym.hydra.optimizer) run() (physicsnemo.active_learning.driver.Driver method) run_freq (physicsnemo.sym.hydra.loss.NTKConf attribute) run_id (physicsnemo.active_learning.config.DriverConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) run_mode (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) S sample() (physicsnemo.active_learning.protocols.QueryStrategy method) (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_warp() (physicsnemo.sym.geometry.tessellation_warp.WarpTessellationSampler method) sampling (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) save_checkpoint() (physicsnemo.active_learning.driver.Driver method) 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) save_training_checkpoint() (physicsnemo.active_learning.loop.DefaultTrainingLoop method) scale() (physicsnemo.sym.geometry.curve.Curve method) (physicsnemo.sym.geometry.geometry.Geometry method) scale_model_targets() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) scale_parameter (physicsnemo.sym.hydra.optimizer.AdafactorConf attribute) ScaleShiftMlp (class in physicsnemo.nn.module.afno_layers) 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_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) scheduler_cls (physicsnemo.active_learning.config.OptimizerConfig attribute), [1] scheduler_kwargs (physicsnemo.active_learning.config.OptimizerConfig attribute), [1] SchedulerConf (class in physicsnemo.sym.hydra.scheduler) ScreenedPoissonDistance (class in physicsnemo.sym.eq.pdes.signed_distance_function) sdf() (physicsnemo.sym.geometry.tessellation_warp.Tessellation static method) second_order_pino_grads() (in module physicsnemo.nn.module.spectral_layers) Seq2SeqRNN (class in physicsnemo.models.rnn.rnn_seq2seq) serialize_records() (physicsnemo.active_learning.protocols.MetrologyStrategy method) set_dataset() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe 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) shard_grid (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) shard_points (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) short_run_id (physicsnemo.active_learning.driver.Driver property) sigma() (physicsnemo.diffusion.preconditioners.VPPrecond method) sigma_inv() (physicsnemo.diffusion.preconditioners.VPPrecond method) signed_distance_field() (in module physicsnemo.nn.functional) silu_backward_for() (in module physicsnemo.nn.module.fused_silu) silu_double_backward_for() (in module physicsnemo.nn.module.fused_silu) silu_triple_backward_for() (in module physicsnemo.nn.module.fused_silu) SimpleFormat (class in physicsnemo.sym.hydra.hydra) SinusoidalTimestepEmbedding (class in physicsnemo.nn.module.embedding_layers) SirenArch (class in physicsnemo.sym.models.siren) SirenConf (class in physicsnemo.sym.hydra.arch) SirenLayer (class in physicsnemo.nn.module.siren_layers) SirenLayerType (class in physicsnemo.nn.module.siren_layers) 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) skip_labeling (physicsnemo.active_learning.config.DriverConfig attribute), [1] skip_metrology (physicsnemo.active_learning.config.DriverConfig attribute), [1] skip_training (physicsnemo.active_learning.config.DriverConfig attribute), [1] 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) SommerfeldBC (class in physicsnemo.sym.eq.pdes.electromagnetic) SongUNet (class in physicsnemo.models.diffusion_unets) SongUNetPosEmbd (class in physicsnemo.models.diffusion_unets) SongUNetPosLtEmbd (class in physicsnemo.models.diffusion_unets) SparseAdamConf (class in physicsnemo.sym.hydra.optimizer) SpectralConv1d (class in physicsnemo.nn.module.spectral_layers) SpectralConv2d (class in physicsnemo.nn.module.spectral_layers) SpectralConv3d (class in physicsnemo.nn.module.spectral_layers) SpectralConv4d (class in physicsnemo.nn.module.spectral_layers) Sphere (class in physicsnemo.sym.geometry.primitives_3d) SquarePlus (class in physicsnemo.nn.module.activations) 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) Stan (class in physicsnemo.nn.module.activations) start_step (physicsnemo.sym.hydra.profiler.NvtxProfiler attribute) (physicsnemo.sym.hydra.profiler.ProfilerConf attribute) (physicsnemo.sym.hydra.profiler.TensorBoardProfiler attribute) state_dict() (physicsnemo.optim.CombinedOptimizer method) StaticCaptureEvaluateNoGrad (class in physicsnemo.utils.capture) StaticCaptureTraining (class in physicsnemo.utils.capture) step() (physicsnemo.optim.CombinedOptimizer method) step_sizes (physicsnemo.sym.hydra.optimizer.RpropConf attribute) StokesDataset (class in physicsnemo.datapipes.gnn.stokes_dataset) stop_criterion (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) StopCriterionConf (class in physicsnemo.sym.hydra.training) strategies_config (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] (physicsnemo.active_learning.driver.Driver attribute) StrategiesConfig (class in physicsnemo.active_learning.config) strategy_dir (physicsnemo.active_learning.protocols.ActiveLearningProtocol property), [1] 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) 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) 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) SwinTransformer (class in physicsnemo.nn.module.transformer_layers) symbolic() (physicsnemo.nn.module.fft.OnnxIrfft static method) (physicsnemo.nn.module.fft.OnnxIrfft2 static method) (physicsnemo.nn.module.fft.OnnxRfft static method) (physicsnemo.nn.module.fft.OnnxRfft2 static method) 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) (class in physicsnemo.sym.geometry.tessellation_warp) test_function (physicsnemo.sym.hydra.training.VariationalTraining attribute) Tetrahedron (class in physicsnemo.sym.geometry.primitives_3d) TFExponentialLRConf (class in physicsnemo.sym.hydra.scheduler) TimeSeriesDataset (class in physicsnemo.datapipes.healpix.timeseries_dataset) to_absolute_path() (in module physicsnemo.sym.hydra.utils) to_dict() (physicsnemo.active_learning.config.OptimizerConfig method) (physicsnemo.active_learning.config.StrategiesConfig method) (physicsnemo.active_learning.config.TrainingConfig method) to_json() (physicsnemo.active_learning.config.DriverConfig method) to_yaml() (in module physicsnemo.sym.hydra.utils) 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) train_datapool (physicsnemo.active_learning.config.TrainingConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) train_loop_fn (physicsnemo.active_learning.config.TrainingConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) Trainer (class in physicsnemo.sym.trainer) TRAINING (physicsnemo.active_learning.protocols.ActiveLearningPhase attribute) training (physicsnemo.sym.hydra.config.PhysicsNeMoConfig attribute) training_config (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] (physicsnemo.active_learning.driver.Driver attribute) training_pool (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] training_step() (physicsnemo.active_learning.protocols.LearnerProtocol method) TrainingConf (class in physicsnemo.sym.hydra.training) TrainingConfig (class in physicsnemo.active_learning.config) TrainingLoop (class in physicsnemo.active_learning.protocols) TrainingProtocol (class in physicsnemo.active_learning.protocols) 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) Transformer2DBlock (class in physicsnemo.nn.module.transformer_layers) Transformer3DBlock (class in physicsnemo.nn.module.transformer_layers) TransformerDecoder (class in physicsnemo.nn.module.transformer_decoder) translate() (physicsnemo.sym.geometry.curve.Curve method) (physicsnemo.sym.geometry.geometry.Geometry method) TransposeConvLayer (class in physicsnemo.nn.module.conv_layers) 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) U UNetAttention (class in physicsnemo.nn.module.attention_layers) UNetBlock (class in physicsnemo.nn.module.unet_layers) unique_axis0_fast() (in module physicsnemo.sym.eq.spatial_grads.spatial_grads) unlabeled_datapool (physicsnemo.active_learning.config.StrategiesConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) unlabeled_pool (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] unmark_module_as_shared() (in module physicsnemo.distributed.utils) unscale_model_outputs() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) 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) UpSample2D (class in physicsnemo.nn.module.resample_layers) UpSample3D (class in physicsnemo.nn.module.resample_layers) use_fp16 (physicsnemo.diffusion.preconditioners.EDMPrecondSuperResolution property) (physicsnemo.models.diffusion_unets.CorrDiffRegressionUNet property) use_ntk (physicsnemo.sym.hydra.loss.NTKConf attribute) use_quadratures (physicsnemo.sym.hydra.training.VariationalTraining attribute) V val_datapool (physicsnemo.active_learning.config.TrainingConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) validation_pool (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] validation_step() (physicsnemo.active_learning.protocols.LearnerProtocol method) ValidationProtocol (class in physicsnemo.active_learning.protocols) 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.diffusion.preconditioners) version (physicsnemo.sym.hydra.hydra.DebugLogging attribute) (physicsnemo.sym.hydra.hydra.DefaultLogging attribute) view_as_complex() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) 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_sample_from_disk (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) volume_variables (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) VortexSheddingDataset (class in physicsnemo.datapipes.gnn.vortex_shedding_dataset) VoxelInferencer (class in physicsnemo.sym.domain.inferencer.voxel) VPPrecond (class in physicsnemo.diffusion.preconditioners) 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) WarpTessellationSampler (class in physicsnemo.sym.geometry.tessellation_warp) 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_fact() (in module physicsnemo.nn.functional) 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) WeightFactLinear (class in physicsnemo.nn.module.weight_fact) WeightNormLinear (class in physicsnemo.nn.module.weight_norm) weights (physicsnemo.sym.hydra.loss.LossConf attribute) world_size (physicsnemo.distributed.manager.DistributedManager property) X xi (physicsnemo.sym.hydra.optimizer.AccSGDConf attribute) Y YogiConf (class in physicsnemo.sym.hydra.optimizer) Z zero_grad() (physicsnemo.optim.CombinedOptimizer method) 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)
