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 A absolute_pos_embed (mfai.pytorch.models.half_unet.HalfUNetSettings attribute) AbsolutePosEmdebding (class in mfai.pytorch.models.utils) Activation (class in mfai.pytorch.models.deeplabv3) activation (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) add_module() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) add_skip_connections (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) add_special_tokens() (mfai.tokenizers.GPT2Tokenizer method), [1] (mfai.tokenizers.MiniGPT2Tokenizer method), [1] add_state() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) AddCoords (class in mfai.pytorch.models.gan_dgmr.layers.coord_conv) aggregate() (mfai.pytorch.models.nlam.interaction_net.InteractionNet method) aggregate_loss() (mfai.pytorch.losses.toolbelt.DiceLoss method), [1] all_gather() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) allow_zero_length_dataloader_with_multiple_devices (mfai.pytorch.lightning_modules.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.SegmentationLightningModule attribute) append() (mfai.pytorch.losses.perceptual.LinLayers method) apply() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) apply_rope() (in module mfai.pytorch.models.llms.llama3) ArchesWeather (class in mfai.pytorch.models.archesweather), [1] ArchesWeatherSettings (class in mfai.pytorch.models.archesweather) ASPP (class in mfai.pytorch.models.deeplabv3) ASPPConv (class in mfai.pytorch.models.deeplabv3) ASPPPooling (class in mfai.pytorch.models.deeplabv3) ASPPSeparableConv (class in mfai.pytorch.models.deeplabv3) assign() (in module mfai.pytorch), [1] Attention (class in mfai.pytorch.models.vit) attention_code (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) attention_einsum() (in module mfai.pytorch.models.gan_dgmr.layers.attention) AttentionLayer (class in mfai.pytorch.models.gan_dgmr.layers.attention) attn_drop_rate (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) attn_tf_compat (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) automatic_optimization (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) autopad_enabled (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) (mfai.pytorch.models.half_unet.HalfUNetSettings attribute) (mfai.pytorch.models.segformer.SegformerSettings attribute) (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) (mfai.pytorch.models.unet.CustomUNetSettings attribute) (mfai.pytorch.models.unet.UNetSettings attribute) (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) (mfai.pytorch.models.vit.ViTEncoderSettings attribute) AutoPaddingModel (class in mfai.pytorch.models.base) aux_params (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) axial_attn (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) axial_attn_heads (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) B backend (mfai.pytorch.models.llms.FreezeMLMMixin attribute), [1] (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) backward() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) BaseGraphModel (class in mfai.pytorch.models.nlam), [1], [2] BaseHiGraphModel (class in mfai.pytorch.models.nlam), [1], [2] BaseModel (class in mfai.pytorch.models.base), [1] BaseNet (class in mfai.pytorch.losses.perceptual), [1] bfloat16() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) bias (mfai.pytorch.models.half_unet.HalfUNetSettings attribute) binary_dilation_() (mfai.pytorch.metrics.CSINeighborhood method), [1] buffers() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) build_graph_for_grid() (in module mfai.pytorch.models.nlam.create_mesh) C call_super_init (mfai.pytorch.lightning_modules.clip.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule attribute) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule attribute) (mfai.pytorch.losses.gan_dgmr.GridCellLoss attribute) (mfai.pytorch.losses.perceptual.BaseNet attribute) (mfai.pytorch.losses.perceptual.LinLayers attribute) (mfai.pytorch.losses.perceptual.LPIPS attribute) (mfai.pytorch.losses.perceptual.PerceptualLoss attribute) (mfai.pytorch.losses.perceptual.VGG16 attribute) (mfai.pytorch.losses.toolbelt.DiceLoss attribute) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss attribute) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss attribute) (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) (mfai.pytorch.models.archesweather.ArchesWeather attribute) (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute) (mfai.pytorch.models.identity.IdentityModel attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute) (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute) (mfai.pytorch.models.segformer.Segformer attribute) (mfai.pytorch.models.swinunetr.SwinUNetR attribute) (mfai.pytorch.models.unet.CustomUNet attribute) (mfai.pytorch.models.unet.UNet attribute) (mfai.pytorch.models.unetrpp.UNetRPP attribute) (mfai.pytorch.models.vit.ViTClassifier attribute) (mfai.pytorch.models.vit.VitEncoder attribute) (mfai.pytorch.transforms.DimensionSubSampler attribute) (mfai.pytorch.transforms.MeanDimensionSubsampler attribute) cast_tuple() (in module mfai.pytorch.models.segformer) check_input_shape() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) check_required_attributes() (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.base.ModelABC method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) checkpoint_activation (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) CHECKPOINT_HYPER_PARAMS_KEY (mfai.pytorch.lightning_modules.clip.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule attribute) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule attribute) CHECKPOINT_HYPER_PARAMS_NAME (mfai.pytorch.lightning_modules.clip.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule attribute) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule attribute) CHECKPOINT_HYPER_PARAMS_TYPE (mfai.pytorch.lightning_modules.clip.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule attribute) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule attribute) CheckpointWrapper (class in mfai.pytorch.models.nlam.interaction_net) children() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) classification_head (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) Clip (class in mfai.pytorch.models.clip) clip_gradients() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) CLIPAccuracySkillScore (class in mfai.pytorch.lightning_modules.clip) CLIPLightningModule (class in mfai.pytorch.lightning_modules), [1] (class in mfai.pytorch.lightning_modules.clip), [1] ClipSettings (class in mfai.pytorch.models.clip) clone() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.namedtensor.NamedTensor method) collate_fn() (mfai.pytorch.namedtensor.NamedTensor static method) collate_fn_fit() (mfai.pytorch.dummy_dataset.DummyMultiModalDataModule method) collate_text() (mfai.pytorch.dummy_dataset.DummyMultiModalDataModule method) compile() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) compute() (mfai.pytorch.lightning_modules.clip.CLIPAccuracySkillScore method) (mfai.pytorch.metrics.CSINeighborhood method), [1] (mfai.pytorch.metrics.FAR method), [1] (mfai.pytorch.metrics.FNR method), [1] (mfai.pytorch.metrics.FSS method), [1] (mfai.pytorch.metrics.PR_AUC method), [1] compute_fbs_wfbs() (mfai.pytorch.metrics.FSS method), [1] compute_perceptual_features() (mfai.pytorch.losses.perceptual.PerceptualLoss method), [1] compute_rope() (in module mfai.pytorch.models.llms.llama2) compute_rope_params() (in module mfai.pytorch.models.llms.llama3) compute_score() (mfai.pytorch.losses.toolbelt.DiceLoss method), [1] concat() (mfai.pytorch.namedtensor.NamedTensor static method) cond_dim (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) CondBasicLayer (class in mfai.pytorch.models.archesweather) configure_callbacks() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) configure_gradient_clipping() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) configure_model() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) configure_optimizers() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method), [1] (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] configure_sharded_model() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) context_length (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLM property) (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.Fuyu property) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) (mfai.pytorch.models.llms.llama2.Llama2Settings attribute) (mfai.pytorch.models.llms.llama3.Llama3Settings attribute) ContextConditioningStack (class in mfai.pytorch.models.gan_dgmr.blocks) conv_head (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) conv_op (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) ConvGRU (class in mfai.pytorch.models.gan_dgmr.layers.conv_gru) ConvGRUCell (class in mfai.pytorch.models.gan_dgmr.layers.conv_gru) CONVOLUTIONAL (mfai.pytorch.models.base.ModelType attribute) CoordConv (class in mfai.pytorch.models.gan_dgmr.layers.coord_conv) cos (mfai.pytorch.models.llms.llama2.MultiHeadAttentionPySDPALlama2 attribute) cpu() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) CrossAttentionGPT2 (class in mfai.pytorch.models.llms.gpt2) CrossAttentionGPT2Settings (class in mfai.pytorch.models.llms.gpt2) CrossAttentionTransformerBlock (class in mfai.pytorch.models.llms.gpt2) CSINeighborhood (class in mfai.pytorch.metrics), [1] cuda() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) current_epoch (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) CustomPad2d (class in mfai.pytorch.models.pangu) CustomPad3d (class in mfai.pytorch.models.pangu) CustomUNet (class in mfai.pytorch.models.unet), [1] CustomUNetSettings (class in mfai.pytorch.models.unet) D DBlock (class in mfai.pytorch.models.gan_dgmr.blocks) decode() (mfai.tokenizers.GPT2Tokenizer method), [1] (mfai.tokenizers.LlamaTokenizer method), [1] (mfai.tokenizers.MiniGPT2Tokenizer method), [1] (mfai.tokenizers.Tokenizer method), [1] decoder (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) decoder_atrous_rates (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) decoder_channels (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) decoder_dim (mfai.pytorch.models.segformer.SegformerSettings attribute) decoder_proj_size (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) DeepLabV3 (class in mfai.pytorch.models.deeplabv3), [1] DeepLabV3Decoder (class in mfai.pytorch.models.deeplabv3) DeepLabV3Plus (class in mfai.pytorch.models.deeplabv3), [1] DeepLabV3PlusDecoder (class in mfai.pytorch.models.deeplabv3) DeepLabV3PlusSettings (class in mfai.pytorch.models.deeplabv3) DeepLabV3Settings (class in mfai.pytorch.models.deeplabv3) define_3d_earth_position_index() (in module mfai.pytorch.models.pangu) depth_multiplier (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) depths (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) device (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) (mfai.pytorch.metrics.CSINeighborhood property) (mfai.pytorch.metrics.FAR property) (mfai.pytorch.metrics.FNR property) (mfai.pytorch.metrics.FSS property) (mfai.pytorch.metrics.PR_AUC property) (mfai.pytorch.namedtensor.NamedTensor property) device_mesh (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) DGMRLightningModule (class in mfai.pytorch.lightning_modules.gan_dgmr), [1] DiceLoss (class in mfai.pytorch.losses.toolbelt), [1] dilation (mfai.pytorch.models.half_unet.HalfUNetSettings attribute) dim_index() (mfai.pytorch.namedtensor.NamedTensor method) dim_size() (mfai.pytorch.namedtensor.NamedTensor method) DimensionSubSampler (class in mfai.pytorch.transforms), [1] dims (mfai.pytorch.models.segformer.SegformerSettings attribute) Discriminator (class in mfai.pytorch.models.gan_dgmr.discriminators) discriminator_step() (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method), [1] do_ds (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) double() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) DoubleConv (class in mfai.pytorch.models.unet) dowload_weights_from_tf_ckpt() (mfai.pytorch.models.llms.gpt2.GPT2 method) download_and_load_gpt2() (in module mfai.tensorflow) download_file() (in module mfai.http) DownSample (class in mfai.pytorch.models.pangu) downsample (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) downsampling_rate (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) drop_rate (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) dropout_path_rate (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) dropout_rate (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) droppath_coeff (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) DsConv2d (class in mfai.pytorch.models.segformer) dtype (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) (mfai.pytorch.metrics.CSINeighborhood property) (mfai.pytorch.metrics.FAR property) (mfai.pytorch.metrics.FNR property) (mfai.pytorch.metrics.FSS property) (mfai.pytorch.metrics.PR_AUC property) DummyDataModule (class in mfai.pytorch.dummy_dataset) DummyDataset (class in mfai.pytorch.dummy_dataset) DummyMultiModalDataModule (class in mfai.pytorch.dummy_dataset) DummyMultiModalDataset (class in mfai.pytorch.dummy_dataset) dump_patches (mfai.pytorch.lightning_modules.clip.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule attribute) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule attribute) (mfai.pytorch.losses.gan_dgmr.GridCellLoss attribute) (mfai.pytorch.losses.perceptual.BaseNet attribute) (mfai.pytorch.losses.perceptual.LinLayers attribute) (mfai.pytorch.losses.perceptual.LPIPS attribute) (mfai.pytorch.losses.perceptual.PerceptualLoss attribute) (mfai.pytorch.losses.perceptual.VGG16 attribute) (mfai.pytorch.losses.toolbelt.DiceLoss attribute) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss attribute) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss attribute) (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) (mfai.pytorch.models.archesweather.ArchesWeather attribute) (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute) (mfai.pytorch.models.identity.IdentityModel attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute) (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute) (mfai.pytorch.models.segformer.Segformer attribute) (mfai.pytorch.models.swinunetr.SwinUNetR attribute) (mfai.pytorch.models.unet.CustomUNet attribute) (mfai.pytorch.models.unet.UNet attribute) (mfai.pytorch.models.unetrpp.UNetRPP attribute) (mfai.pytorch.models.vit.ViTClassifier attribute) (mfai.pytorch.models.vit.VitEncoder attribute) (mfai.pytorch.transforms.DimensionSubSampler attribute) (mfai.pytorch.transforms.MeanDimensionSubsampler attribute) E EarthAttention3D (class in mfai.pytorch.models.pangu) EarthSpecificBlock (class in mfai.pytorch.models.archesweather) (class in mfai.pytorch.models.pangu) EarthSpecificLayer (class in mfai.pytorch.models.archesweather) (class in mfai.pytorch.models.pangu) EfficientSelfAttention (class in mfai.pytorch.models.segformer) emb_dim (mfai.pytorch.models.clip.ClipSettings attribute) (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) (mfai.pytorch.models.llms.llama2.Llama2Settings attribute) (mfai.pytorch.models.llms.llama3.Llama3Settings attribute) (mfai.pytorch.models.vit.ViTEncoderSettings attribute) emb_dropout (mfai.pytorch.models.vit.ViTEncoderSettings attribute) embed_tokens() (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2 method) (mfai.pytorch.models.llms.gpt2.GPT2 method) (mfai.pytorch.models.llms.llama2.Llama2 method) (mfai.pytorch.models.llms.llama3.Llama3 method) embedd_mesh_nodes() (mfai.pytorch.models.nlam.BaseGraphModel method), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel method), [1], [2] (mfai.pytorch.models.nlam.GraphLAM method), [1], [2] (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) embedding_dim (mfai.pytorch.models.weather_projector.WeatherProjectorSettings attribute) encode() (mfai.tokenizers.GPT2Tokenizer method), [1] (mfai.tokenizers.LlamaTokenizer method), [1] (mfai.tokenizers.MiniGPT2Tokenizer method), [1] (mfai.tokenizers.Tokenizer method), [1] encode_text() (mfai.pytorch.models.clip.Clip method) encoder_depth (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) (mfai.pytorch.models.resnet.ResNet50MLMSettings attribute) (mfai.pytorch.models.resnet.ResNet50Settings attribute) (mfai.pytorch.models.unet.CustomUNetSettings attribute) encoder_name (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) (mfai.pytorch.models.unet.CustomUNetSettings attribute) encoder_output_stride (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) encoder_proj_sizes (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) encoder_stride (mfai.pytorch.models.resnet.ResNet50MLMSettings attribute) (mfai.pytorch.models.resnet.ResNet50Settings attribute) encoder_weights (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) (mfai.pytorch.models.resnet.ResNet50MLMSettings attribute) (mfai.pytorch.models.resnet.ResNet50Settings attribute) (mfai.pytorch.models.unet.CustomUNetSettings attribute) eot_token (mfai.tokenizers.GPT2Tokenizer property), [1] (mfai.tokenizers.LlamaTokenizer property), [1] (mfai.tokenizers.MiniGPT2Tokenizer property), [1] (mfai.tokenizers.Tokenizer property), [1] EPA (class in mfai.pytorch.models.unetrpp) eval() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) example_input_array (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) exists() (in module mfai.pytorch.models.segformer) expand_to_batch() (in module mfai.pytorch.models.utils) expand_to_batch_like() (mfai.pytorch.namedtensor.NamedTensor static method) export_to_onnx() (in module mfai.pytorch), [1] extend() (mfai.pytorch.losses.perceptual.LinLayers method) extra_repr() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) F fabric (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) FAR (class in mfai.pytorch.metrics), [1] feat_size (mfai.pytorch.models.unetrpp.UNetRPP attribute) feature_dim_idx (mfai.pytorch.namedtensor.NamedTensor property) feature_size (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) features_last (mfai.pytorch.models.archesweather.ArchesWeather attribute), [1] (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute), [1] (mfai.pytorch.models.segformer.Segformer attribute), [1] (mfai.pytorch.models.swinunetr.SwinUNetR attribute), [1] (mfai.pytorch.models.unet.CustomUNet attribute), [1] (mfai.pytorch.models.unet.UNet attribute), [1] (mfai.pytorch.models.unetrpp.UNetRPP attribute), [1] (mfai.pytorch.models.vit.ViTClassifier attribute), [1] (mfai.pytorch.models.vit.VitEncoder attribute), [1] features_last_to_second() (in module mfai.pytorch.models.utils) features_second (mfai.pytorch.models.archesweather.ArchesWeather property) (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 property) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus property) (mfai.pytorch.models.half_unet.HalfUNet property) (mfai.pytorch.models.identity.IdentityModel property) (mfai.pytorch.models.nlam.BaseGraphModel property) (mfai.pytorch.models.nlam.BaseHiGraphModel property) (mfai.pytorch.models.nlam.GraphLAM property) (mfai.pytorch.models.nlam.HiLAM property) (mfai.pytorch.models.nlam.HiLAMParallel property) (mfai.pytorch.models.pangu.PanguWeather property) (mfai.pytorch.models.segformer.Segformer property) (mfai.pytorch.models.swinunetr.SwinUNetR property) (mfai.pytorch.models.unet.CustomUNet property) (mfai.pytorch.models.unet.UNet property) (mfai.pytorch.models.unetrpp.UNetRPP property) (mfai.pytorch.models.vit.ViTClassifier property) (mfai.pytorch.models.vit.VitEncoder property) features_second_to_last() (in module mfai.pytorch.models.utils) FeedForward (class in mfai.pytorch.models.llms.gpt2) (class in mfai.pytorch.models.vit) FeedForwardLlama2 (class in mfai.pytorch.models.llms.llama2) ff_expansion (mfai.pytorch.models.segformer.SegformerSettings attribute) finalize_graph_model() (mfai.pytorch.models.nlam.BaseGraphModel method), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel method), [1], [2] (mfai.pytorch.models.nlam.GraphLAM method), [1], [2] (mfai.pytorch.models.nlam.HiLAM method), [1], [2] (mfai.pytorch.models.nlam.HiLAMParallel method), [1], [2] first_interaction_layer (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) flatten_() (mfai.pytorch.namedtensor.NamedTensor method) float() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) FNR (class in mfai.pytorch.metrics), [1] forward() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method), [1] (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] (mfai.pytorch.losses.gan_dgmr.GridCellLoss method), [1] (mfai.pytorch.losses.perceptual.BaseNet method), [1] (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method), [1] (mfai.pytorch.losses.perceptual.PerceptualLoss method), [1] (mfai.pytorch.losses.perceptual.VGG16 method), [1] (mfai.pytorch.losses.toolbelt.DiceLoss method), [1] (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method), [1] (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method), [1] (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method), [1] (mfai.pytorch.models.archesweather.CondBasicLayer method) (mfai.pytorch.models.archesweather.EarthSpecificBlock method) (mfai.pytorch.models.archesweather.EarthSpecificLayer method) (mfai.pytorch.models.archesweather.Interpolate method) (mfai.pytorch.models.archesweather.LinVert method) (mfai.pytorch.models.archesweather.PatchRecoveryConv method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.clip.Clip method) (mfai.pytorch.models.deeplabv3.Activation method) (mfai.pytorch.models.deeplabv3.ASPP method) (mfai.pytorch.models.deeplabv3.ASPPPooling method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Decoder method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.deeplabv3.DeepLabV3PlusDecoder method) (mfai.pytorch.models.gan_dgmr.blocks.ContextConditioningStack method) (mfai.pytorch.models.gan_dgmr.blocks.DBlock method) (mfai.pytorch.models.gan_dgmr.blocks.GBlock method) (mfai.pytorch.models.gan_dgmr.blocks.LatentConditioningStack method) (mfai.pytorch.models.gan_dgmr.blocks.LBlock method) (mfai.pytorch.models.gan_dgmr.blocks.UpsampleGBlock method) (mfai.pytorch.models.gan_dgmr.discriminators.Discriminator method) (mfai.pytorch.models.gan_dgmr.discriminators.SpatialDiscriminator method) (mfai.pytorch.models.gan_dgmr.discriminators.TemporalDiscriminator method) (mfai.pytorch.models.gan_dgmr.generators.Generator method) (mfai.pytorch.models.gan_dgmr.generators.Sampler method) (mfai.pytorch.models.gan_dgmr.layers.attention.AttentionLayer method) (mfai.pytorch.models.gan_dgmr.layers.conv_gru.ConvGRU method) (mfai.pytorch.models.gan_dgmr.layers.conv_gru.ConvGRUCell method) (mfai.pytorch.models.gan_dgmr.layers.coord_conv.AddCoords method) (mfai.pytorch.models.gan_dgmr.layers.coord_conv.CoordConv method) (mfai.pytorch.models.half_unet.GhostModule method) (mfai.pytorch.models.half_unet.HalfUNet method), [1] (mfai.pytorch.models.identity.IdentityModel method), [1] (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLM method) (mfai.pytorch.models.llms.fuyu.Fuyu method) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2 method) (mfai.pytorch.models.llms.gpt2.CrossAttentionTransformerBlock method) (mfai.pytorch.models.llms.gpt2.FeedForward method) (mfai.pytorch.models.llms.gpt2.GELU method) (mfai.pytorch.models.llms.gpt2.GPT2 method) (mfai.pytorch.models.llms.gpt2.LayerNorm method) (mfai.pytorch.models.llms.gpt2.MultiHeadAttention method) (mfai.pytorch.models.llms.gpt2.MultiHeadAttentionPySDPA method) (mfai.pytorch.models.llms.gpt2.MultiHeadCrossAttentionPySDPA method) (mfai.pytorch.models.llms.gpt2.TransformerBlock method) (mfai.pytorch.models.llms.llama2.FeedForwardLlama2 method) (mfai.pytorch.models.llms.llama2.Llama2 method) (mfai.pytorch.models.llms.llama2.MultiHeadAttentionPySDPALlama2 method) (mfai.pytorch.models.llms.llama2.RMSNorm method) (mfai.pytorch.models.llms.llama2.SiLU method) (mfai.pytorch.models.llms.llama2.TransformerBlockLlama2 method) (mfai.pytorch.models.llms.llama3.GroupedQueryAttention method) (mfai.pytorch.models.llms.llama3.Llama3 method) (mfai.pytorch.models.llms.llama3.TransformerBlock method) (mfai.pytorch.models.nlam.BaseGraphModel method), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.nlam.interaction_net.CheckpointWrapper method) (mfai.pytorch.models.nlam.interaction_net.InteractionNet method) (mfai.pytorch.models.nlam.interaction_net.SplitMLPs method) (mfai.pytorch.models.pangu.DownSample method) (mfai.pytorch.models.pangu.EarthAttention3D method) (mfai.pytorch.models.pangu.EarthSpecificBlock method) (mfai.pytorch.models.pangu.EarthSpecificLayer method) (mfai.pytorch.models.pangu.MLP method) (mfai.pytorch.models.pangu.PanguWeather method), [1] (mfai.pytorch.models.pangu.PatchEmbedding method) (mfai.pytorch.models.pangu.PatchRecovery method) (mfai.pytorch.models.pangu.UpSample method) (mfai.pytorch.models.resnet.ResNet50 method) (mfai.pytorch.models.resnet.ResNet50MLM method) (mfai.pytorch.models.resnet.ResNetEncoder method) (mfai.pytorch.models.segformer.DsConv2d method) (mfai.pytorch.models.segformer.EfficientSelfAttention method) (mfai.pytorch.models.segformer.LayerNorm method) (mfai.pytorch.models.segformer.MiT method) (mfai.pytorch.models.segformer.MixFeedForward method) (mfai.pytorch.models.segformer.PreNorm method) (mfai.pytorch.models.segformer.Segformer method), [1] (mfai.pytorch.models.swinunetr.SwinUNetR method), [1] (mfai.pytorch.models.swinunetr.UpsampleBlock method) (mfai.pytorch.models.unet.CustomUNet method), [1] (mfai.pytorch.models.unet.DoubleConv method) (mfai.pytorch.models.unet.UNet method), [1] (mfai.pytorch.models.unetrpp.EPA method) (mfai.pytorch.models.unetrpp.LayerNorm method) (mfai.pytorch.models.unetrpp.TransformerBlock method) (mfai.pytorch.models.unetrpp.UNetRPP method), [1] (mfai.pytorch.models.unetrpp.UNetRPPEncoder method) (mfai.pytorch.models.unetrpp.UNetRUpBlock method) (mfai.pytorch.models.utils.AbsolutePosEmdebding method) (mfai.pytorch.models.vit.Attention method) (mfai.pytorch.models.vit.FeedForward method) (mfai.pytorch.models.vit.Transformer method) (mfai.pytorch.models.vit.ViTClassifier method), [1] (mfai.pytorch.models.vit.ViTCore method) (mfai.pytorch.models.vit.VitEncoder method), [1] (mfai.pytorch.models.weather_projector.PatchMaker method) (mfai.pytorch.models.weather_projector.WeatherProjector method) (mfai.pytorch.transforms.DimensionSubSampler method), [1] (mfai.pytorch.transforms.MeanDimensionSubsampler method), [1] forward_features() (mfai.pytorch.models.unetrpp.UNetRPPEncoder method) forward_vectors() (mfai.pytorch.models.llms.gpt2.GPT2 method) (mfai.pytorch.models.llms.llama2.Llama2 method) (mfai.pytorch.models.llms.llama3.Llama3 method) freeze() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) freeze_llm() (mfai.pytorch.models.llms.FreezeMLMMixin method), [1] freeze_vision() (mfai.pytorch.models.llms.FreezeMLMMixin method), [1] FreezeMLMMixin (class in mfai.pytorch.models.llms), [1] from_dict() (mfai.pytorch.models.archesweather.ArchesWeatherSettings class method) (mfai.pytorch.models.clip.ClipSettings class method) (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings class method) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings class method) (mfai.pytorch.models.half_unet.HalfUNetSettings class method) (mfai.pytorch.models.identity.IdentityModelSettings class method) (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings class method) (mfai.pytorch.models.llms.fuyu.FuyuSettings class method) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings class method) (mfai.pytorch.models.llms.gpt2.GPT2Settings class method) (mfai.pytorch.models.llms.llama2.Llama2Settings class method) (mfai.pytorch.models.llms.llama3.Llama3Settings class method) (mfai.pytorch.models.nlam.GraphLamSettings class method), [1] (mfai.pytorch.models.pangu.PanguWeatherSettings class method) (mfai.pytorch.models.resnet.ResNet50MLMSettings class method) (mfai.pytorch.models.resnet.ResNet50Settings class method) (mfai.pytorch.models.segformer.SegformerSettings class method) (mfai.pytorch.models.swinunetr.SwinUNetRSettings class method) (mfai.pytorch.models.unet.CustomUNetSettings class method) (mfai.pytorch.models.unet.UNetSettings class method) (mfai.pytorch.models.unetrpp.UNetRPPSettings class method) (mfai.pytorch.models.vit.ViTClassifierSettings class method) (mfai.pytorch.models.vit.ViTEncoderSettings class method) (mfai.pytorch.models.weather_projector.WeatherProjectorSettings class method) from_json() (mfai.pytorch.models.archesweather.ArchesWeatherSettings class method) (mfai.pytorch.models.clip.ClipSettings class method) (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings class method) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings class method) (mfai.pytorch.models.half_unet.HalfUNetSettings class method) (mfai.pytorch.models.identity.IdentityModelSettings class method) (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings class method) (mfai.pytorch.models.llms.fuyu.FuyuSettings class method) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings class method) (mfai.pytorch.models.llms.gpt2.GPT2Settings class method) (mfai.pytorch.models.llms.llama2.Llama2Settings class method) (mfai.pytorch.models.llms.llama3.Llama3Settings class method) (mfai.pytorch.models.nlam.GraphLamSettings class method), [1] (mfai.pytorch.models.pangu.PanguWeatherSettings class method) (mfai.pytorch.models.resnet.ResNet50MLMSettings class method) (mfai.pytorch.models.resnet.ResNet50Settings class method) (mfai.pytorch.models.segformer.SegformerSettings class method) (mfai.pytorch.models.swinunetr.SwinUNetRSettings class method) (mfai.pytorch.models.unet.CustomUNetSettings class method) (mfai.pytorch.models.unet.UNetSettings class method) (mfai.pytorch.models.unetrpp.UNetRPPSettings class method) (mfai.pytorch.models.vit.ViTClassifierSettings class method) (mfai.pytorch.models.vit.ViTEncoderSettings class method) (mfai.pytorch.models.weather_projector.WeatherProjectorSettings class method) from_networkx_with_start_index() (in module mfai.pytorch.models.nlam.create_mesh) FSS (class in mfai.pytorch.metrics), [1] full_state_update (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) Fuyu (class in mfai.pytorch.models.llms.fuyu) FuyuSettings (class in mfai.pytorch.models.llms.fuyu) G g2m_edges (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1] (mfai.pytorch.models.nlam.GraphLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1] GBlock (class in mfai.pytorch.models.gan_dgmr.blocks) GELU (class in mfai.pytorch.models.llms.gpt2) generate_3d_attention_mask() (in module mfai.pytorch.models.pangu) Generator (class in mfai.pytorch.models.gan_dgmr.generators) generator_step() (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method), [1] get_buffer() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) get_classification_head() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) get_conv_layer() (in module mfai.pytorch.models.gan_dgmr.layers), [1] get_crop() (mfai.pytorch.transforms.RandomCropWithMinPositivePixels method), [1] get_decoder() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) get_extra_state() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) get_hparams() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] get_lr() (mfai.pytorch.lr_scheduler.LinearWarmupCosineAnnealingLR method) get_metrics() (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] get_network() (in module mfai.pytorch.losses.perceptual) get_num_mesh() (mfai.pytorch.models.nlam.BaseGraphModel method), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel method), [1], [2] (mfai.pytorch.models.nlam.GraphLAM method), [1], [2] (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) get_one_crop() (mfai.pytorch.transforms.RandomCropWithMinPositivePixels method), [1] get_parameter() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) get_resnet_encoder() (in module mfai.pytorch.models.resnet) get_segmentation_head() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) get_state_dict() (in module mfai.pytorch.losses.perceptual) get_submodule() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) get_tiktoken_encoding() (in module mfai.encoding) GhostModule (class in mfai.pytorch.models.half_unet) global_rank (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) global_step (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) GPT2 (class in mfai.pytorch.models.llms.gpt2) GPT2Settings (class in mfai.pytorch.models.llms.gpt2) GPT2Tokenizer (class in mfai.tokenizers), [1] GRAPH (mfai.pytorch.models.base.ModelType attribute) GraphLAM (class in mfai.pytorch.models.nlam), [1], [2] GraphLamSettings (class in mfai.pytorch.models.nlam), [1] grid2mesh() (in module mfai.pytorch.models.nlam.create_mesh) GridCellLoss (class in mfai.pytorch.losses.gan_dgmr), [1] GroupedQueryAttention (class in mfai.pytorch.models.llms.llama3) H half() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) HalfUNet (class in mfai.pytorch.models.half_unet), [1] HalfUNetSettings (class in mfai.pytorch.models.half_unet) heads (mfai.pytorch.models.segformer.SegformerSettings attribute) hi_processor_step() (mfai.pytorch.models.nlam.BaseHiGraphModel method), [1], [2] (mfai.pytorch.models.nlam.HiLAM method), [1], [2] (mfai.pytorch.models.nlam.HiLAMParallel method), [1], [2] hidden_dim (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.llama2.Llama2Settings attribute) (mfai.pytorch.models.llms.llama3.Llama3Settings attribute) hidden_dims (mfai.pytorch.models.nlam.GraphLamSettings attribute), [1] hidden_layers (mfai.pytorch.models.nlam.GraphLamSettings attribute), [1] hidden_size (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) hierarchical (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) hierarchical_mesh() (in module mfai.pytorch.models.nlam.create_mesh) higher_is_better (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) HiLAM (class in mfai.pytorch.models.nlam), [1], [2] HiLAMParallel (class in mfai.pytorch.models.nlam), [1], [2] hparams (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) hparams_initial (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) I IDENTITY (mfai.pytorch.models.base.ModelType attribute) IdentityModel (class in mfai.pytorch.models.identity), [1] IdentityModelSettings (class in mfai.pytorch.models.identity) image_encoder (mfai.pytorch.models.clip.ClipSettings attribute) in_channels (mfai.pytorch.models.archesweather.ArchesWeather attribute) (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.base.ModelABC attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute) (mfai.pytorch.models.identity.IdentityModel attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1] (mfai.pytorch.models.nlam.GraphLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1] (mfai.pytorch.models.pangu.PanguWeather attribute) (mfai.pytorch.models.segformer.Segformer attribute) (mfai.pytorch.models.swinunetr.SwinUNetR attribute) (mfai.pytorch.models.unet.CustomUNet attribute) (mfai.pytorch.models.unet.UNet attribute) (mfai.pytorch.models.unetrpp.UNetRPP attribute) (mfai.pytorch.models.vit.ViTClassifier attribute) (mfai.pytorch.models.vit.VitEncoder attribute) index_select_dim() (mfai.pytorch.namedtensor.NamedTensor method) index_select_tensor_dim() (mfai.pytorch.namedtensor.NamedTensor method) init_() (in module mfai.pytorch.models.unetrpp) (in module mfai.pytorch.models.utils) init_features (mfai.pytorch.models.unet.UNetSettings attribute) init_temperature (mfai.pytorch.models.clip.ClipSettings attribute) initialize() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) initialize_decoder() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) initialize_head() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) inject_vision_each_stage (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) input_dims (mfai.pytorch.models.weather_projector.WeatherProjectorSettings attribute) input_shape (mfai.pytorch.models.archesweather.ArchesWeather attribute) (mfai.pytorch.models.base.AutoPaddingModel attribute) (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.base.ModelABC attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute) (mfai.pytorch.models.identity.IdentityModel attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1] (mfai.pytorch.models.nlam.GraphLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1] (mfai.pytorch.models.pangu.PanguWeather attribute) (mfai.pytorch.models.segformer.Segformer attribute) (mfai.pytorch.models.swinunetr.SwinUNetR attribute) (mfai.pytorch.models.unet.CustomUNet attribute) (mfai.pytorch.models.unet.UNet attribute) (mfai.pytorch.models.unetrpp.UNetRPP attribute) (mfai.pytorch.models.vit.ViTClassifier attribute) (mfai.pytorch.models.vit.VitEncoder attribute) insert() (mfai.pytorch.losses.perceptual.LinLayers method) InteractionNet (class in mfai.pytorch.models.nlam.interaction_net) Interpolate (class in mfai.pytorch.models.archesweather) ipu() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) is_differentiable (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) iter_dim() (mfai.pytorch.namedtensor.NamedTensor method) iter_tensor_dim() (mfai.pytorch.namedtensor.NamedTensor method) K kernel_and_stride (mfai.pytorch.models.segformer.SegformerSettings attribute) L label_smoothed_nll_loss() (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method), [1] lam (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) last_activation (mfai.pytorch.models.half_unet.HalfUNetSettings attribute) last_activation() (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] LatentConditioningStack (class in mfai.pytorch.models.gan_dgmr.blocks) layer_depth (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) layer_norm_vis (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) layer_norm_viz (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) layer_norm_viz_txt (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) LayerNorm (class in mfai.pytorch.models.llms.gpt2) (class in mfai.pytorch.models.segformer) (class in mfai.pytorch.models.unetrpp) LBlock (class in mfai.pytorch.models.gan_dgmr.blocks) linear_upsampling (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) LinearWarmupCosineAnnealingLR (class in mfai.pytorch.lr_scheduler) LinLayers (class in mfai.pytorch.losses.perceptual), [1] LinVert (class in mfai.pytorch.models.archesweather) Llama2 (class in mfai.pytorch.models.llms.llama2) Llama2Settings (class in mfai.pytorch.models.llms.llama2) Llama3 (class in mfai.pytorch.models.llms.llama3) Llama3Settings (class in mfai.pytorch.models.llms.llama3) LlamaTokenizer (class in mfai.tokenizers), [1] LLM (mfai.pytorch.models.base.ModelType attribute) load_from() (mfai.pytorch.models.swinunetr.SwinUNetR method) load_from_checkpoint() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) load_from_settings_file() (in module mfai.pytorch.models), [1] load_graph() (mfai.pytorch.models.nlam.BaseGraphModel method), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) load_state_dict() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.resnet.ResNetEncoder method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) load_weights_from_dict() (mfai.pytorch.models.llms.gpt2.GPT2 method) load_weights_from_tf_checkpoint() (in module mfai.tensorflow) local_rank (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) log() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) log_dict() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) logger (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) loggers (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) loss_hinge_disc() (in module mfai.pytorch.losses.gan_dgmr) loss_hinge_gen() (in module mfai.pytorch.losses.gan_dgmr) LPIPS (class in mfai.pytorch.losses.perceptual), [1] lr_scheduler_step() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) lr_schedulers() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) M m2g_edges (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1] (mfai.pytorch.models.nlam.GraphLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1] make_dilated() (mfai.pytorch.models.resnet.ResNetEncoder method) make_down_gnns() (mfai.pytorch.models.nlam.HiLAM method), [1], [2] make_mlp() (in module mfai.pytorch.models.nlam.interaction_net) make_same_gnns() (mfai.pytorch.models.nlam.HiLAM method), [1], [2] make_up_gnns() (mfai.pytorch.models.nlam.HiLAM method), [1], [2] manual_backward() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) MeanDimensionSubsampler (class in mfai.pytorch.transforms), [1] merge_state() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) mesh2grid() (in module mfai.pytorch.models.nlam.create_mesh) mesh_aggr (mfai.pytorch.models.nlam.GraphLamSettings attribute), [1] mesh_down_step() (mfai.pytorch.models.nlam.HiLAM method), [1], [2] mesh_up_step() (mfai.pytorch.models.nlam.HiLAM method), [1], [2] message() (mfai.pytorch.models.nlam.interaction_net.InteractionNet method) metric_state (mfai.pytorch.metrics.CSINeighborhood property) (mfai.pytorch.metrics.FAR property) (mfai.pytorch.metrics.FNR property) (mfai.pytorch.metrics.FSS property) (mfai.pytorch.metrics.PR_AUC property) mfai module, [1] mfai.encoding module mfai.http module mfai.pytorch module, [1] mfai.pytorch.callbacks module mfai.pytorch.dummy_dataset module mfai.pytorch.lightning_modules module, [1] mfai.pytorch.lightning_modules.clip module mfai.pytorch.lightning_modules.gan_dgmr module mfai.pytorch.lightning_modules.segmentation module mfai.pytorch.losses module, [1] mfai.pytorch.losses.gan_dgmr module mfai.pytorch.losses.perceptual module mfai.pytorch.losses.toolbelt module mfai.pytorch.lr_scheduler module mfai.pytorch.metrics module mfai.pytorch.models module, [1] mfai.pytorch.models.archesweather module mfai.pytorch.models.base module mfai.pytorch.models.clip module mfai.pytorch.models.deeplabv3 module mfai.pytorch.models.gan_dgmr module, [1] mfai.pytorch.models.gan_dgmr.blocks module mfai.pytorch.models.gan_dgmr.discriminators module mfai.pytorch.models.gan_dgmr.generators module mfai.pytorch.models.gan_dgmr.layers module, [1] mfai.pytorch.models.gan_dgmr.layers.attention module mfai.pytorch.models.gan_dgmr.layers.conv_gru module mfai.pytorch.models.gan_dgmr.layers.coord_conv module mfai.pytorch.models.half_unet module mfai.pytorch.models.identity module mfai.pytorch.models.llms module, [1] mfai.pytorch.models.llms.cross_attention module mfai.pytorch.models.llms.fuyu module mfai.pytorch.models.llms.gpt2 module mfai.pytorch.models.llms.llama2 module mfai.pytorch.models.llms.llama3 module mfai.pytorch.models.nlam module, [1] mfai.pytorch.models.nlam.create_mesh module mfai.pytorch.models.nlam.interaction_net module mfai.pytorch.models.pangu module mfai.pytorch.models.resnet module mfai.pytorch.models.segformer module mfai.pytorch.models.swinunetr module mfai.pytorch.models.unet module mfai.pytorch.models.unetrpp module mfai.pytorch.models.utils module mfai.pytorch.models.vit module mfai.pytorch.models.weather_projector module mfai.pytorch.namedtensor module mfai.pytorch.padding module mfai.pytorch.transforms module mfai.tensorflow module mfai.tokenizers module, [1] MiniGPT2Tokenizer (class in mfai.tokenizers), [1] MiT (class in mfai.pytorch.models.segformer) MixFeedForward (class in mfai.pytorch.models.segformer) mk_2d_graph() (in module mfai.pytorch.models.nlam.create_mesh) MLFlowSaveConfigCallback (class in mfai.pytorch.callbacks), [1] MLFlowSystemMonitorCallback (class in mfai.pytorch.callbacks), [1] MLP (class in mfai.pytorch.models.pangu) mlp_blueprint_end (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1] (mfai.pytorch.models.nlam.GraphLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1] mlp_dim (mfai.pytorch.models.vit.ViTEncoderSettings attribute) mlp_output (mfai.pytorch.models.resnet.ResNet50MLMSettings attribute) model_size (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) model_type (mfai.pytorch.models.archesweather.ArchesWeather attribute), [1] (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLM attribute) (mfai.pytorch.models.llms.fuyu.Fuyu attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2 attribute) (mfai.pytorch.models.llms.gpt2.GPT2 attribute) (mfai.pytorch.models.llms.llama2.Llama2 attribute) (mfai.pytorch.models.llms.llama3.Llama3 attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute), [1] (mfai.pytorch.models.segformer.Segformer attribute), [1] (mfai.pytorch.models.swinunetr.SwinUNetR attribute), [1] (mfai.pytorch.models.unet.CustomUNet attribute), [1] (mfai.pytorch.models.unet.UNet attribute), [1] (mfai.pytorch.models.unetrpp.UNetRPP attribute), [1] (mfai.pytorch.models.vit.ViTClassifier attribute), [1] (mfai.pytorch.models.vit.VitEncoder attribute), [1] ModelABC (class in mfai.pytorch.models.base), [1] ModelType (class in mfai.pytorch.models.base) module mfai, [1] mfai.encoding mfai.http mfai.pytorch, [1] mfai.pytorch.callbacks mfai.pytorch.dummy_dataset mfai.pytorch.lightning_modules, [1] mfai.pytorch.lightning_modules.clip mfai.pytorch.lightning_modules.gan_dgmr mfai.pytorch.lightning_modules.segmentation mfai.pytorch.losses, [1] mfai.pytorch.losses.gan_dgmr mfai.pytorch.losses.perceptual mfai.pytorch.losses.toolbelt mfai.pytorch.lr_scheduler mfai.pytorch.metrics mfai.pytorch.models, [1] mfai.pytorch.models.archesweather mfai.pytorch.models.base mfai.pytorch.models.clip mfai.pytorch.models.deeplabv3 mfai.pytorch.models.gan_dgmr, [1] mfai.pytorch.models.gan_dgmr.blocks mfai.pytorch.models.gan_dgmr.discriminators mfai.pytorch.models.gan_dgmr.generators mfai.pytorch.models.gan_dgmr.layers, [1] mfai.pytorch.models.gan_dgmr.layers.attention mfai.pytorch.models.gan_dgmr.layers.conv_gru mfai.pytorch.models.gan_dgmr.layers.coord_conv mfai.pytorch.models.half_unet mfai.pytorch.models.identity mfai.pytorch.models.llms, [1] mfai.pytorch.models.llms.cross_attention mfai.pytorch.models.llms.fuyu mfai.pytorch.models.llms.gpt2 mfai.pytorch.models.llms.llama2 mfai.pytorch.models.llms.llama3 mfai.pytorch.models.nlam, [1] mfai.pytorch.models.nlam.create_mesh mfai.pytorch.models.nlam.interaction_net mfai.pytorch.models.pangu mfai.pytorch.models.resnet mfai.pytorch.models.segformer mfai.pytorch.models.swinunetr mfai.pytorch.models.unet mfai.pytorch.models.unetrpp mfai.pytorch.models.utils mfai.pytorch.models.vit mfai.pytorch.models.weather_projector mfai.pytorch.namedtensor mfai.pytorch.padding mfai.pytorch.transforms mfai.tensorflow mfai.tokenizers, [1] modules() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) monai_kwargs() (mfai.pytorch.models.swinunetr.SwinUNetRSettings method) monolevel_mesh() (in module mfai.pytorch.models.nlam.create_mesh) mtia() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) MultiHeadAttention (class in mfai.pytorch.models.llms.gpt2) MultiHeadAttentionPySDPA (class in mfai.pytorch.models.llms.gpt2) MultiHeadAttentionPySDPALlama2 (class in mfai.pytorch.models.llms.llama2) MultiHeadCrossAttentionPySDPA (class in mfai.pytorch.models.llms.gpt2) MULTIMODAL_LLM (mfai.pytorch.models.base.ModelType attribute) N n_heads (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) (mfai.pytorch.models.llms.llama2.Llama2Settings attribute) (mfai.pytorch.models.llms.llama3.Llama3Settings attribute) (mfai.pytorch.models.vit.ViTEncoderSettings attribute) n_layers (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) (mfai.pytorch.models.llms.llama2.Llama2Settings attribute) (mfai.pytorch.models.llms.llama3.Llama3Settings attribute) (mfai.pytorch.models.vit.ViTEncoderSettings attribute) name() (mfai.tokenizers.GPT2Tokenizer method), [1] (mfai.tokenizers.LlamaTokenizer method), [1] (mfai.tokenizers.MiniGPT2Tokenizer method), [1] (mfai.tokenizers.Tokenizer method), [1] named_buffers() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) named_children() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) named_modules() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) named_parameters() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) NamedTensor (class in mfai.pytorch.namedtensor) ndims (mfai.pytorch.namedtensor.NamedTensor property) new_like() (mfai.pytorch.namedtensor.NamedTensor static method) no_weight_decay() (mfai.pytorch.models.unetrpp.EPA method) norm_name (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) normalize (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) normalize_activation() (in module mfai.pytorch.losses.perceptual) num_downsampling_chans (mfai.pytorch.models.segformer.SegformerSettings attribute) num_filters (mfai.pytorch.models.half_unet.HalfUNetSettings attribute) num_heads (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) num_heads_decoder (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) num_heads_encoder (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) num_kv_groups (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.llama3.Llama3Settings attribute) num_layers (mfai.pytorch.models.segformer.SegformerSettings attribute) num_spatial_dims (mfai.pytorch.models.archesweather.ArchesWeather property), [1] (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather property), [1] (mfai.pytorch.models.segformer.Segformer attribute), [1] (mfai.pytorch.models.swinunetr.SwinUNetR attribute), [1] (mfai.pytorch.models.unet.CustomUNet attribute), [1] (mfai.pytorch.models.unet.UNet attribute), [1] (mfai.pytorch.models.unetrpp.UNetRPP property), [1] (mfai.pytorch.models.vit.ViTClassifier attribute), [1] (mfai.pytorch.models.vit.VitEncoder attribute), [1] (mfai.pytorch.namedtensor.NamedTensor property) num_tokens (mfai.pytorch.models.resnet.ResNet50MLMSettings attribute) O offload_to_cpu (mfai.pytorch.models.nlam.GraphLamSettings attribute), [1] offload_to_cpu() (in module mfai.pytorch.models.nlam), [1] on_after_backward() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_after_batch_transfer() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_before_backward() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_before_batch_transfer() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_before_optimizer_step() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_before_zero_grad() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_exception() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) on_fit_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method), [1] (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_fit_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method), [1] (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_gpu (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) on_load_checkpoint() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_predict_batch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_predict_batch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_predict_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_predict_epoch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_predict_epoch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_predict_model_eval() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_predict_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_sanity_check_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) on_sanity_check_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) on_save_checkpoint() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_test_batch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_test_batch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_test_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_test_epoch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] on_test_epoch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_test_model_eval() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_test_model_train() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_test_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] on_train_batch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_train_batch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_train_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_train_epoch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] on_train_epoch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_train_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] on_validation_batch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_validation_batch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_validation_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_validation_epoch_end() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method), [1] (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] on_validation_epoch_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_validation_model_eval() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_validation_model_train() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_validation_model_zero_grad() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) on_validation_start() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] onnx_load_and_infer() (in module mfai.pytorch), [1] onnx_supported (mfai.pytorch.models.archesweather.ArchesWeather attribute), [1] (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute), [1] (mfai.pytorch.models.segformer.Segformer attribute), [1] (mfai.pytorch.models.swinunetr.SwinUNetR attribute), [1] (mfai.pytorch.models.unet.CustomUNet attribute), [1] (mfai.pytorch.models.unet.UNet attribute), [1] (mfai.pytorch.models.unetrpp.UNetRPP attribute), [1] (mfai.pytorch.models.vit.ViTClassifier attribute), [1] (mfai.pytorch.models.vit.VitEncoder attribute), [1] optimizer_step() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) optimizer_zero_grad() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) optimizers() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) out_channels (mfai.pytorch.models.archesweather.ArchesWeather attribute) (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.base.ModelABC attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute) (mfai.pytorch.models.identity.IdentityModel attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1] (mfai.pytorch.models.nlam.GraphLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1] (mfai.pytorch.models.pangu.PanguWeather attribute) (mfai.pytorch.models.resnet.ResNetEncoder property) (mfai.pytorch.models.segformer.Segformer attribute) (mfai.pytorch.models.swinunetr.SwinUNetR attribute) (mfai.pytorch.models.unet.CustomUNet attribute) (mfai.pytorch.models.unet.UNet attribute) (mfai.pytorch.models.unetrpp.UNetRPP attribute) (mfai.pytorch.models.vit.ViTClassifier attribute) (mfai.pytorch.models.vit.VitEncoder attribute) output_stride (mfai.pytorch.models.resnet.ResNetEncoder property) P pad_batch() (in module mfai.pytorch.padding) pair() (in module mfai.pytorch.models.vit) PANGU (mfai.pytorch.models.base.ModelType attribute) PanguWeather (class in mfai.pytorch.models.pangu), [1] PanguWeatherSettings (class in mfai.pytorch.models.pangu) parameters() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) patch_first_conv() (in module mfai.pytorch.models.utils) patch_size (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.vit.ViTEncoderSettings attribute) (mfai.pytorch.models.weather_projector.WeatherProjectorSettings attribute) PatchEmbedding (class in mfai.pytorch.models.pangu) PatchMaker (class in mfai.pytorch.models.weather_projector) PatchRecovery (class in mfai.pytorch.models.pangu) patchrecovery (mfai.pytorch.models.archesweather.ArchesWeather attribute) PatchRecoveryConv (class in mfai.pytorch.models.archesweather) PerceptualLoss (class in mfai.pytorch.losses.perceptual), [1] persistent() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) pin_memory_() (mfai.pytorch.namedtensor.NamedTensor method) plevel_patch_size (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) plevel_variables (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) plevels (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) plot() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) plot_graph() (in module mfai.pytorch.models.nlam.create_mesh) plot_legend_name (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) plot_lower_bound (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) plot_probabilities_matrix() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.CLIPLightningModule method), [1] plot_upper_bound (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) pool (mfai.pytorch.models.vit.ViTClassifierSettings attribute) pop() (mfai.pytorch.losses.perceptual.LinLayers method) pos_embed (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) pos_embedding (mfai.pytorch.models.resnet.ResNet50MLMSettings attribute) position_embs_dim (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) PR_AUC (class in mfai.pytorch.metrics), [1] precompute_rope_params() (in module mfai.pytorch.models.llms.llama2) predict() (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) predict_dataloader() (mfai.pytorch.dummy_dataset.DummyDataModule method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) predict_step() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) PreNorm (class in mfai.pytorch.models.segformer) prepare_data() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) prepare_data_per_node (mfai.pytorch.lightning_modules.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.SegmentationLightningModule attribute) prepend_node_index() (in module mfai.pytorch.models.nlam.create_mesh) print() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) probabilities_to_classes() (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] process_step() (mfai.pytorch.models.nlam.BaseGraphModel method), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel method), [1], [2] (mfai.pytorch.models.nlam.GraphLAM method), [1], [2] (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) processor_layers (mfai.pytorch.models.nlam.GraphLamSettings attribute), [1] proj_feat() (mfai.pytorch.models.unetrpp.UNetRPP method), [1] Q qkv_bias (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) (mfai.pytorch.models.llms.gpt2.GPT2Settings attribute) R RandomCropWithMinPositivePixels (class in mfai.pytorch.transforms), [1] rank_zero_setup() (mfai.pytorch.models.nlam.BaseGraphModel class method), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel class method) (mfai.pytorch.models.nlam.GraphLAM class method) (mfai.pytorch.models.nlam.HiLAM class method) (mfai.pytorch.models.nlam.HiLAMParallel class method) rearrange_() (mfai.pytorch.namedtensor.NamedTensor method) reduction (mfai.pytorch.losses.toolbelt.DiceLoss attribute) register (mfai.pytorch.models.archesweather.ArchesWeather attribute), [1] (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.base.ModelABC attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.nlam.BaseGraphModel attribute) (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute), [1], [2] (mfai.pytorch.models.nlam.HiLAM attribute), [1], [2] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1], [2] (mfai.pytorch.models.pangu.PanguWeather attribute), [1] (mfai.pytorch.models.segformer.Segformer attribute), [1] (mfai.pytorch.models.swinunetr.SwinUNetR attribute), [1] (mfai.pytorch.models.unet.CustomUNet attribute), [1] (mfai.pytorch.models.unet.UNet attribute), [1] (mfai.pytorch.models.unetrpp.UNetRPP attribute), [1] (mfai.pytorch.models.vit.ViTClassifier attribute), [1] (mfai.pytorch.models.vit.VitEncoder attribute), [1] register_backward_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_buffer() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_forward_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_forward_pre_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_full_backward_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_full_backward_pre_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_load_state_dict_post_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_load_state_dict_pre_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_module() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_parameter() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_state_dict_post_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) register_state_dict_pre_hook() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) remove_ignored_hparams() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) replace_strides_with_dilation() (in module mfai.pytorch.models.utils) requires_grad_() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) reset() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) ResNet50 (class in mfai.pytorch.models.resnet) ResNet50MLM (class in mfai.pytorch.models.resnet) ResNet50MLMSettings (class in mfai.pytorch.models.resnet) ResNet50Settings (class in mfai.pytorch.models.resnet) resnet_mlp_output (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) resnet_num_tokens (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) resnet_pos_embedding (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) ResNetEncoder (class in mfai.pytorch.models.resnet) RMSNorm (class in mfai.pytorch.models.llms.llama2) rope_base (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) (mfai.pytorch.models.llms.llama3.Llama3Settings attribute) S Sampler (class in mfai.pytorch.models.gan_dgmr.generators) save_config() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method), [1] save_edges() (in module mfai.pytorch.models.nlam.create_mesh) save_edges_list() (in module mfai.pytorch.models.nlam.create_mesh) save_hyperparameters() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) save_vision_encoder() (mfai.pytorch.models.clip.Clip method) SaveCLIPVisualEncoderWeights (class in mfai.pytorch.lightning_modules.clip), [1] schema() (mfai.pytorch.models.archesweather.ArchesWeatherSettings class method) (mfai.pytorch.models.clip.ClipSettings class method) (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings class method) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings class method) (mfai.pytorch.models.half_unet.HalfUNetSettings class method) (mfai.pytorch.models.identity.IdentityModelSettings class method) (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings class method) (mfai.pytorch.models.llms.fuyu.FuyuSettings class method) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings class method) (mfai.pytorch.models.llms.gpt2.GPT2Settings class method) (mfai.pytorch.models.llms.llama2.Llama2Settings class method) (mfai.pytorch.models.llms.llama3.Llama3Settings class method) (mfai.pytorch.models.nlam.GraphLamSettings class method), [1] (mfai.pytorch.models.pangu.PanguWeatherSettings class method) (mfai.pytorch.models.resnet.ResNet50MLMSettings class method) (mfai.pytorch.models.resnet.ResNet50Settings class method) (mfai.pytorch.models.segformer.SegformerSettings class method) (mfai.pytorch.models.swinunetr.SwinUNetRSettings class method) (mfai.pytorch.models.unet.CustomUNetSettings class method) (mfai.pytorch.models.unet.UNetSettings class method) (mfai.pytorch.models.unetrpp.UNetRPPSettings class method) (mfai.pytorch.models.vit.ViTClassifierSettings class method) (mfai.pytorch.models.vit.ViTEncoderSettings class method) (mfai.pytorch.models.weather_projector.WeatherProjectorSettings class method) Segformer (class in mfai.pytorch.models.segformer), [1] SegformerSettings (class in mfai.pytorch.models.segformer) SegmentationLightningModule (class in mfai.pytorch.lightning_modules), [1] (class in mfai.pytorch.lightning_modules.segmentation), [1] select_dim() (mfai.pytorch.namedtensor.NamedTensor method) select_tensor_dim() (mfai.pytorch.namedtensor.NamedTensor method) SeparableConv2d (class in mfai.pytorch.models.deeplabv3) set_dtype() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) set_extra_state() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) set_in_channels() (mfai.pytorch.models.resnet.ResNetEncoder method) set_requires_grad() (mfai.pytorch.losses.perceptual.BaseNet method), [1] (mfai.pytorch.losses.perceptual.VGG16 method) set_submodule() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) settings (mfai.pytorch.models.archesweather.ArchesWeather property), [1] (mfai.pytorch.models.base.AutoPaddingModel property) (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus property), [1] (mfai.pytorch.models.half_unet.HalfUNet property), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.nlam.BaseGraphModel property), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel property) (mfai.pytorch.models.nlam.GraphLAM property) (mfai.pytorch.models.nlam.HiLAM property) (mfai.pytorch.models.nlam.HiLAMParallel property) (mfai.pytorch.models.pangu.PanguWeather property), [1] (mfai.pytorch.models.segformer.Segformer property), [1] (mfai.pytorch.models.swinunetr.SwinUNetR property), [1] (mfai.pytorch.models.unet.CustomUNet property), [1] (mfai.pytorch.models.unet.UNet property), [1] (mfai.pytorch.models.unetrpp.UNetRPP property), [1] (mfai.pytorch.models.vit.ViTClassifier property), [1] (mfai.pytorch.models.vit.VitEncoder property), [1] settings_kls (mfai.pytorch.models.archesweather.ArchesWeather attribute), [1] (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute), [1] (mfai.pytorch.models.half_unet.HalfUNet attribute), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLM attribute) (mfai.pytorch.models.llms.fuyu.Fuyu attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2 attribute) (mfai.pytorch.models.llms.gpt2.GPT2 attribute) (mfai.pytorch.models.llms.llama2.Llama2 attribute) (mfai.pytorch.models.llms.llama3.Llama3 attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute), [1] (mfai.pytorch.models.resnet.ResNet50 attribute) (mfai.pytorch.models.resnet.ResNet50MLM attribute) (mfai.pytorch.models.segformer.Segformer attribute), [1] (mfai.pytorch.models.swinunetr.SwinUNetR attribute), [1] (mfai.pytorch.models.unet.CustomUNet attribute), [1] (mfai.pytorch.models.unet.UNet attribute), [1] (mfai.pytorch.models.unetrpp.UNetRPP attribute), [1] (mfai.pytorch.models.vit.ViTClassifier attribute), [1] (mfai.pytorch.models.vit.VitEncoder attribute), [1] setup() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.dummy_dataset.DummyDataModule method) (mfai.pytorch.dummy_dataset.DummyMultiModalDataModule method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) share_memory() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) SiLU (class in mfai.pytorch.models.llms.llama2) sin (mfai.pytorch.models.llms.llama2.MultiHeadAttentionPySDPALlama2 attribute) SoftBCEWithLogitsLoss (class in mfai.pytorch.losses.toolbelt), [1] SoftCrossEntropyLoss (class in mfai.pytorch.losses.toolbelt), [1] sort_nodes_internally() (in module mfai.pytorch.models.nlam.create_mesh) spatial_dim_idx (mfai.pytorch.namedtensor.NamedTensor property) SPATIAL_DIM_NAMES (mfai.pytorch.namedtensor.NamedTensor attribute) spatial_dims (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) (mfai.pytorch.models.unetrpp.UNetRPPSettings attribute) SpatialDiscriminator (class in mfai.pytorch.models.gan_dgmr.discriminators) SplitMLPs (class in mfai.pytorch.models.nlam.interaction_net) squeeze_() (mfai.pytorch.namedtensor.NamedTensor method) stack() (mfai.pytorch.namedtensor.NamedTensor static method) state_dict() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) state_key (mfai.pytorch.callbacks.MLFlowSaveConfigCallback property) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback property) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights property) static_length (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) strict_loading (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) supported_num_spatial_dims (mfai.pytorch.models.archesweather.ArchesWeather attribute), [1] (mfai.pytorch.models.base.BaseModel property) (mfai.pytorch.models.base.ModelABC property), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute), [1] (mfai.pytorch.models.identity.IdentityModel attribute), [1] (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1], [2] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute), [1] (mfai.pytorch.models.segformer.Segformer attribute), [1] (mfai.pytorch.models.swinunetr.SwinUNetR attribute), [1] (mfai.pytorch.models.unet.CustomUNet attribute), [1] (mfai.pytorch.models.unet.UNet attribute), [1] (mfai.pytorch.models.unetrpp.UNetRPP attribute), [1] (mfai.pytorch.models.vit.ViTClassifier attribute), [1] (mfai.pytorch.models.vit.VitEncoder attribute), [1] surface_variables (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) SwinUNetR (class in mfai.pytorch.models.swinunetr), [1] SwinUNetRSettings (class in mfai.pytorch.models.swinunetr) sync() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) sync_context() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) T T_destination (mfai.pytorch.lightning_modules.clip.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule attribute) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule attribute) (mfai.pytorch.losses.gan_dgmr.GridCellLoss attribute) (mfai.pytorch.losses.perceptual.BaseNet attribute) (mfai.pytorch.losses.perceptual.LinLayers attribute) (mfai.pytorch.losses.perceptual.LPIPS attribute) (mfai.pytorch.losses.perceptual.PerceptualLoss attribute) (mfai.pytorch.losses.perceptual.VGG16 attribute) (mfai.pytorch.losses.toolbelt.DiceLoss attribute) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss attribute) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss attribute) (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) (mfai.pytorch.models.archesweather.ArchesWeather attribute) (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute) (mfai.pytorch.models.identity.IdentityModel attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute) (mfai.pytorch.models.nlam.BaseHiGraphModel attribute) (mfai.pytorch.models.nlam.GraphLAM attribute) (mfai.pytorch.models.nlam.HiLAM attribute) (mfai.pytorch.models.nlam.HiLAMParallel attribute) (mfai.pytorch.models.pangu.PanguWeather attribute) (mfai.pytorch.models.segformer.Segformer attribute) (mfai.pytorch.models.swinunetr.SwinUNetR attribute) (mfai.pytorch.models.unet.CustomUNet attribute) (mfai.pytorch.models.unet.UNet attribute) (mfai.pytorch.models.unetrpp.UNetRPP attribute) (mfai.pytorch.models.vit.ViTClassifier attribute) (mfai.pytorch.models.vit.VitEncoder attribute) (mfai.pytorch.transforms.DimensionSubSampler attribute) (mfai.pytorch.transforms.MeanDimensionSubsampler attribute) teardown() (mfai.pytorch.callbacks.MLFlowSaveConfigCallback method) (mfai.pytorch.callbacks.MLFlowSystemMonitorCallback method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) TemporalDiscriminator (class in mfai.pytorch.models.gan_dgmr.discriminators) tensor (mfai.pytorch.namedtensor.NamedTensor attribute) (mfai.pytorch.namedtensor.TensorWrapper attribute) TensorWrapper (class in mfai.pytorch.namedtensor) test_dataloader() (mfai.pytorch.dummy_dataset.DummyDataModule method) (mfai.pytorch.dummy_dataset.DummyMultiModalDataModule method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) test_step() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] text_encoder (mfai.pytorch.models.clip.ClipSettings attribute) tmp_dir (mfai.pytorch.models.nlam.GraphLamSettings attribute), [1] to() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) to_() (mfai.pytorch.namedtensor.NamedTensor method) to_category() (mfai.pytorch.metrics.FSS static method), [1] to_dict() (mfai.pytorch.models.archesweather.ArchesWeatherSettings method) (mfai.pytorch.models.clip.ClipSettings method) (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings method) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings method) (mfai.pytorch.models.half_unet.HalfUNetSettings method) (mfai.pytorch.models.identity.IdentityModelSettings method) (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings method) (mfai.pytorch.models.llms.fuyu.FuyuSettings method) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings method) (mfai.pytorch.models.llms.gpt2.GPT2Settings method) (mfai.pytorch.models.llms.llama2.Llama2Settings method) (mfai.pytorch.models.llms.llama3.Llama3Settings method) (mfai.pytorch.models.nlam.GraphLamSettings method), [1] (mfai.pytorch.models.pangu.PanguWeatherSettings method) (mfai.pytorch.models.resnet.ResNet50MLMSettings method) (mfai.pytorch.models.resnet.ResNet50Settings method) (mfai.pytorch.models.segformer.SegformerSettings method) (mfai.pytorch.models.swinunetr.SwinUNetRSettings method) (mfai.pytorch.models.unet.CustomUNetSettings method) (mfai.pytorch.models.unet.UNetSettings method) (mfai.pytorch.models.unetrpp.UNetRPPSettings method) (mfai.pytorch.models.vit.ViTClassifierSettings method) (mfai.pytorch.models.vit.ViTEncoderSettings method) (mfai.pytorch.models.weather_projector.WeatherProjectorSettings method) to_empty() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) to_json() (mfai.pytorch.models.archesweather.ArchesWeatherSettings method) (mfai.pytorch.models.clip.ClipSettings method) (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings method) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings method) (mfai.pytorch.models.half_unet.HalfUNetSettings method) (mfai.pytorch.models.identity.IdentityModelSettings method) (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings method) (mfai.pytorch.models.llms.fuyu.FuyuSettings method) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings method) (mfai.pytorch.models.llms.gpt2.GPT2Settings method) (mfai.pytorch.models.llms.llama2.Llama2Settings method) (mfai.pytorch.models.llms.llama3.Llama3Settings method) (mfai.pytorch.models.nlam.GraphLamSettings method), [1] (mfai.pytorch.models.pangu.PanguWeatherSettings method) (mfai.pytorch.models.resnet.ResNet50MLMSettings method) (mfai.pytorch.models.resnet.ResNet50Settings method) (mfai.pytorch.models.segformer.SegformerSettings method) (mfai.pytorch.models.swinunetr.SwinUNetRSettings method) (mfai.pytorch.models.unet.CustomUNetSettings method) (mfai.pytorch.models.unet.UNetSettings method) (mfai.pytorch.models.unetrpp.UNetRPPSettings method) (mfai.pytorch.models.vit.ViTClassifierSettings method) (mfai.pytorch.models.vit.ViTEncoderSettings method) (mfai.pytorch.models.weather_projector.WeatherProjectorSettings method) to_numpy() (in module mfai.pytorch), [1] to_onnx() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) to_tensorrt() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) to_torchscript() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) toggle_optimizer() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) toggled_optimizer() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) token_size (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) Tokenizer (class in mfai.tokenizers), [1] tokens() (mfai.tokenizers.MiniGPT2Tokenizer method), [1] torch_save() (in module mfai.pytorch.models.nlam.create_mesh) train() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) train_dataloader() (mfai.pytorch.dummy_dataset.DummyDataModule method) (mfai.pytorch.dummy_dataset.DummyMultiModalDataModule method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) trainer (mfai.pytorch.lightning_modules.clip.CLIPLightningModule property) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule property) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule property) training (mfai.pytorch.lightning_modules.clip.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.CLIPLightningModule attribute) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule attribute) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule attribute) (mfai.pytorch.lightning_modules.SegmentationLightningModule attribute) (mfai.pytorch.losses.gan_dgmr.GridCellLoss attribute) (mfai.pytorch.losses.perceptual.BaseNet attribute) (mfai.pytorch.losses.perceptual.LinLayers attribute) (mfai.pytorch.losses.perceptual.LPIPS attribute) (mfai.pytorch.losses.perceptual.PerceptualLoss attribute) (mfai.pytorch.losses.perceptual.VGG16 attribute) (mfai.pytorch.losses.toolbelt.DiceLoss attribute) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss attribute) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss attribute) (mfai.pytorch.metrics.CSINeighborhood attribute) (mfai.pytorch.metrics.FAR attribute) (mfai.pytorch.metrics.FNR attribute) (mfai.pytorch.metrics.FSS attribute) (mfai.pytorch.metrics.PR_AUC attribute) (mfai.pytorch.models.archesweather.ArchesWeather attribute) (mfai.pytorch.models.base.BaseModel attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3 attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus attribute) (mfai.pytorch.models.half_unet.HalfUNet attribute) (mfai.pytorch.models.identity.IdentityModel attribute) (mfai.pytorch.models.nlam.BaseGraphModel attribute), [1] (mfai.pytorch.models.nlam.BaseHiGraphModel attribute), [1] (mfai.pytorch.models.nlam.GraphLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAM attribute), [1] (mfai.pytorch.models.nlam.HiLAMParallel attribute), [1] (mfai.pytorch.models.pangu.PanguWeather attribute) (mfai.pytorch.models.segformer.Segformer attribute) (mfai.pytorch.models.swinunetr.SwinUNetR attribute) (mfai.pytorch.models.unet.CustomUNet attribute) (mfai.pytorch.models.unet.UNet attribute) (mfai.pytorch.models.unetrpp.UNetRPP attribute) (mfai.pytorch.models.vit.ViTClassifier attribute) (mfai.pytorch.models.vit.VitEncoder attribute) (mfai.pytorch.transforms.DimensionSubSampler attribute) (mfai.pytorch.transforms.MeanDimensionSubsampler attribute) training_step() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method), [1] (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] transfer_batch_to_device() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) Transformer (class in mfai.pytorch.models.vit) transformer_dropout (mfai.pytorch.models.vit.ViTEncoderSettings attribute) TransformerBlock (class in mfai.pytorch.models.llms.gpt2) (class in mfai.pytorch.models.llms.llama3) (class in mfai.pytorch.models.unetrpp) TransformerBlockLlama2 (class in mfai.pytorch.models.llms.llama2) trunc_normal_() (in module mfai.pytorch.models.unetrpp) type() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) type_() (mfai.pytorch.namedtensor.NamedTensor method) U undo_padding() (in module mfai.pytorch.padding) UNet (class in mfai.pytorch.models.unet), [1] UNetRPP (class in mfai.pytorch.models.unetrpp), [1] UNetRPPEncoder (class in mfai.pytorch.models.unetrpp) UNetRPPSettings (class in mfai.pytorch.models.unetrpp) UNetRUpBlock (class in mfai.pytorch.models.unetrpp) UNetSettings (class in mfai.pytorch.models.unet) unflatten_() (mfai.pytorch.namedtensor.NamedTensor method) unfreeze() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) unfreeze_llm() (mfai.pytorch.models.llms.FreezeMLMMixin method), [1] unfreeze_vision() (mfai.pytorch.models.llms.FreezeMLMMixin method), [1] unsqueeze_() (mfai.pytorch.namedtensor.NamedTensor method) unsqueeze_and_expand_from_() (mfai.pytorch.namedtensor.NamedTensor method) unsync() (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) untoggle_optimizer() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) update() (mfai.pytorch.lightning_modules.clip.CLIPAccuracySkillScore method) (mfai.pytorch.metrics.CSINeighborhood method), [1] (mfai.pytorch.metrics.FAR method), [1] (mfai.pytorch.metrics.FNR method), [1] (mfai.pytorch.metrics.FSS method), [1] (mfai.pytorch.metrics.PR_AUC method), [1] update_called (mfai.pytorch.metrics.CSINeighborhood property) (mfai.pytorch.metrics.FAR property) (mfai.pytorch.metrics.FNR property) (mfai.pytorch.metrics.FSS property) (mfai.pytorch.metrics.PR_AUC property) update_count (mfai.pytorch.metrics.CSINeighborhood property) (mfai.pytorch.metrics.FAR property) (mfai.pytorch.metrics.FNR property) (mfai.pytorch.metrics.FSS property) (mfai.pytorch.metrics.PR_AUC property) UpSample (class in mfai.pytorch.models.pangu) UpsampleBlock (class in mfai.pytorch.models.swinunetr) UpsampleGBlock (class in mfai.pytorch.models.gan_dgmr.blocks) upsampling (mfai.pytorch.models.deeplabv3.DeepLabV3PlusSettings attribute) (mfai.pytorch.models.deeplabv3.DeepLabV3Settings attribute) use_checkpoint (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) use_checkpointing (mfai.pytorch.models.nlam.GraphLamSettings attribute), [1] use_ghost (mfai.pytorch.models.half_unet.HalfUNetSettings attribute) use_prev (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) use_skip (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) use_v2 (mfai.pytorch.models.swinunetr.SwinUNetRSettings attribute) V val_dataloader() (mfai.pytorch.dummy_dataset.DummyDataModule method) (mfai.pytorch.dummy_dataset.DummyMultiModalDataModule method) (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) val_plot_step() (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] validate_input_shape() (mfai.pytorch.models.base.AutoPaddingModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method), [1] (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method), [1] (mfai.pytorch.models.segformer.Segformer method), [1] (mfai.pytorch.models.swinunetr.SwinUNetR method), [1] (mfai.pytorch.models.unet.CustomUNet method), [1] (mfai.pytorch.models.unet.UNet method), [1] (mfai.pytorch.models.unetrpp.UNetRPP method), [1] (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.models.vit.VitPaddingMixin method) validation_step() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.CLIPLightningModule method), [1] (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method), [1] (mfai.pytorch.lightning_modules.SegmentationLightningModule method), [1] VGG16 (class in mfai.pytorch.losses.perceptual), [1] vision_encoder (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.FreezeMLMMixin attribute), [1] (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) vision_input_shape (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.fuyu.FuyuSettings attribute) VISION_TRANSFORMER (mfai.pytorch.models.base.ModelType attribute) ViTClassifier (class in mfai.pytorch.models.vit), [1] ViTClassifierSettings (class in mfai.pytorch.models.vit) ViTCore (class in mfai.pytorch.models.vit) VitEncoder (class in mfai.pytorch.models.vit), [1] ViTEncoderSettings (class in mfai.pytorch.models.vit) VitPaddingMixin (class in mfai.pytorch.models.vit) vocab_size (mfai.tokenizers.GPT2Tokenizer property), [1] (mfai.tokenizers.LlamaTokenizer property), [1] (mfai.tokenizers.MiniGPT2Tokenizer property), [1] (mfai.tokenizers.Tokenizer property), [1] W WeatherProjector (class in mfai.pytorch.models.weather_projector) WeatherProjectorSettings (class in mfai.pytorch.models.weather_projector) weight_fn() (mfai.pytorch.losses.gan_dgmr.GridCellLoss method), [1] window_size (mfai.pytorch.models.archesweather.ArchesWeatherSettings attribute) (mfai.pytorch.models.pangu.PanguWeatherSettings attribute) X x_att_ratio (mfai.pytorch.models.llms.cross_attention.XAttMultiModalLMSettings attribute) (mfai.pytorch.models.llms.gpt2.CrossAttentionGPT2Settings attribute) XAttMultiModalLM (class in mfai.pytorch.models.llms.cross_attention) XAttMultiModalLMSettings (class in mfai.pytorch.models.llms.cross_attention) xpu() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method) Z z_score() (mfai.pytorch.losses.perceptual.BaseNet method), [1] (mfai.pytorch.losses.perceptual.VGG16 method) zero_grad() (mfai.pytorch.lightning_modules.clip.CLIPLightningModule method) (mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule method) (mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule method) (mfai.pytorch.losses.gan_dgmr.GridCellLoss method) (mfai.pytorch.losses.perceptual.BaseNet method) (mfai.pytorch.losses.perceptual.LinLayers method) (mfai.pytorch.losses.perceptual.LPIPS method) (mfai.pytorch.losses.perceptual.PerceptualLoss method) (mfai.pytorch.losses.perceptual.VGG16 method) (mfai.pytorch.losses.toolbelt.DiceLoss method) (mfai.pytorch.losses.toolbelt.SoftBCEWithLogitsLoss method) (mfai.pytorch.losses.toolbelt.SoftCrossEntropyLoss method) (mfai.pytorch.metrics.CSINeighborhood method) (mfai.pytorch.metrics.FAR method) (mfai.pytorch.metrics.FNR method) (mfai.pytorch.metrics.FSS method) (mfai.pytorch.metrics.PR_AUC method) (mfai.pytorch.models.archesweather.ArchesWeather method) (mfai.pytorch.models.base.BaseModel method) (mfai.pytorch.models.deeplabv3.DeepLabV3 method) (mfai.pytorch.models.deeplabv3.DeepLabV3Plus method) (mfai.pytorch.models.half_unet.HalfUNet method) (mfai.pytorch.models.identity.IdentityModel method) (mfai.pytorch.models.nlam.BaseGraphModel method) (mfai.pytorch.models.nlam.BaseHiGraphModel method) (mfai.pytorch.models.nlam.GraphLAM method) (mfai.pytorch.models.nlam.HiLAM method) (mfai.pytorch.models.nlam.HiLAMParallel method) (mfai.pytorch.models.pangu.PanguWeather method) (mfai.pytorch.models.segformer.Segformer method) (mfai.pytorch.models.swinunetr.SwinUNetR method) (mfai.pytorch.models.unet.CustomUNet method) (mfai.pytorch.models.unet.UNet method) (mfai.pytorch.models.unetrpp.UNetRPP method) (mfai.pytorch.models.vit.ViTClassifier method) (mfai.pytorch.models.vit.VitEncoder method) (mfai.pytorch.transforms.DimensionSubSampler method) (mfai.pytorch.transforms.MeanDimensionSubsampler method)