gensbi.models.healswin

Mirror of the standalone heal_swin_nnx package.

GenSBI depends on heal-swin-nnx (the HEALPix-native spherical Swin V2 U-Net in Flax NNX) and re-exports its public API here, so spherical encoders are importable alongside the other gensbi models. The SBI-relevant names — HealSwinEncoder and HealSwinParams — are also exported from gensbi.models directly; everything else (full U-Nets, decoders, HealConv, planar Swin) is available from this module or from heal_swin_nnx itself.