Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | 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] _local_tensor (physicsnemo.domain_parallel.ShardTensor attribute) _registry (in module physicsnemo.active_learning) _setup_logger() (physicsnemo.active_learning.protocols.ActiveLearningProtocol method), [1] _spec (physicsnemo.domain_parallel.ShardTensor attribute) A AbstractQueue (class in physicsnemo.active_learning.protocols) acc() (in module physicsnemo.metrics.climate.acc) activation() (physicsnemo.models.dlwp.dlwp.DLWP method) 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] 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_noise() (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.NoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) add_param_group() (physicsnemo.optim.CombinedOptimizer method) AFNO (class in physicsnemo.models.afno.afno) AFNO2DLayer (class in physicsnemo.models.afno.afno) (class in physicsnemo.nn.module.afno_layers) AFNOMlp (class in physicsnemo.models.afno.afno) (class in physicsnemo.nn.module.afno_layers) AFNOPatchEmbed (class in physicsnemo.nn.module.afno_layers) 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.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) alpha_dot() (physicsnemo.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) amp_mode (physicsnemo.models.diffusion_unets.DhariwalUNet property) (physicsnemo.models.diffusion_unets.SongUNet property) amp_type (physicsnemo.active_learning.loop.DefaultTrainingLoop property) append() (physicsnemo.active_learning.protocols.DataPool method), [1] (physicsnemo.active_learning.protocols.MetrologyStrategy method) (physicsnemo.datapipes.transforms.compose.Compose method) apply_activation() (physicsnemo.nn.module.fully_connected_layers.ConvFCLayer method) 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) azimuthal_angle() (in module physicsnemo.models.graphcast.utils.graph_utils) B backward() (physicsnemo.distributed.autograd.AllGatherVAutograd static method) (physicsnemo.distributed.autograd.GatherVAutograd static method) (physicsnemo.distributed.autograd.IndexedAllToAllVAutograd static method) (physicsnemo.distributed.autograd.ScatterVAutograd static method) (physicsnemo.distributed.fft.DistributedIRFFT2 static method) (physicsnemo.distributed.fft.DistributedRFFT2 static method) (physicsnemo.domain_parallel.ShardTensor method) (physicsnemo.nn.module.attention_layers.AttentionOp static method) barrier() (physicsnemo.active_learning.driver.Driver method) BaseAffinePreconditioner (class in physicsnemo.diffusion.preconditioners) batch_size (physicsnemo.active_learning.config.DriverConfig attribute), [1] BiStrideMeshGraphNet (class in physicsnemo.models.meshgraphnet.bsms_mgn) 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) BoundingBoxFilter (class in physicsnemo.datapipes.transforms.spatial) BQWarp (class in physicsnemo.nn.module.ball_query) broadcast_buffers (physicsnemo.distributed.manager.DistributedManager property) broadcast_global_features (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) BroadcastGlobalFeatures (class in physicsnemo.datapipes.transforms.field_processing) 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) calculate_fid_from_inception_stats() (in module physicsnemo.diffusion.metrics.fid) cancel_prefetch() (physicsnemo.datapipes.multi_dataset.MultiDataset method) CappedGELU (class in physicsnemo.nn.module.activations) CappedLeakyReLU (class in physicsnemo.nn.module.activations) cdf() (in module physicsnemo.metrics.general.histogram) cell_areas (physicsnemo.mesh.mesh.Mesh property) cell_centroids (physicsnemo.mesh.mesh.Mesh property) cell_data_to_point_data() (physicsnemo.mesh.mesh.Mesh method) cell_normals (physicsnemo.mesh.mesh.Mesh property) cell_to_adj() (in module physicsnemo.models.graphcast.utils.graph_utils) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) CenterOfMass (class in physicsnemo.datapipes.transforms.spatial) 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] clean() (physicsnemo.mesh.mesh.Mesh 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) close() (physicsnemo.datapipes.dataset.Dataset method) (physicsnemo.datapipes.multi_dataset.MultiDataset method) (physicsnemo.datapipes.readers.base.Reader method) (physicsnemo.datapipes.readers.hdf5.HDF5Reader method) (physicsnemo.datapipes.readers.numpy.NumpyReader method) (physicsnemo.datapipes.readers.zarr.ZarrReader method) codimension (physicsnemo.mesh.mesh.Mesh property) collate_fn (physicsnemo.active_learning.config.DriverConfig attribute), [1] Collator (class in physicsnemo.datapipes.collate) 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 (class in physicsnemo.datapipes.transforms.compose) compute() (physicsnemo.active_learning.protocols.MetrologyStrategy method) compute_cell_derivatives() (physicsnemo.mesh.mesh.Mesh method) compute_coefficients() (physicsnemo.diffusion.preconditioners.BaseAffinePreconditioner method) (physicsnemo.diffusion.preconditioners.EDMPreconditioner method) (physicsnemo.diffusion.preconditioners.IDDPMPreconditioner method) (physicsnemo.diffusion.preconditioners.VEPreconditioner method) (physicsnemo.diffusion.preconditioners.VPPreconditioner method) compute_connectivity_tensor() (in module physicsnemo.sym.eq.gradients) compute_gradients() (physicsnemo.sym.eq.gradients.GradientCalculator method) compute_point_derivatives() (physicsnemo.mesh.mesh.Mesh method) compute_point_normals() (physicsnemo.mesh.mesh.Mesh method) compute_scaling_factors (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) compute_scaling_factors() (in module physicsnemo.datapipes.cae.domino_datapipe) 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) ComputeNormals (class in physicsnemo.datapipes.transforms.geometric) ComputeSDF (class in physicsnemo.datapipes.transforms.geometric) ConcatCollator (class in physicsnemo.datapipes.collate) ConcatFields (class in physicsnemo.datapipes.transforms.concat_fields) config (physicsnemo.active_learning.driver.Driver attribute) configure_optimizer() (physicsnemo.active_learning.driver.Driver method) ConstantField (class in physicsnemo.datapipes.transforms.utility) 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) ConvBlock (class in physicsnemo.nn.module.conv_layers) ConvFCLayer (class in physicsnemo.nn.module.fully_connected_layers) 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) copy_to_parallel_region() (in module physicsnemo.distributed.mappings) create_ddp_group_tag() (in module physicsnemo.utils.logging.utils) create_g2m_graph() (physicsnemo.models.graphcast.utils.graph.Graph method) create_graph() (physicsnemo.datapipes.gnn.ahmed_body_dataset.AhmedBodyDataset method) (physicsnemo.datapipes.gnn.vortex_shedding_dataset.VortexSheddingDataset static method) create_m2g_graph() (physicsnemo.models.graphcast.utils.graph.Graph method) create_mesh_graph() (physicsnemo.models.graphcast.utils.graph.Graph method) create_orthogonal_process_group() (physicsnemo.distributed.manager.DistributedManager static method) create_process_subgroup() (physicsnemo.distributed.manager.DistributedManager static method) CreateGrid (class in physicsnemo.datapipes.transforms.spatial) crps() (in module physicsnemo.metrics.general.crps) CubeEmbedding (class in physicsnemo.nn.module.conv_layers) cuda (physicsnemo.distributed.manager.DistributedManager property) current_phase (physicsnemo.active_learning.driver.Driver attribute) custom_forward() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) D 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) (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) DataConsistencyDPSGuidance (class in physicsnemo.diffusion.guidance) DataLoader (class in physicsnemo.datapipes.dataloader) DataPool (class in physicsnemo.active_learning.protocols) Dataset (class in physicsnemo.datapipes.dataset) decode() (physicsnemo.models.mesh_reduced.mesh_reduced.Mesh_Reduced method) decoder_forward() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) DecoderLayer (class in physicsnemo.nn.module.transformer_layers) DecoderOnlyLayer (class in physicsnemo.nn.module.transformer_decoder) DeepSpeed DefaultCollator (class in physicsnemo.datapipes.collate) DefaultTrainingLoop (class in physicsnemo.active_learning.loop) deg2rad() (in module physicsnemo.models.graphcast.utils.graph_utils) Denoiser (class in physicsnemo.diffusion) 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) deterministic (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) device (physicsnemo.active_learning.config.DriverConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) (physicsnemo.core.module.Module property) (physicsnemo.datapipes.transforms.base.Transform property) (physicsnemo.distributed.manager.DistributedManager property) (physicsnemo.mesh.mesh.Mesh property) device_mesh (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler property) DeviceMesh DGMLayer (class in physicsnemo.nn.module.dgm_layers) DhariwalUNet (class in physicsnemo.models.diffusion_unets) diffusion() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) DiffusionModel (class in physicsnemo.diffusion) dimensions_compatible() (physicsnemo.datapipes.climate.climate.ClimateDataSourceSpec method) disable_prefetch() (physicsnemo.datapipes.dataloader.DataLoader method) 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) DistributedDataParallel DistributedIRFFT2 (class in physicsnemo.distributed.fft) DistributedManager (class in physicsnemo.distributed.manager) DistributedRFFT2 (class in physicsnemo.distributed.fft) DiT (class in physicsnemo.models.dit) DLWP (class in physicsnemo.models.dlwp.dlwp) Domain Parallel DomainParallelNoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) 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) DPSGuidance (class in physicsnemo.diffusion.guidance) DPSScorePredictor (class in physicsnemo.diffusion.guidance) draw() (physicsnemo.mesh.mesh.Mesh method) draw_mesh() (in module physicsnemo.mesh.visualization.draw_mesh) drift() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler 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] dumps() (physicsnemo.mesh.mesh.Mesh method) E EarthAttention2D (class in physicsnemo.nn.module.attention_layers) EarthAttention3D (class in physicsnemo.nn.module.attention_layers) EDMLogUniformNoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) EDMNoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) EDMPreconditioner (class in physicsnemo.diffusion.preconditioners) EDMStochasticEulerSolver (class in physicsnemo.diffusion.samplers.solvers) EDMStochasticHeunSolver (class in physicsnemo.diffusion.samplers.solvers) efi() (in module physicsnemo.metrics.climate.efi) efi_gaussian() (in module physicsnemo.metrics.climate.efi) empty() (physicsnemo.active_learning.protocols.AbstractQueue method) enable_prefetch() (physicsnemo.datapipes.dataloader.DataLoader method) encode() (physicsnemo.models.mesh_reduced.mesh_reduced.Mesh_Reduced method) encoder_forward() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) EncoderLayer (class in physicsnemo.nn.module.transformer_layers) EnsembleMetrics (class in physicsnemo.metrics.general.ensemble_metrics) entropy_from_counts() (in module physicsnemo.metrics.general.entropy) epsilon_to_score() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) epsilon_to_x0() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) ERA5DaliExternalSource (class in physicsnemo.datapipes.climate.era5_hdf5) ERA5HDF5Datapipe (class in physicsnemo.datapipes.climate.era5_hdf5) error() (physicsnemo.utils.logging.console.PythonLogger method) EulerSolver (class in physicsnemo.diffusion.samplers.solvers) extra_repr() (physicsnemo.datapipes.transforms.base.Transform method) (physicsnemo.datapipes.transforms.compose.Compose method) (physicsnemo.datapipes.transforms.concat_fields.ConcatFields method) (physicsnemo.datapipes.transforms.concat_fields.NormalizeVectors method) (physicsnemo.datapipes.transforms.field_slice.FieldSlice method) (physicsnemo.datapipes.transforms.normalize.Normalize method) (physicsnemo.datapipes.transforms.utility.ConstantField method) (physicsnemo.datapipes.transforms.utility.Purge method) (physicsnemo.datapipes.transforms.utility.Rename method) (physicsnemo.nn.module.drop.DropPath method) (physicsnemo.nn.module.weight_fact.WeightFactLinear method) (physicsnemo.nn.module.weight_norm.WeightNormLinear method) F factory() (physicsnemo.core.registry.ModelRegistry method) FCLayer (class in physicsnemo.nn.module.fully_connected_layers) Fengwu (class in physicsnemo.models.fengwu.fengwu) field_names (physicsnemo.datapipes.dataset.Dataset property) (physicsnemo.datapipes.multi_dataset.MultiDataset property) (physicsnemo.datapipes.readers.base.Reader property) fields (physicsnemo.datapipes.readers.numpy.NumpyReader property) (physicsnemo.datapipes.readers.tensorstore_zarr.TensorStoreZarrReader property) (physicsnemo.datapipes.readers.zarr.ZarrReader property) fields() (physicsnemo.mesh.mesh.Mesh class method) FieldSlice (class in physicsnemo.datapipes.transforms.field_slice) file_logging() (physicsnemo.utils.logging.console.PythonLogger method) FileInfo (class in physicsnemo.datapipes.gnn.ahmed_body_dataset) 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) fine_tuning_lr (physicsnemo.active_learning.config.DriverConfig attribute), [1] first_order_pino_grads() (in module physicsnemo.nn.module.spectral_layers) 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) 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.graphcast.graph_cast_net.GraphCastNet 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.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.gradients.GradientsAutoDiff method) (physicsnemo.sym.eq.gradients.GradientsFiniteDifference method) (physicsnemo.sym.eq.gradients.GradientsLeastSquares method) (physicsnemo.sym.eq.gradients.GradientsMeshlessFiniteDifference method) (physicsnemo.sym.eq.gradients.GradientsSpectral method) fourier_derivatives() (in module physicsnemo.nn.module.spectral_layers) fourier_encode() (in module physicsnemo.nn.module.fourier_layers) 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) from_checkpoint() (physicsnemo.core.module.Module class method) from_dict() (physicsnemo.active_learning.config.OptimizerConfig class method) (physicsnemo.active_learning.config.StrategiesConfig class method) (physicsnemo.active_learning.config.TrainingConfig class method) from_dtensor() (physicsnemo.domain_parallel.ShardTensor class method) from_json() (physicsnemo.active_learning.config.DriverConfig class method) from_local() (physicsnemo.domain_parallel.ShardTensor static method) from_tensordict() (physicsnemo.mesh.mesh.Mesh class method) from_torch() (physicsnemo.core.module.Module static method) FSDP full_tensor() (physicsnemo.domain_parallel.ShardTensor method) FullyConnected (class in physicsnemo.models.mlp.fully_connected) FunctionCollator (class in physicsnemo.datapipes.collate) FusedSiLU (class in physicsnemo.nn.module.fused_silu) FuserLayer (class in physicsnemo.nn.module.transformer_layers) G GaborFilter (class in physicsnemo.nn.module.fourier_layers) 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) gaussian_curvature_cells (physicsnemo.mesh.mesh.Mesh property) gaussian_curvature_vertices (physicsnemo.mesh.mesh.Mesh property) 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) geospatial_rotation() (in module physicsnemo.models.graphcast.utils.graph_utils) get() (physicsnemo.active_learning.protocols.AbstractQueue method) (physicsnemo.core.filesystem.Package method) (physicsnemo.mesh.mesh.Mesh 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_boundary_mesh() (physicsnemo.mesh.mesh.Mesh method) get_cell_to_cells_adjacency() (physicsnemo.mesh.mesh.Mesh method) get_cell_to_points_adjacency() (physicsnemo.mesh.mesh.Mesh method) get_checkpoint_dir() (in module physicsnemo.utils.checkpoint) get_constants() (physicsnemo.datapipes.healpix.timeseries_dataset.TimeSeriesDataset method) get_denoiser() (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.NoiseScheduler method) get_face_centroids() (in module physicsnemo.models.graphcast.utils.graph_utils) get_facet_mesh() (physicsnemo.mesh.mesh.Mesh method) get_gradient_module() (physicsnemo.sym.eq.gradients.GradientCalculator method) get_grid() (physicsnemo.models.transolver.transolver.Transolver method) get_group_norm() (in module physicsnemo.nn.module.group_norm) get_lat_lon_partition_separators() (in module physicsnemo.models.graphcast.graph_cast_net) 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) get_point_to_cells_adjacency() (physicsnemo.mesh.mesh.Mesh method) get_point_to_points_adjacency() (physicsnemo.mesh.mesh.Mesh 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) GradientCalculator (class in physicsnemo.sym.eq.gradients) GradientsAutoDiff (class in physicsnemo.sym.eq.gradients) GradientsFiniteDifference (class in physicsnemo.sym.eq.gradients) GradientsLeastSquares (class in physicsnemo.sym.eq.gradients) GradientsMeshlessFiniteDifference (class in physicsnemo.sym.eq.gradients) GradientsSpectral (class in physicsnemo.sym.eq.gradients) Graph (class in physicsnemo.models.graphcast.utils.graph) GraphCastNet (class in physicsnemo.models.graphcast.graph_cast_net) grid_resolution (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) grid_to_point_interpolation() (in module physicsnemo.nn.functional) 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) 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 has_label_queue (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] has_query_queue (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] HDF5Reader (class in physicsnemo.datapipes.readers.hdf5) HEALPixRecUNet (class in physicsnemo.models.dlwp_healpix.HEALPixRecUNet) HeunSolver (class in physicsnemo.diffusion.samplers.solvers) Histogram (class in physicsnemo.metrics.general.histogram) histogram() (in module physicsnemo.metrics.general.histogram) I IDDPMNoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) IDDPMPreconditioner (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) include_normals (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) include_sdf (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) 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) inference_fn (physicsnemo.active_learning.driver.Driver attribute) inference_step() (physicsnemo.active_learning.protocols.LearnerProtocol method) InferenceProtocol (class in physicsnemo.active_learning.protocols) InfiniteSampler (class in physicsnemo.diffusion.utils) info() (physicsnemo.utils.logging.console.PythonLogger method) init_latents() (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.NoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) initialise_parameters() (physicsnemo.nn.module.fully_connected_layers.ConvNdFCLayer method) initialize() (physicsnemo.distributed.manager.DistributedManager static method) (physicsnemo.utils.logging.launch.LaunchLogger static method) initialize_batch() (physicsnemo.datapipes.benchmarks.darcy.Darcy2D method) (physicsnemo.datapipes.benchmarks.kelvin_helmholtz.KelvinHelmholtz2D method) initialize_env() (physicsnemo.distributed.manager.DistributedManager static method) initialize_mesh() (physicsnemo.distributed.manager.DistributedManager method) initialize_open_mpi() (physicsnemo.distributed.manager.DistributedManager static method) initialize_slurm() (physicsnemo.distributed.manager.DistributedManager static method) initialize_weights() (physicsnemo.models.dit.DiT method) (physicsnemo.models.transolver.transolver.Transolver method) inner_scheduler (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler property) insolation() (in module physicsnemo.utils.insolation) instantiate() (physicsnemo.core.module.Module class method) integrate() (physicsnemo.mesh.mesh.Mesh method) integrate_flux() (physicsnemo.mesh.mesh.Mesh method) integration_steps (physicsnemo.models.dlwp_healpix.HEALPixRecUNet.HEALPixRecUNet property) Inter-Node interpolation() (in module physicsnemo.nn.functional) Intra-Node inverse() (physicsnemo.datapipes.transforms.normalize.Normalize method) 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_manifold() (physicsnemo.mesh.mesh.Mesh method) is_optimizer_configured (physicsnemo.active_learning.driver.Driver property) is_registered() (in module physicsnemo.active_learning) is_watertight() (physicsnemo.mesh.mesh.Mesh method) K kcrps() (in module physicsnemo.metrics.general.crps) KelvinHelmholtz2D (class in physicsnemo.datapipes.benchmarks.kelvin_helmholtz) KNearestNeighbors (class in physicsnemo.datapipes.transforms.spatial) knn() (in module physicsnemo.nn.functional) knn_interpolate() (physicsnemo.models.mesh_reduced.mesh_reduced.Mesh_Reduced method) 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) last_checkpoint (physicsnemo.active_learning.driver.Driver property) latlon2xyz() (in module physicsnemo.models.graphcast.utils.graph_utils) LaunchLogger (class in physicsnemo.utils.logging.launch) 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) Linear (class in physicsnemo.nn.module.fully_connected_layers) LinearGaussianNoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) list_models() (physicsnemo.core.registry.ModelRegistry method) load() (physicsnemo.core.module.Module method) (physicsnemo.mesh.mesh.Mesh class method) load_() (physicsnemo.mesh.mesh.Mesh method) load_checkpoint() (in module physicsnemo.utils.checkpoint) (physicsnemo.active_learning.driver.Driver class method) load_json() (in module physicsnemo.datapipes.gnn.utils) load_memmap() (physicsnemo.mesh.mesh.Mesh class method) load_model_weights() (in module physicsnemo.utils.checkpoint) load_records() (physicsnemo.active_learning.protocols.MetrologyStrategy method) load_state_dict() (physicsnemo.datapipes.transforms.base.Transform method) (physicsnemo.datapipes.transforms.normalize.Normalize method) (physicsnemo.mesh.mesh.Mesh method) (physicsnemo.optim.CombinedOptimizer method) load_statistics() (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) log() (physicsnemo.utils.logging.console.PythonLogger method) log_dir (physicsnemo.active_learning.driver.Driver property) log_epoch() (physicsnemo.utils.logging.launch.LaunchLogger method) log_figure() (physicsnemo.utils.logging.launch.LaunchLogger method) log_minibatch() (physicsnemo.utils.logging.launch.LaunchLogger method) logger (physicsnemo.active_learning.driver.Driver attribute) (physicsnemo.active_learning.protocols.ActiveLearningProtocol property), [1] loss_weight() (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.NoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) lr_scheduler (physicsnemo.active_learning.driver.Driver attribute) lr_scheduler_state (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] M make_computations() (physicsnemo.sym.eq.pde.PDE method) mark_module_as_shared() (in module physicsnemo.distributed.utils) max_active_learning_steps (physicsnemo.active_learning.config.DriverConfig attribute), [1] max_edge_length() (in module physicsnemo.models.graphcast.utils.graph_utils) max_fine_tuning_epochs (physicsnemo.active_learning.config.TrainingConfig attribute), [1] max_samples (physicsnemo.active_learning.protocols.QueryStrategy attribute), [1] max_training_epochs (physicsnemo.active_learning.config.TrainingConfig attribute), [1] Mean (class in physicsnemo.metrics.general.ensemble_metrics) mean (physicsnemo.metrics.general.ensemble_metrics.Variance property) mean_curvature_vertices (physicsnemo.mesh.mesh.Mesh property) memmap() (physicsnemo.mesh.mesh.Mesh method) memmap_() (physicsnemo.mesh.mesh.Mesh method) memmap_like() (physicsnemo.mesh.mesh.Mesh method) memmap_refresh_() (physicsnemo.mesh.mesh.Mesh method) merge() (physicsnemo.mesh.mesh.Mesh class method) Mesh (class in physicsnemo.mesh.mesh) mesh() (physicsnemo.distributed.manager.DistributedManager method) mesh_dims (physicsnemo.distributed.manager.DistributedManager property) mesh_green_gauss_gradient() (in module physicsnemo.nn.functional) mesh_lsq_gradient() (in module physicsnemo.nn.functional) mesh_names() (physicsnemo.distributed.manager.DistributedManager method) mesh_poisson_disk_sample() (in module physicsnemo.nn.functional) Mesh_Reduced (class in physicsnemo.models.mesh_reduced.mesh_reduced) mesh_sizes() (physicsnemo.distributed.manager.DistributedManager method) mesh_to_voxel_fraction() (in module physicsnemo.nn.functional) MeshGraphNet (class in physicsnemo.models.meshgraphnet.meshgraphnet) MeshGraphNetProcessor (class in physicsnemo.models.meshgraphnet.meshgraphnet) meshless_fd_derivatives() (in module physicsnemo.nn.functional) MetaData (class in physicsnemo.datapipes.benchmarks.darcy) (class in physicsnemo.datapipes.benchmarks.kelvin_helmholtz) (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) 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) 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) Model Parallel model_checkpoint_frequency (physicsnemo.active_learning.config.DriverConfig attribute), [1] model_type (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) ModelConsistencyDPSGuidance (class in physicsnemo.diffusion.guidance) ModelRegistry (class in physicsnemo.core.registry) module physicsnemo.active_learning.config physicsnemo.active_learning.driver physicsnemo.active_learning.loop physicsnemo.active_learning.protocols physicsnemo.core physicsnemo.core.filesystem physicsnemo.datapipes physicsnemo.datapipes.benchmarks.darcy physicsnemo.datapipes.benchmarks.kelvin_helmholtz physicsnemo.datapipes.cae.domino_datapipe physicsnemo.datapipes.cae.transolver_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.distributed physicsnemo.distributed.autograd physicsnemo.distributed.fft physicsnemo.distributed.manager physicsnemo.distributed.mappings physicsnemo.distributed.utils physicsnemo.mesh.boundaries physicsnemo.mesh.calculus physicsnemo.mesh.curvature physicsnemo.mesh.geometry physicsnemo.mesh.io physicsnemo.mesh.neighbors physicsnemo.mesh.projections physicsnemo.mesh.remeshing physicsnemo.mesh.repair physicsnemo.mesh.sampling physicsnemo.mesh.smoothing physicsnemo.mesh.spatial physicsnemo.mesh.subdivision physicsnemo.mesh.transformations.geometric physicsnemo.mesh.validation physicsnemo.mesh.visualization.draw_mesh physicsnemo.metrics physicsnemo.metrics.climate.acc physicsnemo.metrics.climate.efi physicsnemo.metrics.climate.reduction physicsnemo.metrics.general.calibration physicsnemo.metrics.general.crps physicsnemo.metrics.general.ensemble_metrics physicsnemo.metrics.general.entropy physicsnemo.metrics.general.histogram physicsnemo.metrics.general.mse physicsnemo.metrics.general.reduction physicsnemo.metrics.general.wasserstein physicsnemo.models physicsnemo.models.domino.utils physicsnemo.models.graphcast.utils.graph physicsnemo.models.graphcast.utils.graph_utils 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.utils physicsnemo.utils.capture physicsnemo.utils.checkpoint physicsnemo.utils.insolation physicsnemo.utils.logging physicsnemo.utils.logging.console physicsnemo.utils.logging.launch physicsnemo.utils.logging.mlflow physicsnemo.utils.logging.utils physicsnemo.utils.logging.wandb physicsnemo.utils.profiling physicsnemo.utils.zenith_angle Module (class in physicsnemo.core.module) mse() (in module physicsnemo.metrics.general.mse) MSEDSMLoss (class in physicsnemo.diffusion.metrics.losses) MultiDataset (class in physicsnemo.datapipes.multi_dataset) N Node NoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) normal_cdf() (in module physicsnemo.metrics.general.histogram) normal_pdf() (in module physicsnemo.metrics.general.histogram) Normalize (class in physicsnemo.datapipes.transforms.normalize) 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) NormalizeVectors (class in physicsnemo.datapipes.transforms.concat_fields) num_dataloader_workers (physicsnemo.active_learning.config.DriverConfig attribute), [1] num_parameters() (physicsnemo.core.module.Module method) num_surface_neighbors (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) NumpyReader (class in physicsnemo.datapipes.readers.numpy) O offsets() (physicsnemo.domain_parallel.ShardTensor method) 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) 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] OptimizerConfig (class in physicsnemo.active_learning.config) P Package (class in physicsnemo.core.filesystem) pad() (physicsnemo.mesh.mesh.Mesh method) pad_helper() (in module physicsnemo.distributed.utils) pad_to_next_power() (physicsnemo.mesh.mesh.Mesh method) Pangu (class in physicsnemo.models.pangu.pangu) parameters (physicsnemo.active_learning.protocols.LearnerProtocol property) parse_dataset_files() (physicsnemo.datapipes.climate.climate.ClimateDataSourceSpec method) (physicsnemo.datapipes.climate.era5_hdf5.ERA5HDF5Datapipe method) PatchEmbed (in module physicsnemo.models.afno.afno) (in module physicsnemo.nn.module.afno_layers) patches_enabled() (physicsnemo.domain_parallel.ShardTensor class method) PDE (class in physicsnemo.sym.eq.pde) 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.core module physicsnemo.core.filesystem module physicsnemo.datapipes module physicsnemo.datapipes.benchmarks.darcy module physicsnemo.datapipes.benchmarks.kelvin_helmholtz module physicsnemo.datapipes.cae.domino_datapipe module physicsnemo.datapipes.cae.transolver_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.distributed module physicsnemo.distributed.autograd module physicsnemo.distributed.fft module physicsnemo.distributed.manager module physicsnemo.distributed.mappings module physicsnemo.distributed.utils module physicsnemo.mesh.boundaries module physicsnemo.mesh.calculus module physicsnemo.mesh.curvature module physicsnemo.mesh.geometry module physicsnemo.mesh.io module physicsnemo.mesh.neighbors module physicsnemo.mesh.projections module physicsnemo.mesh.remeshing module physicsnemo.mesh.repair module physicsnemo.mesh.sampling module physicsnemo.mesh.smoothing module physicsnemo.mesh.spatial module physicsnemo.mesh.subdivision module physicsnemo.mesh.transformations.geometric module physicsnemo.mesh.validation module physicsnemo.mesh.visualization.draw_mesh module physicsnemo.metrics module physicsnemo.metrics.climate.acc module physicsnemo.metrics.climate.efi module physicsnemo.metrics.climate.reduction module physicsnemo.metrics.general.calibration module physicsnemo.metrics.general.crps module physicsnemo.metrics.general.ensemble_metrics module physicsnemo.metrics.general.entropy module physicsnemo.metrics.general.histogram module physicsnemo.metrics.general.mse module physicsnemo.metrics.general.reduction module physicsnemo.metrics.general.wasserstein module physicsnemo.models module physicsnemo.models.domino.utils module physicsnemo.models.graphcast.utils.graph module physicsnemo.models.graphcast.utils.graph_utils 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.utils module physicsnemo.utils.capture module physicsnemo.utils.checkpoint module physicsnemo.utils.insolation module physicsnemo.utils.logging module physicsnemo.utils.logging.console module physicsnemo.utils.logging.launch module physicsnemo.utils.logging.mlflow module physicsnemo.utils.logging.utils module physicsnemo.utils.logging.wandb module physicsnemo.utils.profiling module physicsnemo.utils.zenith_angle module physicsnemo_version (physicsnemo.active_learning.driver.ActiveLearningCheckpoint attribute), [1] PhysicsNeMoUndefinedGroupError PhysicsNeMoUninitializedDistributedManagerWarning Pipeline Parallel Pix2Pix (class in physicsnemo.models.pix2pix.pix2pix) Pix2PixUnet (class in physicsnemo.models.pix2pix.pix2pixunet) PixelShuffle3d (class in physicsnemo.models.srrn.super_res_net) point_data_to_cell_data() (physicsnemo.mesh.mesh.Mesh method) point_normals (physicsnemo.mesh.mesh.Mesh property) point_to_grid_interpolation() (in module physicsnemo.nn.functional) 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) poisson_sample_indices_fixed() (in module physicsnemo.datapipes.cae.transolver_datapipe) polar_angle() (in module physicsnemo.models.graphcast.utils.graph_utils) 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) pprint() (physicsnemo.sym.eq.pde.PDE method) Predictor (class in physicsnemo.diffusion) prefetch() (physicsnemo.datapipes.dataset.Dataset method) (physicsnemo.datapipes.multi_dataset.MultiDataset method) prepare_input() (physicsnemo.models.fengwu.fengwu.Fengwu method) (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) (physicsnemo.models.pangu.pangu.Pangu method) prepare_output() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) process_data() (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataPipe method) process_geometry() (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataPipe method) process_volume() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) profile_mode (physicsnemo.models.diffusion_unets.DhariwalUNet property) (physicsnemo.models.diffusion_unets.SongUNet property) 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) Purge (class in physicsnemo.datapipes.transforms.utility) put() (physicsnemo.active_learning.protocols.AbstractQueue method) PythonLogger (class in physicsnemo.utils.logging.console) Q quality_metrics (physicsnemo.mesh.mesh.Mesh property) 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 rad2deg() (in module physicsnemo.models.graphcast.utils.graph_utils) radius_search() (in module physicsnemo.nn.functional) randint() (physicsnemo.diffusion.utils.StackedRandomGenerator method) randn() (physicsnemo.diffusion.utils.StackedRandomGenerator method) randn_like() (physicsnemo.diffusion.utils.StackedRandomGenerator method) randt() (physicsnemo.diffusion.utils.StackedRandomGenerator method) Rank rank (physicsnemo.distributed.manager.DistributedManager property) rank_probability_score() (in module physicsnemo.metrics.general.calibration) RankZeroLoggingWrapper (class in physicsnemo.utils.logging.console) read_vtp_file() (in module physicsnemo.datapipes.gnn.utils) Reader (class in physicsnemo.datapipes.readers.base) real() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) records (physicsnemo.active_learning.protocols.MetrologyStrategy attribute), [1] rectilinear_grid_gradient() (in module physicsnemo.nn.functional) redistribute() (physicsnemo.domain_parallel.ShardTensor method) reduce_from_parallel_region() (in module physicsnemo.distributed.mappings) reduce_loss() (in module physicsnemo.distributed.utils) reference_origin (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) register() (in module physicsnemo.active_learning) (physicsnemo.core.registry.ModelRegistry method) register_dispatch_handler() (physicsnemo.domain_parallel.ShardTensor class method) register_function_handler() (physicsnemo.domain_parallel.ShardTensor class method) register_named_function_handler() (physicsnemo.domain_parallel.ShardTensor class method) registry (in module physicsnemo.active_learning) relative_entropy_from_counts() (in module physicsnemo.metrics.general.entropy) remove_extra_state_hook_for_torch() (in module physicsnemo.nn.module.layer_norm) Rename (class in physicsnemo.datapipes.transforms.utility) 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) resolution (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) return_mesh_features (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) rfft() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) rfft2() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) rmse() (in module physicsnemo.metrics.general.mse) root_log_dir (physicsnemo.active_learning.config.DriverConfig attribute), [1] rotate() (in module physicsnemo.mesh.transformations.geometric) (physicsnemo.mesh.mesh.Mesh method) rotation_matrix() (in module physicsnemo.mesh.transformations.geometric) run() (physicsnemo.active_learning.driver.Driver method) run_id (physicsnemo.active_learning.config.DriverConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) S sample() (in module physicsnemo.diffusion.samplers) (physicsnemo.active_learning.protocols.QueryStrategy method) sample_data_at_points() (physicsnemo.mesh.mesh.Mesh method) sample_in_bbox (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) sample_random_points_on_cells() (physicsnemo.mesh.mesh.Mesh method) sample_time() (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.EDMLogUniformNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.NoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) sampling (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) save() (physicsnemo.core.module.Module method) (physicsnemo.mesh.mesh.Mesh method) save_checkpoint() (in module physicsnemo.utils.checkpoint) (physicsnemo.active_learning.driver.Driver method) save_json() (in module physicsnemo.datapipes.gnn.utils) save_training_checkpoint() (physicsnemo.active_learning.loop.DefaultTrainingLoop method) Scale (class in physicsnemo.datapipes.transforms.geometric) scale() (in module physicsnemo.mesh.transformations.geometric) (physicsnemo.mesh.mesh.Mesh method) scale_matrix() (in module physicsnemo.mesh.transformations.geometric) scale_model_targets() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataPipe method) ScaleShiftMlp (class in physicsnemo.nn.module.afno_layers) scaling_type (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) scatter_tensor() (in module physicsnemo.domain_parallel) 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_cls (physicsnemo.active_learning.config.OptimizerConfig attribute), [1] scheduler_kwargs (physicsnemo.active_learning.config.OptimizerConfig attribute), [1] score_to_epsilon() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) score_to_x0() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler 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() (physicsnemo.mesh.mesh.Mesh method) set_checkpoint_decoder() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_checkpoint_encoder() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_checkpoint_model() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_checkpoint_processor() (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) set_dataset() (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataPipe method) (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataPipe method) set_epoch() (physicsnemo.datapipes.dataloader.DataLoader method) (physicsnemo.datapipes.dataset.Dataset method) (physicsnemo.datapipes.multi_dataset.MultiDataset method) (physicsnemo.datapipes.readers.base.Reader method) (physicsnemo.datapipes.readers.numpy.NumpyReader method) (physicsnemo.datapipes.readers.tensorstore_zarr.TensorStoreZarrReader method) (physicsnemo.datapipes.readers.zarr.ZarrReader method) (physicsnemo.datapipes.transforms.base.Transform method) (physicsnemo.datapipes.transforms.compose.Compose method) set_generator() (physicsnemo.datapipes.dataset.Dataset method) (physicsnemo.datapipes.multi_dataset.MultiDataset method) (physicsnemo.datapipes.readers.base.Reader method) (physicsnemo.datapipes.readers.numpy.NumpyReader method) (physicsnemo.datapipes.readers.tensorstore_zarr.TensorStoreZarrReader method) (physicsnemo.datapipes.readers.zarr.ZarrReader method) (physicsnemo.datapipes.transforms.base.Transform method) (physicsnemo.datapipes.transforms.compose.Compose method) setup() (physicsnemo.distributed.manager.DistributedManager static method) shard_grid (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) shard_points (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) ShardTensor (class in physicsnemo.domain_parallel) short_run_id (physicsnemo.active_learning.driver.Driver property) sigma() (physicsnemo.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) (physicsnemo.diffusion.preconditioners.BaseAffinePreconditioner method) (physicsnemo.diffusion.preconditioners.VPPreconditioner method) sigma_dot() (physicsnemo.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) sigma_inv() (physicsnemo.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) signed_distance_field() (in module physicsnemo.nn.functional) SinusoidalTimestepEmbedding (class in physicsnemo.nn.module.embedding_layers) SirenLayer (class in physicsnemo.nn.module.siren_layers) SirenLayerType (class in physicsnemo.nn.module.siren_layers) 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] slice_cells() (physicsnemo.mesh.mesh.Mesh method) slice_points() (physicsnemo.mesh.mesh.Mesh method) Solver (class in physicsnemo.diffusion.samplers.solvers) SongUNet (class in physicsnemo.models.diffusion_unets) SongUNetPosEmbd (class in physicsnemo.models.diffusion_unets) SongUNetPosLtEmbd (class in physicsnemo.models.diffusion_unets) spectral_grid_gradient() (in module physicsnemo.nn.functional) 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) SquarePlus (class in physicsnemo.nn.module.activations) SRResNet (class in physicsnemo.models.srrn.super_res_net) StackedRandomGenerator (class in physicsnemo.diffusion.utils) Stan (class in physicsnemo.nn.module.activations) state_dict() (physicsnemo.datapipes.transforms.base.Transform method) (physicsnemo.datapipes.transforms.compose.Compose method) (physicsnemo.datapipes.transforms.normalize.Normalize method) (physicsnemo.mesh.mesh.Mesh method) (physicsnemo.optim.CombinedOptimizer method) StaticCaptureEvaluateNoGrad (class in physicsnemo.utils.capture) StaticCaptureTraining (class in physicsnemo.utils.capture) statistics (physicsnemo.mesh.mesh.Mesh property) step() (physicsnemo.diffusion.samplers.solvers.EDMStochasticEulerSolver method) (physicsnemo.diffusion.samplers.solvers.EDMStochasticHeunSolver method) (physicsnemo.diffusion.samplers.solvers.EulerSolver method) (physicsnemo.diffusion.samplers.solvers.HeunSolver method) (physicsnemo.diffusion.samplers.solvers.Solver method) (physicsnemo.optim.CombinedOptimizer method) stochastic (physicsnemo.datapipes.transforms.base.Transform property) (physicsnemo.datapipes.transforms.compose.Compose property) StokesDataset (class in physicsnemo.datapipes.gnn.stokes_dataset) 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] strip_caches() (physicsnemo.mesh.mesh.Mesh method) Strong Scaling StudentTEDMNoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) subdivide() (physicsnemo.mesh.mesh.Mesh method) SubPixel_ConvolutionalBlock3d (class in physicsnemo.models.srrn.super_res_net) subs() (physicsnemo.sym.eq.pde.PDE method) SubsamplePoints (class in physicsnemo.datapipes.transforms.subsample) success() (physicsnemo.utils.logging.console.PythonLogger method) 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) 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) SyntheticWeatherDataLoader (class in physicsnemo.datapipes.climate.synthetic) SyntheticWeatherDataset (class in physicsnemo.datapipes.climate.synthetic) T Tensor Parallel TensorStoreZarrReader (class in physicsnemo.datapipes.readers.tensorstore_zarr) TimeSeriesDataset (class in physicsnemo.datapipes.healpix.timeseries_dataset) timesteps() (physicsnemo.diffusion.noise_schedulers.DomainParallelNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.EDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.IDDPMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.NoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.StudentTEDMNoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VENoiseScheduler method) (physicsnemo.diffusion.noise_schedulers.VPNoiseScheduler method) to() (physicsnemo.datapipes.transforms.base.Transform method) (physicsnemo.datapipes.transforms.compose.Compose method) (physicsnemo.datapipes.transforms.normalize.Normalize method) (physicsnemo.models.graphcast.graph_cast_net.GraphCastNet method) to_dict() (physicsnemo.active_learning.config.OptimizerConfig method) (physicsnemo.active_learning.config.StrategiesConfig method) (physicsnemo.active_learning.config.TrainingConfig method) to_dual_graph() (physicsnemo.mesh.mesh.Mesh method) to_edge_graph() (physicsnemo.mesh.mesh.Mesh method) to_json() (physicsnemo.active_learning.config.DriverConfig method) to_local() (physicsnemo.domain_parallel.ShardTensor method) to_point_cloud() (physicsnemo.mesh.mesh.Mesh method) to_tensordict() (physicsnemo.mesh.mesh.Mesh method) toggle_mlflow() (physicsnemo.utils.logging.launch.LaunchLogger class method) toggle_wandb() (physicsnemo.utils.logging.launch.LaunchLogger class method) 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) TRAINING (physicsnemo.active_learning.protocols.ActiveLearningPhase 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) TrainingConfig (class in physicsnemo.active_learning.config) TrainingLoop (class in physicsnemo.active_learning.protocols) TrainingProtocol (class in physicsnemo.active_learning.protocols) Transform (class in physicsnemo.datapipes.transforms.base) transform() (in module physicsnemo.mesh.transformations.geometric) (physicsnemo.mesh.mesh.Mesh method) 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 (class in physicsnemo.datapipes.transforms.geometric) translate() (in module physicsnemo.mesh.transformations.geometric) (physicsnemo.mesh.mesh.Mesh method) translational_invariance (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) Transolver (class in physicsnemo.models.transolver.transolver) TransolverBlock (class in physicsnemo.models.transolver.transolver) TransolverDataConfig (class in physicsnemo.datapipes.cae.transolver_datapipe) TransolverDataPipe (class in physicsnemo.datapipes.cae.transolver_datapipe) TransposeConvLayer (class in physicsnemo.nn.module.conv_layers) truncate_helper() (in module physicsnemo.distributed.utils) U unbind() (physicsnemo.mesh.mesh.Mesh method) UNetAttention (class in physicsnemo.nn.module.attention_layers) UNetBlock (class in physicsnemo.nn.module.unet_layers) uniform_grid_gradient() (in module physicsnemo.nn.functional) 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) unscale_model_targets() (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataPipe 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) UpSample2D (class in physicsnemo.nn.module.resample_layers) UpSample3D (class in physicsnemo.nn.module.resample_layers) V val_datapool (physicsnemo.active_learning.config.TrainingConfig attribute), [1] (physicsnemo.active_learning.driver.Driver property) validate() (physicsnemo.mesh.mesh.Mesh method) validation_pool (physicsnemo.active_learning.protocols.DriverProtocol attribute), [1] validation_step() (physicsnemo.active_learning.protocols.LearnerProtocol method) ValidationProtocol (class in physicsnemo.active_learning.protocols) Variance (class in physicsnemo.metrics.general.ensemble_metrics) VENoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) VEPreconditioner (class in physicsnemo.diffusion.preconditioners) view_as_complex() (in module physicsnemo.nn.functional) (in module physicsnemo.nn.module.fft) 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) (physicsnemo.datapipes.cae.transolver_datapipe.TransolverDataConfig attribute) volume_variables (physicsnemo.datapipes.cae.domino_datapipe.DoMINODataConfig attribute) VortexSheddingDataset (class in physicsnemo.datapipes.gnn.vortex_shedding_dataset) VPNoiseScheduler (class in physicsnemo.diffusion.noise_schedulers) VPPreconditioner (class in physicsnemo.diffusion.preconditioners) VTKReader (class in physicsnemo.datapipes.readers.vtk) W warning() (physicsnemo.utils.logging.console.PythonLogger method) wasserstein_from_cdf() (in module physicsnemo.metrics.general.wasserstein) wasserstein_from_normal() (in module physicsnemo.metrics.general.wasserstein) wasserstein_from_samples() (in module physicsnemo.metrics.general.wasserstein) Weak Scaling weight_fact() (in module physicsnemo.nn.functional) WeightedMean (class in physicsnemo.metrics.general.reduction) WeightedMSEDSMLoss (class in physicsnemo.diffusion.metrics.losses) 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) world_size (physicsnemo.distributed.manager.DistributedManager property) X x0_to_epsilon() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) x0_to_score() (physicsnemo.diffusion.noise_schedulers.LinearGaussianNoiseScheduler method) xyz2latlon() (in module physicsnemo.models.graphcast.utils.graph_utils) Z ZarrReader (class in physicsnemo.datapipes.readers.zarr) zero_grad() (physicsnemo.optim.CombinedOptimizer method) zonal_mean() (in module physicsnemo.metrics.climate.reduction) zonal_var() (in module physicsnemo.metrics.climate.reduction)
