PhysicsNeMo General Release v2.2.0

Summary

PhysicsNeMo General Release v2.2.0 Added Adds ShardTensor support for GeoTransolver and FLARE models. Adds zarr save/load for Mesh and DomainMesh via tensordict's zarr storage backend: physicsnemo.mesh.io.to_zarr / from_zarr , with training-appropriate chunking and zstd compression. MeshReader and DomainMeshReader transparently read zarr stores alongside .pmsh / .pdmsh (opt in via pattern ). Requires optional zarr >= 3 and a tensordict release with the zarr backend. Promotes GeoTransolver out of experimental to physicsnemo.models.geotransolver.GeoTransolver , together with the FLARE model ( physicsnemo.models.flare.FLARE ) and the reusable GALE and FLARE attention layers ( physicsnemo.nn.GALE , physicsnemo.nn.GALEBlock , physicsnemo.nn.FLARE ). The embedded OOD guard is decoupled from the model. Wrap a constructed GeoTransolver with physicsnemo.experimental.guardrails.embedded.GuardedGeoTransolver (or attach_ood_guard ) to enable out-of-distribution guarding. PhysicsNeMo removes the guard_config model argument. Legacy import shims keep the pre-move physicsnemo.experimental import paths working and emit a LegacyFeatureWarning pointing to the new locations. Adds zenith_azimuth_angles and zenith_azimuth_angles_from_timestamp to physicsnemo.utils.zenith_angle , returning (sin_zenith, cos_zenith, sin_azimuth, cos_azimuth) alongside the existing cos_zenith_angle / cos_zenith_angle_from_timestamp helpers. The azimuth follows the north-clockwise convention. Adds physicsnemo.nn.shrink_and_perturb_ , an in-place shrink-and-perturb weight re-initialization for warm-starting from pretrained weights. Adds dimension-generic volume mesh generation for implicit domains to physicsnemo.mesh.generate . mesh_implicit_domain meshes {x : phi(x) < 0} , clipped to the bounding box (box faces are honored as a boundary, so that external-flow "box minus obstacle" domains work directly), for any implicit function (signed-distance functions, level sets, or neural fields). Adds compile-safe subclassing extension points to domain_parallel.ShardTensor ( _extra_inner_tensors , __subclass_flatten_context__ / __subclass_unflatten__ , _stable_inner_sentinel , and a DTensor-style __metadata_guard__ ), so a subclass can carry extra inner tensors and opaque metadata through torch.compile without re-implementing the flatten protocol. Base behavior unchanged. Adds ShardTensor._subclass_propagated_…