All modules for which code is available

• physicsnemo.active_learning.config
• physicsnemo.active_learning.driver
• physicsnemo.active_learning.loop
• physicsnemo.active_learning.protocols
• physicsnemo.core.filesystem
• physicsnemo.core.module
• physicsnemo.core.registry
• 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.collate
• physicsnemo.datapipes.dataloader
• physicsnemo.datapipes.dataset
• 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.datapipes.multi_dataset
• physicsnemo.datapipes.readers.base
• physicsnemo.datapipes.readers.hdf5
• physicsnemo.datapipes.readers.numpy
• physicsnemo.datapipes.readers.tensorstore_zarr
• physicsnemo.datapipes.readers.vtk
• physicsnemo.datapipes.readers.zarr
• physicsnemo.datapipes.transforms.base
• physicsnemo.datapipes.transforms.compose
• physicsnemo.datapipes.transforms.concat_fields
• physicsnemo.datapipes.transforms.field_processing
• physicsnemo.datapipes.transforms.field_slice
• physicsnemo.datapipes.transforms.geometric
• physicsnemo.datapipes.transforms.normalize
• physicsnemo.datapipes.transforms.spatial
• physicsnemo.datapipes.transforms.subsample
• physicsnemo.datapipes.transforms.utility
• physicsnemo.diffusion.base
• physicsnemo.diffusion.guidance.dps_guidance
• physicsnemo.diffusion.metrics.fid
• physicsnemo.diffusion.metrics.losses
• physicsnemo.diffusion.noise_schedulers.domain_parallel
• physicsnemo.diffusion.noise_schedulers.noise_schedulers
• physicsnemo.diffusion.preconditioners.preconditioners
• physicsnemo.diffusion.samplers.samplers
• physicsnemo.diffusion.samplers.solvers
• physicsnemo.diffusion.utils.utils
• physicsnemo.distributed.autograd
• physicsnemo.distributed.fft
• physicsnemo.distributed.manager
• physicsnemo.distributed.mappings
• physicsnemo.distributed.utils
• physicsnemo.domain_parallel.shard_tensor
• physicsnemo.mesh.mesh
• physicsnemo.mesh.transformations.geometric
• physicsnemo.mesh.visualization.draw_mesh
• 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.afno.afno
• physicsnemo.models.afno.modafno
• physicsnemo.models.diffusion_unets.dhariwal_unet
• physicsnemo.models.diffusion_unets.song_unet
• physicsnemo.models.dit.dit
• physicsnemo.models.dlwp.dlwp
• physicsnemo.models.dlwp_healpix.HEALPixRecUNet
• physicsnemo.models.domino.model
• physicsnemo.models.fengwu.fengwu
• physicsnemo.models.fno.fno
• physicsnemo.models.graphcast.graph_cast_net
• physicsnemo.models.graphcast.utils.graph
• physicsnemo.models.graphcast.utils.graph_utils
• physicsnemo.models.mesh_reduced.mesh_reduced
• physicsnemo.models.meshgraphnet.bsms_mgn
• physicsnemo.models.meshgraphnet.meshgraphnet
• physicsnemo.models.mlp.fully_connected
• physicsnemo.models.pangu.pangu
• physicsnemo.models.pix2pix.pix2pix
• physicsnemo.models.pix2pix.pix2pixunet
• physicsnemo.models.rnn.rnn_one2many
• physicsnemo.models.rnn.rnn_seq2seq
• physicsnemo.models.srrn.super_res_net
• physicsnemo.models.swinvrnn.swinvrnn
• physicsnemo.models.transolver.transolver
• physicsnemo.nn.functional.interpolation.grid_to_point_interpolation.grid_to_point_interpolation
• 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.group_norm
• physicsnemo.nn.module.gumbel_softmax
• 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.combined_optimizer
• physicsnemo.sym.eq.gradients
• physicsnemo.sym.eq.pde
• physicsnemo.sym.eq.phy_informer
• physicsnemo.utils.capture
• physicsnemo.utils.checkpoint
• physicsnemo.utils.insolation
• physicsnemo.utils.logging.console
• physicsnemo.utils.logging.launch
• physicsnemo.utils.logging.utils