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.mesh_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.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.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.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.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.utils.capture
• physicsnemo.utils.checkpoint
• physicsnemo.utils.insolation
• physicsnemo.utils.logging.console
• physicsnemo.utils.logging.launch
• physicsnemo.utils.logging.utils