Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Y | Z _ __call__() (_transforms.NormalizeState method) (_transforms.NormalizeState_Quantile method) (_transforms.ToTensor method) (credit.ensemble.bred_vector.BredVector method) (credit.ensemble.color.ColorNoise method) (credit.ensemble.gaussian.GaussianNoise method) (credit.ensemble.spherical.SphericalNoise method) (credit.ensemble.temporal.TemporalNoise method) (credit.metrics.LatWeightedMetrics method) (credit.metrics.LatWeightedMetricsClimatology method) (credit.metrics.LatWeightedMetricsEnsemble method) (credit.metrics_downscaling.UnWeightedMetrics method) (credit.replay_buffer.TOADataLoader method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeLES method) (credit.transforms.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.NormalizeWRF method) (credit.transforms.ToTensor_BridgeScaler method) (credit.transforms.ToTensor_ERA5_and_Forcing method) (credit.transforms.ToTensorLES method) (credit.transforms.ToTensorWRF method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_les.ToTensorLES method) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler method) (credit.transforms.transforms_wrf.NormalizeWRF method) (credit.transforms.transforms_wrf.ToTensorWRF method) (credit.transforms_downscaling.Clip method) (credit.transforms_downscaling.DataTransforms method) (credit.transforms_downscaling.Expand method) (credit.transforms_downscaling.Identity method) (credit.transforms_downscaling.Minmax method) (credit.transforms_downscaling.Pad method) (credit.transforms_downscaling.Power method) (credit.transforms_downscaling.Zscore method) (credit.xr_sampler.XRSamplerByYear method) (trainerERA5_v1.TOADataLoader method) __commit_id__ (in module credit._version) __del__() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) __enter__() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) __eq__() (credit.regrid.Grid method) __exit__() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) __getitem__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.datasets.datamap.DataMap method) (credit.datasets.downscaling_dataset.DownscalingDataset method) (credit.datasets.era5.ERA5Dataset method) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.era5_singlestep.ERA5_and_Forcing_SingleStep method) (credit.datasets.les_singlestep.LESDataset method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.realtime_predict.RealtimePredictDataset method) (credit.datasets.wrf_singlestep.WRFDataset method) (credit.datasets.wrfmultistep.WRFMultiStep method) __hash__() (credit.regrid.Grid method) __iter__() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep.RepeatingIndexSampler method) (credit.datasets.les_singlestep.LESPredict method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.sequential_multistep.DistributedSequentialDatasetBasic method) (credit.datasets.wrf_singlestep.WRFPredict method) (credit.datasets.wrfmultistep.RepeatingIndexSampler method) (credit.samplers.DistributedMultiStepBatchSampler method) (credit.samplers.MultiStepBatchSamplerSubset method) (credit.trainers.ic_optimization.TimeStepper method) __len__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.data.Predict_Dataset method) (credit.datasets.datamap.DataMap method) (credit.datasets.downscaling_dataset.DownscalingDataset method) (credit.datasets.era5.ERA5Dataset method) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep.RepeatingIndexSampler method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.les_singlestep.LESDataset method) (credit.datasets.les_singlestep.LESPredict method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) (credit.datasets.realtime_predict.RealtimePredictDataset method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.wrf_singlestep.WRFDataset method) (credit.datasets.wrf_singlestep.WRFPredict method) (credit.datasets.wrfmultistep.RepeatingIndexSampler method) (credit.datasets.wrfmultistep.WRFMultiStep method) (credit.samplers.DistributedMultiStepBatchSampler method) (credit.samplers.MultiStepBatchSamplerSubset method) __next__() (credit.trainers.ic_optimization.TimeStepper method) __post_init__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.datasets.datamap.DataMap method) (credit.datasets.downscaling_dataset.DownscalingDataset method) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.les_singlestep.LESDataset method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.wrf_singlestep.WRFDataset method) (credit.datasets.wrfmultistep.WRFMultiStep method) (credit.output_downscaling.OutputWrangler method) (credit.output_downscaling.OutputWriter method) __repr__() (credit.ensemble.color.ColorNoise method) (credit.ensemble.gaussian.GaussianNoise method) (credit.models.graph.TransformerConv method) __version__ (in module credit._version) __version_tuple__ (in module credit._version) _active (credit.trainers.ic_optimization.TimeStepper attribute) _add_metadata() (credit.datasets.era5.ERA5Dataset method) _align_coords() (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) _align_phase_with() (in module credit.regrid) _alpha (credit.models.graph.TransformerConv attribute) _assert_increasing() (in module credit.regrid) _boundary_data (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) _build_valid_indices() (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) _calc_window_shift() (credit.models.swin.SwinTransformerV2CrBlock method) _capture_enabled (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _collect_noise_layers() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) _conservative_latitude_weights() (in module credit.regrid) _conservative_longitude_weights() (in module credit.regrid) _convert_cf_time() (credit.datasets.era5.ERA5Dataset method) _convert_windows_to_indices() (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) _create_correlated_noise() (credit.ensemble.color.ColorNoise method) _current_latents (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _current_timestep_map (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _do_special_eval_on_variable() (in module ensemble_eval) _do_standard_eval_on_variable() (in module ensemble_eval) _earth_padding() (credit.boundary_padding.TensorPadding method) _earth_unpad() (credit.boundary_padding.TensorPadding method) _fetch_batch() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) _forward_with_latent_control() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) _get_batch_samples() (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) _get_ocean_sample() (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) _initialize() (credit.ocean.samudra_constants.TensorMap method) (credit.ocean.samudra_data.Normalize method) _instance (credit.ocean.samudra_constants.TensorMap attribute) (credit.ocean.samudra_data.Normalize attribute) _kernel_crps() (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss method) _kernel_crps_implementation() (credit.losses.kcrps.KCRPSLoss method) _latitude_cell_bounds() (in module credit.regrid) _latitude_overlap() (in module credit.regrid) _longitude_overlap() (in module credit.regrid) _make_attention_mask() (credit.models.swin.SwinTransformerV2CrBlock method) _make_pair_wise_relative_positions() (credit.models.swin.WindowMultiHeadAttention method) _map_files() (credit.datasets.era5.ERA5Dataset method) _match_spatial() (credit.models.swin_wrf.WRFTransformer method) (credit.models.WRFTransformer method) _mean() (credit.regrid.ConservativeRegridder method) _mirror_padding() (credit.boundary_padding.TensorPadding method) _mirror_unpad() (credit.boundary_padding.TensorPadding method) _mode (credit.datasets.downscaling_dataset.DownscalingDataset attribute) _nanmean() (credit.regrid.ConservativeRegridder method) _nearest_neighbor_2d() (credit.regrid.NearestRegridder method) _noise_layers (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _open_ds_extract_fields() (credit.datasets.era5.ERA5Dataset method) _original_factors (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _output (credit.datasets.downscaling_dataset.DownscalingDataset attribute) _periodic_lower_bounds() (in module credit.regrid) _periodic_overlap() (in module credit.regrid) _periodic_upper_bounds() (in module credit.regrid) _pop_and_merge_targets() (credit.datasets.era5.ERA5Dataset method) _populate_dp_3d_idx() (credit.ocean.samudra_constants.TensorMap method) _populate_var_3d_idx() (credit.ocean.samudra_constants.TensorMap method) _process_chunk() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) _prognostic_data (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) _prognostic_mean_np (credit.datasets.om4_multistep_batcher.StandardScaler attribute) _prognostic_std_np (credit.datasets.om4_multistep_batcher.StandardScaler attribute) _relative_positional_encodings() (credit.models.swin.WindowMultiHeadAttention method) _reshape_and_concat() (credit.datasets.era5.ERA5Dataset method) _setup_arrangement() (credit.datasets.downscaling_dataset.DownscalingDataset method) _setup_datasets() (credit.datasets.downscaling_dataset.DownscalingDataset method) _shifted_window_attn() (credit.models.swin.SwinTransformerV2CrBlock method) _stored_latents (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _timestamps() (credit.datasets.era5.ERA5Dataset method) _to_tensor() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) _to_tuple() (credit.regrid.Grid method) _transforms module _wet_mask_np (credit.datasets.om4_multistep_batcher.StandardScaler attribute) A acc() (credit.metrics.LatWeightedMetricsClimatology method) acc_vars (credit.metrics.LatWeightedMetricsClimatology attribute) accum_log() (in module credit.replay_buffer) (in module credit.trainers.utils) (in module trainer_multistep_deprecated) act (credit.models.unet_attention_modules.CoordinateAttention attribute) (credit.models.unet_diffusion.Block attribute) add() (credit.replay_buffer.ReplayBuffer method) add_param_group() (credit.models.checkpoint.OptimizerWrapper method) add_pressure_and_geopotential() (in module credit.nwp) add_spatially_correlated_noise() (in module credit.ensemble.utils) adjust_start_times() (in module credit.ensemble.bred_vector) all_files (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) all_start_indices (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) allvars (_transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) AlmostFairKCRPSLoss (class in credit.losses.almost_fair_crps) alpha (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) alpha_init (credit.skebs.SKEBS attribute) amplitude (credit.ensemble.color.ColorNoise attribute) (credit.ensemble.gaussian.GaussianNoise attribute) (credit.ensemble.spherical.SphericalNoise attribute) annealed_probability() (in module credit.scheduler) app (in module data_server) apply_noise_perturbation_step() (in module credit.ensemble.color) apply_spectral_norm() (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.graph) (in module credit.models.swin) (in module credit.models.swin_wrf) (in module credit.models.wxformer.crossformer) (in module credit.models.wxformer.crossformer_downscaling) area (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) arrangement (credit.datasets.downscaling_dataset.DownscalingDataset attribute) Array (in module credit.data) (in module credit.datasets.downscaling_dataset) (in module credit.regrid) Attend (class in credit.attend) attend (credit.models.unet_diffusion.Attention attribute) Attention (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.unet_diffusion) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) attention (credit.models.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer.UpBlock attribute) AttentionConfig (class in credit.attend) attn (credit.models.swin.SwinTransformerV2CrBlock attribute) attn_drop (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) attn_dropout (credit.attend.Attend attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) attn_type (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) average_div_rot_spectrum() (in module credit.verification.standard) average_zonal_spectrum() (in module credit.verification.standard) avg_pool (credit.models.unet_attention_modules.ECABlock attribute) B b (credit.models.camulator.LayerNorm attribute) (credit.models.camulator.UpBlock attribute) (credit.models.camulator.UpBlockPS attribute) (credit.models.crossformer.LayerNorm attribute) (credit.models.crossformer.UpBlock attribute) (credit.models.dscale_wrf.DownBlock attribute) (credit.models.dscale_wrf.UpBlock attribute) (credit.models.fuxi.DownBlock attribute) (credit.models.fuxi.UpBlock attribute) (credit.models.swin_wrf.DownBlock attribute) (credit.models.swin_wrf.UpBlock attribute) (credit.models.wxformer.crossformer.LayerNorm attribute) (credit.models.wxformer.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) (credit.models.wxformer.crossformer_downscaling.LayerNorm attribute) (credit.models.wxformer.crossformer_downscaling.UpBlock attribute) backscatter_array (credit.skebs.BackscatterFixedCol attribute) (credit.skebs.BackscatterPrescribed attribute) BackscatterCNN (class in credit.skebs) BackscatterFCNN (class in credit.skebs) BackscatterFCNNWide (class in credit.skebs) BackscatterFixedCol (class in credit.skebs) BackscatterPrescribed (class in credit.skebs) BackscatterUnet (class in credit.skebs) backward() (credit.models.checkpoint.OptimizerWrapper method) (credit.trainers.trainerERA5_ensemble.Gather static method) backward_by_grad() (credit.models.checkpoint.OptimizerWrapper method) base_input_channels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) base_losses() (in module credit.losses) (in module credit.losses.base_losses) base_max_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) base_output_channels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) base_path (_transforms.NormalizeTendency attribute) BaseModel (class in credit.models.base_model) BaseTrainer (class in credit.trainers.base_trainer) batch_call_count (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) batch_indices (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) batch_indices_splits (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) batch_normalize (credit.losses.covariance.CovarianceWeightedMSELoss attribute) batch_size (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.samplers.DistributedMultiStepBatchSampler attribute) (credit.samplers.MultiStepBatchSamplerSubset attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) (rollout_to_netcdf.ForecastProcessor attribute) batched_forward (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) (credit.losses.kcrps.KCRPSLoss attribute) batches_per_epoch() (credit.datasets.era5_multistep.RepeatingIndexSampler method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.wrfmultistep.RepeatingIndexSampler method) BatchForecastLenDataLoader (class in credit.datasets.load_dataset_and_dataloader) BatchForecastLenSampler (class in credit.datasets.load_dataset_and_dataloader) BatchForecastLenSamplerSamudra (class in credit.datasets.load_dataset_and_dataloader) bchw_to_bhwc() (in module credit.models.swin) beta (credit.models.graph.TransformerConv attribute) bhwc_to_bchw() (in module credit.models.swin) bias (credit.models.graph.LayerNorm attribute) biased (credit.losses.kcrps.KCRPSLoss attribute) BilinearRegridder (class in credit.regrid) binned_spread_skill() (in module credit.verification.ensemble) Block (class in credit.models.unet_diffusion) block1 (credit.models.unet_diffusion.ResnetBlock attribute) block2 (credit.models.unet_diffusion.ResnetBlock attribute) blocks (credit.models.swin.SwinTransformerV2CrStage attribute) bn1 (credit.models.unet_attention_modules.CoordinateAttention attribute) bottom (credit.transforms_downscaling.Pad attribute) BOUND_VARS_MAP (in module credit.ocean.samudra_constants) boundary (credit.datasets.datamap.VarDict attribute) boundary_input (credit.data.Sample_WRF attribute) boundary_mean (credit.datasets.om4_multistep_batcher.StandardScaler attribute) boundary_std (credit.datasets.om4_multistep_batcher.StandardScaler attribute) BredVector (class in credit.ensemble.bred_vector) BridgescalerScaleState (class in credit.transforms) (class in credit.transforms.transforms_quantile) buffer_size (credit.replay_buffer.ReplayBuffer attribute) build_file_path() (in module credit.nwp) build_GFS_init() (in module credit.nwp) by (credit.transforms_downscaling.Expand attribute) C calc_edges() (in module graph_edges) calc_global_solar module calculate_crps_per_channel() (in module credit.ensemble.crps) calculate_ensemble_metrics() (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) calculate_mslp_and_append() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) Camulator (class in credit.models) (class in credit.models.camulator) cartopy_panel2() (in module credit.visualization_tools) cartopy_panel4() (in module credit.visualization_tools) cartopy_panel6() (in module credit.visualization_tools) cartopy_single_panel() (in module credit.visualization_tools) cast_tuple() (in module credit.diffusion_utils) (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.wxformer.crossformer) (in module credit.models.wxformer.crossformer_downscaling) channel_attention (credit.models.unet_attention_modules.EfficientMixedAttention attribute) channels (credit.diffusion.GaussianDiffusion attribute) (credit.diffusion.ModifiedGaussianDiffusion attribute) (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.ModifiedGaussianDiffusion attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.skebs.SKEBS attribute) channels_out (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) check_rollout_files() (in module ensemble_eval) checkpoint_stages (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) circular_pad1d() (in module credit.models.swin) clamp (credit.ensemble.bred_vector.BredVector attribute) clamp_max (credit.ensemble.bred_vector.BredVector attribute) clamp_min (credit.ensemble.bred_vector.BredVector attribute) cleanup() (in module credit.replay_buffer) (in module credit.trainers.utils) (in module trainer_multistep_deprecated) clear_stored_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) climatology (credit.metrics.LatWeightedMetricsClimatology attribute) Clip (class in credit.transforms_downscaling) clip_grad_by_norm() (credit.models.checkpoint.OptimizerWrapper method) clip_grad_by_value() (credit.models.checkpoint.OptimizerWrapper method) clip_parameters() (credit.skebs.SKEBS method) clone_dataset() (in module credit.ensemble.bred_vector) cmap_combine() (in module credit.visualization_tools) cmax (credit.transforms_downscaling.Clip attribute) cmin (credit.transforms_downscaling.Clip attribute) coef_a (credit.physics_core.physics_hybrid_sigma_level attribute), [1] coef_b (credit.physics_core.physics_hybrid_sigma_level attribute), [1] collate_fn() (in module credit.datasets.load_dataset_and_dataloader) colorbar_opt() (in module credit.visualization_tools) ColorNoise (class in credit.ensemble.color) combine_data() (in module credit.nwp) combine_microphysics_terms() (in module credit.gefs) combine_tiles() (in module credit.gefs) commit_id (in module credit._version) compute_anomalies() (in module credit.ocean.samudra_data) compute_density() (in module credit.physics_core) compute_hlevs() (credit.physics_core.ModelLevelPressures method) compute_metrics() (in module rollout_metrics) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) compute_mlev_thickness() (credit.physics_core.ModelLevelPressures method) compute_p() (credit.physics_core.ModelLevelPressures method) compute_spread_skill_metric() (in module rollout_metrics_noisy_model) compute_virtual_temperature() (in module credit.physics_core) concat (credit.models.graph.TransformerConv attribute) concat_and_reshape() (credit.models.base_model.BaseModel method) (credit.models.checkpoint.TorchFSDPModel method) (credit.replay_buffer.ReplayBuffer method) (in module credit.data) concat_fix() (in module credit.postblock._postblock) concat_for_inplace_ops() (in module credit.skebs) condition (credit.diffusion.GaussianDiffusion attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) conditional_dimensions (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) conf (_transforms.ToTensor attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) (in module credit.datasets.era5_singlestep) (in module credit.datasets.om4_multistep_batcher) (rollout_samudra.ForecastProcessor attribute) (rollout_to_netcdf.ForecastProcessor attribute) config (in module credit.xr_sampler) (rollout_metrics_noisy_model.Objective attribute) (train.Objective attribute) (train_les.Objective attribute) (train_multistep.Objective attribute) (train_wrf.Objective attribute) (train_wrf_multi.Objective attribute) config_file (in module credit.pbs) ConservativeRegridder (class in credit.regrid) construct_metadata() (in module credit.ocean.samudra_constants) conv (credit.models.camulator.UpBlock attribute) (credit.models.camulator.UpBlockPS attribute) (credit.models.dscale_wrf.DownBlock attribute) (credit.models.dscale_wrf.UpBlock attribute) (credit.models.fuxi.DownBlock attribute) (credit.models.fuxi.UpBlock attribute) (credit.models.swin_wrf.DownBlock attribute) (credit.models.swin_wrf.UpBlock attribute) (credit.models.unet_attention_modules.ECABlock attribute) (credit.models.unet_attention_modules.LightSpatialAttention attribute) (credit.models.unet_diffusion.PeriodicConv2d attribute) (credit.models.wxformer.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) (credit.models.wxformer.crossformer_downscaling.UpBlock attribute) (credit.skebs.BackscatterCNN attribute) conv1 (credit.models.unet_attention_modules.CoordinateAttention attribute) conv4up (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) conv_h (credit.models.unet_attention_modules.CoordinateAttention attribute) conv_w (credit.models.unet_attention_modules.CoordinateAttention attribute) convert_image_to_fn() (in module credit.diffusion_utils) convert_longitude() (credit.credit_ptype.CreditPostProcessor method) convert_tensor_out_to_dict() (in module credit.ocean.samudra_data) convs (credit.models.camulator.CrossEmbedLayer attribute) (credit.models.crossformer.CrossEmbedLayer attribute) (credit.models.wxformer.crossformer.CrossEmbedLayer attribute) (credit.models.wxformer.crossformer_downscaling.CrossEmbedLayer attribute) CoordinateAttention (class in credit.models.unet_attention_modules) copy_checkpoint() (in module credit.models.checkpoint) correlated (credit.models.CrossFormerWithNoise attribute) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) cos_lat (credit.skebs.SKEBS attribute) cosine_beta_schedule() (in module credit.diffusion) CosineAnnealingWarmupRestarts (class in credit.scheduler) count_channels() (in module credit.datasets.count_channels) CovarianceWeightedMSELoss (class in credit.losses.covariance) CP_DRY (in module credit.physics_constants) CP_ICE (in module credit.physics_constants) CP_LIQUID (in module credit.physics_constants) CP_VAPOR (in module credit.physics_constants) cpu_config (credit.attend.Attend attribute) create_diffusion() (in module credit.models.unet_diffusion) (in module credit.models.wxformer.crossformer_diffusion) create_model() (in module credit.models.unet_diffusion) (in module credit.models.wxformer.crossformer_diffusion) create_pressure_grid() (in module credit.interp) create_reduced_pressure_grid() (in module credit.interp) create_sigmoid_ramp_function() (in module credit.pol_lapdiff_filt) credit module credit._version module credit.animation module credit.attend module credit.boundary_padding module credit.credit_ptype module credit.data module credit.datasets module credit.datasets.count_channels module credit.datasets.datamap module credit.datasets.downscaling_dataset module credit.datasets.era5 module credit.datasets.era5_multistep module credit.datasets.era5_multistep_batcher module credit.datasets.era5_singlestep module credit.datasets.les_singlestep module credit.datasets.load_dataset_and_dataloader module credit.datasets.om4_multistep_batcher module credit.datasets.realtime_predict module credit.datasets.sequential_multistep module credit.datasets.wrf_singlestep module credit.datasets.wrfmultistep module credit.diffusion module credit.diffusion_utils module credit.distributed module credit.ensemble module credit.ensemble.bred_vector module credit.ensemble.color module credit.ensemble.crps module credit.ensemble.gaussian module credit.ensemble.spherical module credit.ensemble.temporal module credit.ensemble.utils module credit.forecast module credit.gefs module credit.grid module credit.interp module credit.losses module credit.losses.almost_fair_crps module credit.losses.base_losses module credit.losses.covariance module credit.losses.downscaling_loss module credit.losses.kcrps module credit.losses.les_loss module credit.losses.logcosh module credit.losses.msle module credit.losses.power module credit.losses.spectral module credit.losses.weighted_loss module credit.losses.xsigmoid module credit.losses.xtanh module credit.metadata module credit.metrics module credit.metrics_downscaling module credit.mixed_precision module credit.models module credit.models.base_model module credit.models.camulator module credit.models.checkpoint module credit.models.crossformer module credit.models.debugger_model module credit.models.dscale_wrf module credit.models.fuxi module credit.models.graph module credit.models.reset module credit.models.swin module credit.models.swin_wrf module credit.models.unet module credit.models.unet_attention_modules module credit.models.unet_diffusion module credit.models.unet_downscaling module credit.models.wxformer module credit.models.wxformer.crossformer module credit.models.wxformer.crossformer_diffusion module credit.models.wxformer.crossformer_downscaling module credit.models.wxformer.crossformer_ensemble module credit.models.wxformer.sdl_inference_wrapper module credit.models.wxformer.stochastic_decomposition_layer module credit.nwp module credit.ocean module credit.ocean.samudra_constants module credit.ocean.samudra_data module credit.output module credit.output_downscaling module credit.parser module credit.pbs module credit.physics_constants module credit.physics_core module credit.pol_lapdiff_filt module credit.postblock module credit.postblock._postblock module credit.postblock.wet_mask_samudra module credit.regrid module credit.replay_buffer module credit.samplers module credit.scheduler module credit.seed module credit.skebs module credit.solar module credit.trainers module credit.trainers.base_trainer module credit.trainers.ic_optimization module credit.trainers.trainer_downscaling module credit.trainers.trainer_om4_samudra module credit.trainers.trainerERA5 module credit.trainers.trainerERA5_Diffusion module credit.trainers.trainerERA5_ensemble module credit.trainers.trainerLES module credit.trainers.trainerWRF module credit.trainers.trainerWRF_multi module credit.trainers.utils module credit.transforms module credit.transforms.transforms_global module credit.transforms.transforms_les module credit.transforms.transforms_quantile module credit.transforms.transforms_wrf module credit.transforms_downscaling module credit.verification module credit.verification.ensemble module credit.verification.standard module credit.visualization_tools module credit.xr_sampler module credit_main_parser() (in module credit.parser) CreditPostProcessor (class in credit.credit_ptype) criterion (credit.diffusion.ModifiedGaussianDiffusion attribute) (credit.models.ModifiedGaussianDiffusion attribute) CrossEmbedLayer (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) CrossFormer (class in credit.models) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) crossformer_config (in module credit.models.wxformer.crossformer_diffusion) CrossFormerDiffusion (class in credit.models) (class in credit.models.wxformer.crossformer_diffusion) CrossFormerWithNoise (class in credit.models) (class in credit.models.wxformer.crossformer_ensemble) cse (credit.models.unet_attention_modules.SCSEAttention attribute) cube_embedding (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) cube_embedding_inside (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) cube_embedding_outside (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) CubeEmbedding (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) cuda_config (credit.attend.Attend attribute) cur_cycle_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) current_epoch (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) current_file_year (credit.xr_sampler.XRSamplerByYear attribute) current_index (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) cycle (credit.scheduler.CosineAnnealingWarmupRestarts attribute) cycle() (in module credit.diffusion_utils) (in module credit.replay_buffer) (in module credit.trainers.utils) (in module trainer_multistep_deprecated) cycle_mult (credit.scheduler.CosineAnnealingWarmupRestarts attribute) cycle_pattern() (credit.skebs.SKEBS method) D data_lookup (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) data_path (in module data_server) data_server module DataMap (class in credit.datasets.datamap) dataset (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.output_downscaling.OutputWrangler attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.samplers.MultiStepBatchSamplerSubset attribute) (credit.trainers.ic_optimization.TimeStepper attribute) dataset_id (in module credit.datasets.load_dataset_and_dataloader) datasets (credit.datasets.downscaling_dataset.DownscalingDataset attribute) DataTransforms (class in credit.transforms_downscaling) date2tindex() (credit.datasets.datamap.DataMap method) datetime_index (credit.data.Sample attribute) (credit.data.Sample_diag attribute) (credit.data.Sample_dscale attribute) (credit.data.Sample_LES attribute) (credit.data.Sample_WRF attribute) datetimes (credit.datasets.era5.ERA5Dataset attribute) days_of_year (trainerERA5_v1.TOADataLoader attribute) ddim_sample() (credit.diffusion.GaussianDiffusion method) ddim_sampling_eta (credit.diffusion.GaussianDiffusion attribute) debug_save_loc (credit.skebs.SKEBS attribute) DebuggerModel (class in credit.models) (class in credit.models.debugger_model) dec_norm (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) decoder (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) decoder_noise_factor (credit.models.CrossFormerWithNoise attribute) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) default() (in module credit.attend) (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) default_metadata (in module credit.ocean.samudra_constants) defaults (credit.models.checkpoint.OptimizerWrapper property) depth (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) DEPTH_I_LEVELS (in module credit.ocean.samudra_constants) DEPTH_LEVELS (in module credit.ocean.samudra_constants) description (in module ensemble_eval) (in module model_summary) (in module rollout_downscaling) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) (in module train) (in module train_multistep) device (credit.diffusion.GaussianDiffusion property) (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (in module credit.pol_lapdiff_filt) (rollout_metrics_noisy_model.Objective attribute) (rollout_samudra.ForecastProcessor attribute) (rollout_to_netcdf.ForecastProcessor attribute) (train.Objective attribute) (train_les.Objective attribute) (train_multistep.Objective attribute) (train_wrf.Objective attribute) (train_wrf_multi.Objective attribute) (trainer_multistep_deprecated.Trainer attribute) device_compatible_to() (in module credit.data) dewpoint_temp() (credit.credit_ptype.CreditPostProcessor method) diagnostic (credit.datasets.datamap.VarDict attribute) diff_lap2d_filt() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) Diffusion_and_Pole_Filter (class in credit.pol_lapdiff_filt) dim (credit.datasets.datamap.DataMap attribute) (credit.models.swin.SwinTransformerV2CrBlock attribute) (credit.models.unet_diffusion.SinusoidalPosEmb attribute) (credit.output_downscaling.OutputWriter attribute) dim_head (credit.models.DownscalingCrossFormer attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) disable_latent_capture() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) dissipation_scaling_coefficient (credit.skebs.SKEBS attribute) dissipation_type (credit.skebs.SKEBS attribute) distributed_model_wrapper() (in module credit.distributed) DistributedMultiStepBatchSampler (class in credit.samplers) DistributedSequentialDataset (class in credit.datasets.sequential_multistep) DistributedSequentialDatasetBasic (class in credit.datasets.sequential_multistep) div_rot_spectrum() (in module credit.verification.standard) divisible_by() (in module credit.diffusion_utils) do_eval() (in module ensemble_eval) down (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) DownBlock (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) download_gefs_run() (in module credit.gefs) downs (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) Downsample() (in module credit.models.unet_diffusion) downsample_factor (credit.models.unet_diffusion.UnetDiffusion property) (credit.models.UnetDiffusion property) downscale (credit.models.swin.SwinTransformerV2CrStage attribute) DownscalingCrossFormer (class in credit.models) (class in credit.models.wxformer.crossformer_downscaling) DownscalingDataset (class in credit.datasets.downscaling_dataset) DownscalingLoss (class in credit.losses) (class in credit.losses.downscaling_loss) DownscalingSegmentationModel (class in credit.models) (class in credit.models.unet_downscaling) dpb (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) draw_variables() (in module credit.visualization_tools) drop_path1 (credit.models.swin.SwinTransformerV2CrBlock attribute) drop_path2 (credit.models.swin.SwinTransformerV2CrBlock attribute) drop_var_from_dataset() (in module credit.data) dropout (credit.attend.Attend attribute) (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.graph.TransformerConv attribute) (credit.models.GraphResTransfGRU attribute) (credit.models.unet_diffusion.Block attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) ds_read_and_subset() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) DscaleTransformer (class in credit.models) (class in credit.models.dscale_wrf) dt (credit.datasets.era5.ERA5Dataset attribute) dtype (credit.replay_buffer.ReplayBuffer attribute) DynamicPositionBias (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) E ECABlock (class in credit.models.unet_attention_modules) edge_dim (credit.models.graph.TransformerConv attribute) edge_path (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) EfficientMixedAttention (class in credit.models.unet_attention_modules) embed_dim (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) enable_flash (credit.attend.AttentionConfig attribute) enable_latent_capture() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) enable_math (credit.attend.AttentionConfig attribute) enable_mem_efficient (credit.attend.AttentionConfig attribute) enc_norm (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) encode_datetime64() (in module credit.data) encoder (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) encoder_noise (credit.models.CrossFormerWithNoise attribute), [1] (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute), [1] encoder_noise_factor (credit.models.CrossFormerWithNoise attribute) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) end_datetime (credit.datasets.era5.ERA5Dataset attribute) ensemble_eval module ensemble_perturb (credit.ensemble.bred_vector.BredVector attribute) ensemble_size (credit.metrics_downscaling.UnWeightedMetrics attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) (rollout_to_netcdf.ForecastProcessor attribute) ensure_numpy_datetime() (in module credit.data) eps (credit.models.camulator.LayerNorm attribute) (credit.models.crossformer.LayerNorm attribute) (credit.models.graph.LayerNorm attribute) (credit.models.wxformer.crossformer.LayerNorm attribute) (credit.models.wxformer.crossformer_downscaling.LayerNorm attribute) (credit.skebs.SKEBS attribute) EPSGAS (in module credit.physics_constants) ERA5_and_Forcing_Dataset (class in credit.data) ERA5_and_Forcing_MultiStep (class in credit.datasets.era5_multistep) ERA5_and_Forcing_SingleStep (class in credit.datasets.era5_singlestep) ERA5_Dataset_Distributed (class in credit.data) ERA5_indices (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) ERA5_MultiStep_Batcher (class in credit.datasets.era5_multistep_batcher) era5_tsi_data() (in module credit.solar) ERA5Dataset (class in credit.datasets.era5) evaluate() (in module ensemble_eval) evapor_ind (credit.postblock._postblock.GlobalWaterFixer attribute) (credit.postblock.GlobalWaterFixer attribute) exists() (in module credit.attend) (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) Expand (class in credit.transforms_downscaling) exponent (credit.transforms_downscaling.Power attribute) extract() (in module credit.diffusion) (in module credit.models.wxformer.crossformer_diffusion) extract_month_day_hour() (in module credit.data) extract_variable_levels() (credit.credit_ptype.CreditPostProcessor method) extract_wet_mask() (in module credit.ocean.samudra_data) F fast_state_interp_loop() (in module credit.interp) fc (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) fc1 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) fc2 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) fc3 (credit.skebs.BackscatterFCNNWide attribute) fc4 (credit.skebs.BackscatterFCNNWide attribute) feat_size (credit.models.swin.SwinTransformerV2CrBlock attribute) (credit.models.swin.SwinTransformerV2CrStage attribute) feature_info (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) FeedForward (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) ff_dropout (credit.models.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) figure_panel_planner() (in module credit.visualization_tools) file_dict (credit.datasets.era5.ERA5Dataset attribute) filename (in module credit.datasets.era5_multistep) filename_diagnostic (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) filename_dyn_forcing (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) filename_forcing (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) filename_static (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) filename_surface (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) filenames (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.xr_sampler.XRSamplerByYear attribute) film (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) filter_ds() (in module credit.data) final_conv (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) final_res_block (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) find_common_indices() (in module credit.data) find_key() (in module credit.datasets.count_channels) find_key_for_number() (in module credit.data) find_nearest() (in module credit.pol_lapdiff_filt) find_start_stop_indices() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.wrf_singlestep.WRFPredict method) first_cycle_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) first_date (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) fit() (credit.trainers.base_trainer.BaseTrainer method) (trainer_multistep_deprecated.Trainer method) (trainerERA5_multistep_v1.Trainer method) fit_deprecated() (trainerERA5_v2.Trainer method) fit_era5_scaler_times() (in module scaler) fit_scaled_era5_time_residuals() (in module scaler) flag_diagnostic (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_dyn_forcing (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_energy_conserve (credit.ensemble.bred_vector.BredVector attribute) flag_forcing (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_mass_conserve (credit.ensemble.bred_vector.BredVector attribute) flag_static (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_surface (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_surface_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_upper_air (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) flag_water_conserve (credit.ensemble.bred_vector.BredVector attribute) flash (credit.attend.Attend attribute) flash_attn() (credit.attend.Attend method) flatten_list() (in module credit.data) for_len (_transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) for_len_outside (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) forecast_details (in module credit.forecast) forecast_end_time (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_hour (credit.replay_buffer.ReplayBuffer attribute) forecast_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSamplerSamudra attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) (credit.skebs.SKEBS attribute) forecast_len_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) forecast_period (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) forecast_start_time (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_step_count (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) forecast_step_counts (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) forecast_times (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_timestep (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_windows (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) ForecastProcessor (class in credit.trainers.ic_optimization) (class in rollout_samudra) (class in rollout_to_netcdf) format_data() (in module credit.nwp) format_datetime() (in module credit.nwp) forward() (credit.attend.Attend method) (credit.diffusion.GaussianDiffusion method) (credit.diffusion.ModifiedGaussianDiffusion method) (credit.ensemble.spherical.SphericalRandomField method) (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss method) (credit.losses.covariance.CovarianceWeightedMSELoss method) (credit.losses.downscaling_loss.DownscalingLoss method) (credit.losses.DownscalingLoss method) (credit.losses.kcrps.KCRPSLoss method) (credit.losses.les_loss.LESLoss2D method) (credit.losses.logcosh.LogCoshLoss method) (credit.losses.msle.MSLELoss method) (credit.losses.power.PSDLoss method) (credit.losses.spectral.SpectralLoss2D method) (credit.losses.VariableTotalLoss2D method) (credit.losses.weighted_loss.VariableTotalLoss2D method) (credit.losses.xsigmoid.XSigmoidLoss method) (credit.losses.xtanh.XTanhLoss method) (credit.models.Camulator method) (credit.models.camulator.Attention method) (credit.models.camulator.Camulator method) (credit.models.camulator.CrossEmbedLayer method) (credit.models.camulator.CubeEmbedding method) (credit.models.camulator.DynamicPositionBias method) (credit.models.camulator.FeedForward method) (credit.models.camulator.LayerNorm method) (credit.models.camulator.Transformer method) (credit.models.camulator.UpBlock method) (credit.models.camulator.UpBlockPS method) (credit.models.checkpoint.ModelWrapper method) (credit.models.CrossFormer method) (credit.models.crossformer.Attention method) (credit.models.crossformer.CrossEmbedLayer method) (credit.models.crossformer.CrossFormer method) (credit.models.crossformer.CubeEmbedding method) (credit.models.crossformer.DynamicPositionBias method) (credit.models.crossformer.FeedForward method) (credit.models.crossformer.LayerNorm method) (credit.models.crossformer.Transformer method) (credit.models.crossformer.UpBlock method) (credit.models.CrossFormerDiffusion method) (credit.models.CrossFormerWithNoise method) (credit.models.debugger_model.DebuggerModel method) (credit.models.DebuggerModel method) (credit.models.DownscalingCrossFormer method) (credit.models.DownscalingSegmentationModel method) (credit.models.dscale_wrf.CubeEmbedding method) (credit.models.dscale_wrf.DownBlock method) (credit.models.dscale_wrf.DscaleTransformer method) (credit.models.dscale_wrf.UpBlock method) (credit.models.dscale_wrf.UTransformer method) (credit.models.DscaleTransformer method) (credit.models.Fuxi method) (credit.models.fuxi.CubeEmbedding method) (credit.models.fuxi.DownBlock method) (credit.models.fuxi.Fuxi method) (credit.models.fuxi.UpBlock method) (credit.models.fuxi.UTransformer method) (credit.models.graph.GateCell method) (credit.models.graph.GraphResidualBlock method) (credit.models.graph.GraphResTransfGRU method) (credit.models.graph.LayerNorm method) (credit.models.graph.TransformerConv method) (credit.models.GraphResTransfGRU method) (credit.models.ModifiedGaussianDiffusion method) (credit.models.SegmentationModel method) (credit.models.swin.PatchEmbed method) (credit.models.swin.PatchMerging method) (credit.models.swin.SwinTransformerV2Cr method) (credit.models.swin.SwinTransformerV2CrBlock method) (credit.models.swin.SwinTransformerV2CrStage method) (credit.models.swin.WindowMultiHeadAttention method) (credit.models.swin.WindowMultiHeadAttentionNoPos method) (credit.models.swin_wrf.CubeEmbedding method) (credit.models.swin_wrf.DownBlock method) (credit.models.swin_wrf.UpBlock method) (credit.models.swin_wrf.UTransformer method) (credit.models.swin_wrf.WRFTransformer method) (credit.models.SwinTransformerV2Cr method) (credit.models.unet.SegmentationModel method) (credit.models.unet_attention_modules.CoordinateAttention method) (credit.models.unet_attention_modules.ECABlock method) (credit.models.unet_attention_modules.EfficientMixedAttention method) (credit.models.unet_attention_modules.LightSpatialAttention method) (credit.models.unet_attention_modules.SCSEAttention method) (credit.models.unet_diffusion.Attention method) (credit.models.unet_diffusion.Block method) (credit.models.unet_diffusion.LinearAttention method) (credit.models.unet_diffusion.PeriodicConv2d method) (credit.models.unet_diffusion.RandomOrLearnedSinusoidalPosEmb method) (credit.models.unet_diffusion.ResnetBlock method) (credit.models.unet_diffusion.RMSNorm method) (credit.models.unet_diffusion.SinusoidalPosEmb method) (credit.models.unet_diffusion.UnetDiffusion method) (credit.models.unet_downscaling.DownscalingSegmentationModel method) (credit.models.UnetDiffusion method) (credit.models.WRFTransformer method) (credit.models.WXFormer method) (credit.models.wxformer.crossformer.Attention method) (credit.models.wxformer.crossformer.CrossEmbedLayer method) (credit.models.wxformer.crossformer.CrossFormer method) (credit.models.wxformer.crossformer.CubeEmbedding method) (credit.models.wxformer.crossformer.DynamicPositionBias method) (credit.models.wxformer.crossformer.FeedForward method) (credit.models.wxformer.crossformer.LayerNorm method) (credit.models.wxformer.crossformer.Transformer method) (credit.models.wxformer.crossformer.UpBlock method) (credit.models.wxformer.crossformer.UpBlockPS method) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion method) (credit.models.wxformer.crossformer_diffusion.Transformer method) (credit.models.wxformer.crossformer_diffusion.UpBlock method) (credit.models.wxformer.crossformer_downscaling.Attention method) (credit.models.wxformer.crossformer_downscaling.CrossEmbedLayer method) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding method) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer method) (credit.models.wxformer.crossformer_downscaling.DynamicPositionBias method) (credit.models.wxformer.crossformer_downscaling.FeedForward method) (credit.models.wxformer.crossformer_downscaling.LayerNorm method) (credit.models.wxformer.crossformer_downscaling.Transformer method) (credit.models.wxformer.crossformer_downscaling.UpBlock method) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise method) (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer method), [1] (credit.postblock._postblock.GlobalEnergyFixer method) (credit.postblock._postblock.GlobalMassFixer method) (credit.postblock._postblock.GlobalWaterFixer method) (credit.postblock._postblock.PostBlock method) (credit.postblock._postblock.TracerFixer method) (credit.postblock.GlobalEnergyFixer method) (credit.postblock.GlobalMassFixer method) (credit.postblock.GlobalWaterFixer method) (credit.postblock.PostBlock method) (credit.postblock.TracerFixer method) (credit.postblock.wet_mask_samudra.WetMaskBlock method) (credit.postblock.WetMaskBlock method) (credit.replay_buffer.WeightedRMSE method) (credit.skebs.BackscatterCNN method) (credit.skebs.BackscatterFCNN method) (credit.skebs.BackscatterFCNNWide method) (credit.skebs.BackscatterFixedCol method) (credit.skebs.BackscatterPrescribed method) (credit.skebs.BackscatterUnet method) (credit.skebs.SKEBS method) (credit.trainers.trainerERA5_ensemble.Gather static method) forward_features() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) forward_head() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) frames (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.SegmentationModel attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (credit.skebs.SKEBS attribute) freeze (credit.models.CrossFormerWithNoise attribute) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) freeze_pattern_weights (credit.skebs.SKEBS attribute) from_degrees() (credit.regrid.Grid class method) FSDPOptimizerWrapper (class in credit.models.checkpoint) full_pos_embed (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) full_state_pressure_interpolation() (in module credit.interp) Fuxi (class in credit.models) (class in credit.models.fuxi) G g (credit.models.camulator.LayerNorm attribute) (credit.models.crossformer.LayerNorm attribute) (credit.models.unet_diffusion.RMSNorm attribute) (credit.models.wxformer.crossformer.LayerNorm attribute) (credit.models.wxformer.crossformer_downscaling.LayerNorm attribute) gamma (credit.scheduler.CosineAnnealingWarmupRestarts attribute) GateCell (class in credit.models.graph) gated_unit (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) Gather (class in credit.trainers.trainerERA5_ensemble) gather_tensor() (in module credit.trainers.trainerERA5_ensemble) gaussian_grid (in module data_server) GaussianDiffusion (class in credit.diffusion) GaussianNoise (class in credit.ensemble.gaussian) gefs_init module generate_bred_vectors() (in module credit.ensemble.bred_vector) generate_bred_vectors_cycle() (in module credit.ensemble.bred_vector) generate_datetime() (in module credit.data) generate_forecasts() (in module credit.forecast) generate_integer_list_around() (in module credit.data) generate_interpolation_sequence() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) generate_scaled_ensemble() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) geopotential_from_model_vars() (in module credit.interp) get_classifier() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) get_climatology() (credit.metrics.LatWeightedMetricsClimatology method) get_colormap() (in module credit.visualization_tools) get_colormap_extend() (in module credit.visualization_tools) get_data() (in module ensemble_eval) get_file_extension() (in module credit.models.checkpoint) get_forward_data() (in module credit.data) get_inference_field() (in module data_server) get_inference_steps() (in module credit.ocean.samudra_data) get_instance() (credit.ocean.samudra_constants.TensorMap class method) (credit.ocean.samudra_data.Normalize class method) get_lr() (credit.scheduler.CosineAnnealingWarmupRestarts method) get_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) get_norm_unnorm_dicts() (in module credit.ocean.samudra_data) get_num_cpus() (in module credit.diffusion_utils) (in module credit.pbs) get_pad2d() (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin_wrf) get_pad3d() (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin_wrf) get_projection() (in module credit.visualization_tools) get_psd() (credit.losses.power.PSDLoss method) get_rank_info() (in module credit.distributed) get_solar_index() (in module credit.solar) get_solar_radiation_loc() (in module credit.solar) get_stored_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) get_time_from (credit.datasets.downscaling_dataset.DownscalingDataset attribute) get_time_variable() (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) get_toa_radiation() (in module credit.solar) get_tsi() (in module credit.solar) get_variable_range() (in module credit.visualization_tools) get_variable_range_with_rounding() (in module credit.visualization_tools) getdata() (credit.datasets.downscaling_dataset.DownscalingDataset method) getgrad() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) getuv() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) gfs_init module gfs_map (in module credit.nwp) glob (credit.datasets.datamap.DataMap attribute) GlobalEnergyFixer (class in credit.postblock) (class in credit.postblock._postblock) GlobalMassFixer (class in credit.postblock) (class in credit.postblock._postblock) GlobalWaterFixer (class in credit.postblock) (class in credit.postblock._postblock) gp_to_height_dam() (in module credit.animation) grad_checkpointing (credit.models.swin.SwinTransformerV2CrStage attribute) graph_blocks (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) graph_edges module graph_edges_knn module graph_norm_layers (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) GraphResidualBlock (class in credit.models.graph) GraphResTransfGRU (class in credit.models) (class in credit.models.graph) gravity (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) GRAVITY (in module credit.physics_constants) Grid (class in credit.regrid) grid (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.SKEBS attribute) grid2spec() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) grid_file (in module credit.grid) grid_predictions() (credit.credit_ptype.CreditPostProcessor method) grid_size (credit.models.swin.PatchEmbed attribute) group_matcher() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) H h_h_ln (credit.models.graph.GateCell attribute) h_x_ln (credit.models.graph.GateCell attribute) hamp (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) handle_signal() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) has_int_squareroot() (in module credit.diffusion_utils) havg (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) head (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) heads (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.graph.TransformerConv attribute) (credit.models.GraphResTransfGRU attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) hemispheric_rescale (credit.ensemble.bred_vector.BredVector attribute) (credit.ensemble.temporal.TemporalNoise attribute) hemispheric_rescale() (in module credit.ensemble.utils) hidden_size (credit.models.graph.GateCell attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) hist_len (_transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) hist_len_outside (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) historical_ERA5_images (credit.data.Sample attribute) history_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) (credit.diffusion.ModifiedGaussianDiffusion attribute) (credit.models.ModifiedGaussianDiffusion attribute) history_len_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) histroy_len (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) hour_to_nanoseconds() (in module credit.data) hours_of_day (trainerERA5_v1.TOADataLoader attribute) HR_input (credit.data.Sample_dscale attribute) HR_target (credit.data.Sample_dscale attribute) I Identity (class in credit.transforms_downscaling) identity() (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) IMAGE_ATTR_NAMES (in module credit.data) image_height (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.debugger_model) (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin) (in module credit.models.wxformer.crossformer) image_size (credit.diffusion.GaussianDiffusion attribute) image_width (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (in module credit.models.wxformer.crossformer_downscaling) img_size (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.WRFTransformer attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) immiscible (credit.diffusion.GaussianDiffusion attribute) in_channels (credit.models.graph.LayerNorm attribute) (credit.models.graph.TransformerConv attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) (credit.skebs.BackscatterUnet attribute) in_chans (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) in_features (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) index (credit.replay_buffer.ReplayBuffer attribute) index_subset (credit.samplers.MultiStepBatchSamplerSubset attribute) indices() (credit.regrid.NearestRegridder method) indpol (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) init_datetime (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) init_instance() (credit.ocean.samudra_constants.TensorMap class method) (credit.ocean.samudra_data.Normalize class method) init_lr() (credit.scheduler.CosineAnnealingWarmupRestarts method) init_times (credit.samplers.MultiStepBatchSamplerSubset attribute) init_values (credit.models.swin.SwinTransformerV2CrBlock attribute) init_weights() (credit.models.swin.SwinTransformerV2CrBlock method) (in module credit.models.swin) initial_index (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) initialize_batch() (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) initialize_filters() (credit.skebs.SKEBS method) initialize_other_properties() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) initialize_pattern() (credit.skebs.SKEBS method) initialize_sht() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) initialize_skebs_parameters() (credit.skebs.SKEBS method) input (credit.datasets.downscaling_dataset.Sample attribute) input_channels (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) input_length (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) input_only_channels (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.skebs.SKEBS attribute) input_resolution (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) input_shape (credit.replay_buffer.ReplayBuffer attribute) input_size (credit.models.graph.GraphResidualBlock attribute) input_static_dim (credit.ensemble.bred_vector.BredVector attribute) integral (credit.physics_core.physics_hybrid_sigma_level attribute) integration_steps (credit.ensemble.bred_vector.BredVector attribute) interp (credit.models.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) interp_geopotential_to_pressure_levels() (in module credit.interp) interp_hybrid_to_height_agl() (in module credit.interp) interp_hybrid_to_hybrid_levels() (in module credit.interp) interp_hybrid_to_pressure_levels() (in module credit.interp) interp_pressure_to_hybrid_levels() (in module credit.interp) interp_temperature_to_pressure_levels() (in module credit.interp) interpolate() (credit.diffusion.GaussianDiffusion method) interpolate_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) interpolate_to_model_level() (in module credit.nwp) interpolate_vertical_levels() (in module credit.gefs) inverse_sht (credit.ensemble.spherical.SphericalRandomField attribute) inverse_transform() (_transforms.NormalizeState method) (_transforms.NormalizeState_Quantile method) (_transforms.NormalizeTendency method) (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.transforms.BridgescalerScaleState method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeLES method) (credit.transforms.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.NormalizeWRF method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_quantile.BridgescalerScaleState method) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.transforms_wrf.NormalizeWRF method) inverse_transform_dataset() (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) inverse_transform_input() (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeLES method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) is_ddim_sampling (credit.diffusion.GaussianDiffusion attribute) is_dtensor_checkpoint() (in module credit.models.checkpoint) is_fully_initialized (credit.physics_core.ModelLevelPressures attribute) is_safetensor_checkpoint() (in module credit.models.checkpoint) is_safetensors_available() (in module credit.models.checkpoint) is_training (credit.skebs.SKEBS attribute) iteration (credit.skebs.SKEBS attribute) iteration_stop (credit.skebs.SKEBS attribute) K k_to_c() (in module credit.animation) k_to_f() (in module credit.animation) KCRPSLoss (class in credit.losses.kcrps) keep_dataset_vars() (in module credit.data) kgkg_to_gkg() (in module credit.animation) L last_date (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) lat (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) (credit.regrid.Grid attribute) lat_weights (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) (credit.replay_buffer.WeightedRMSE attribute) latitude_modes (credit.ensemble.spherical.SphericalRandomField attribute) latitude_slices (in module credit.verification.ensemble) latitude_weights() (in module credit.losses.weighted_loss) latlon_grid (in module data_server) latlons (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) (rollout_to_netcdf.ForecastProcessor attribute) latN (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) LatWeightedMetrics (class in credit.metrics) LatWeightedMetricsClimatology (class in credit.metrics) LatWeightedMetricsEnsemble (class in credit.metrics) launch_script() (in module credit.pbs) launch_script_mpi() (in module credit.pbs) layer (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) LayerNorm (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.graph) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) layers (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.camulator.DynamicPositionBias attribute) (credit.models.camulator.FeedForward attribute) (credit.models.camulator.Transformer attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.DynamicPositionBias attribute) (credit.models.crossformer.FeedForward attribute) (credit.models.crossformer.Transformer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.DynamicPositionBias attribute) (credit.models.wxformer.crossformer.FeedForward attribute) (credit.models.wxformer.crossformer.Transformer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.Transformer attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_downscaling.DynamicPositionBias attribute) (credit.models.wxformer.crossformer_downscaling.FeedForward attribute) (credit.models.wxformer.crossformer_downscaling.Transformer attribute) lead_time_periods (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) (rollout_to_netcdf.ForecastProcessor attribute) left (credit.transforms_downscaling.Pad attribute) len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSamplerSamudra attribute) length_scale (credit.ensemble.spherical.SphericalNoise attribute) LES_file_indices (credit.datasets.les_singlestep.LESDataset attribute) LES_input (credit.data.Sample_LES attribute) LES_target (credit.data.Sample_LES attribute) LESDataset (class in credit.datasets.les_singlestep) LESLoss2D (class in credit.losses.les_loss) LESPredict (class in credit.datasets.les_singlestep) level_map (in module credit.nwp) levels (_transforms.NormalizeState attribute) (_transforms.NormalizeState_Quantile attribute) (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) (credit.skebs.BackscatterUnet attribute) (credit.skebs.SKEBS attribute) (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) levels_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) LH_ICE (in module credit.physics_constants) LH_WATER (in module credit.physics_constants) LightSpatialAttention (class in credit.models.unet_attention_modules) lin_key (credit.models.graph.TransformerConv attribute) lin_query (credit.models.graph.TransformerConv attribute) lin_value (credit.models.graph.TransformerConv attribute) linear (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) linear_beta_schedule() (in module credit.diffusion) LinearAttention (class in credit.models.unet_diffusion) list_diag_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_dyn_forcing_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_stored_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) list_surf_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_surf_ds_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_upper_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_upper_ds_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) lmax (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.SKEBS attribute) load_data() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) load_dataloader() (in module credit.datasets.load_dataset_and_dataloader) load_dataset() (in module credit.datasets.load_dataset_and_dataloader) load_dataset_and_sampler() (in module train_les) (in module train_wrf) (in module train_wrf_multi) load_dataset_and_sampler_zscore_only() (in module train) load_forecasts() (in module credit.forecast) load_fsdp_or_checkpoint_policy() (in module credit.models) load_gfs_data() (in module credit.nwp) load_gfs_variable() (in module credit.nwp) load_graph() (credit.models.graph.GraphResTransfGRU method) (credit.models.GraphResTransfGRU method) load_inputs() (credit.replay_buffer.ReplayBuffer method) load_loss() (credit.diffusion.ModifiedGaussianDiffusion method) (credit.models.ModifiedGaussianDiffusion method) (in module credit.losses) load_member_tiles() (in module credit.gefs) load_metadata() (in module credit.output) load_model() (credit.credit_ptype.CreditPostProcessor method) (credit.models.base_model.BaseModel class method) (in module credit.models) load_model_name() (credit.models.base_model.BaseModel class method) (in module credit.models) load_model_state() (in module credit.models.checkpoint) load_model_states_and_optimizer() (in module train) (in module train_les) (in module train_multistep) (in module train_wrf) (in module train_wrf_multi) load_premade_encoder_model() (in module credit.models.unet) (in module credit.models.unet_downscaling) (in module credit.skebs) load_scaler() (credit.credit_ptype.CreditPostProcessor method) load_scheduler() (in module credit.scheduler) load_state_dict() (credit.models.checkpoint.OptimizerWrapper method) (in module credit.models.checkpoint) load_state_dict_error_handler() (in module credit.models.checkpoint) load_trainer() (in module credit.trainers) load_transform() (in module credit.datasets.om4_multistep_batcher) load_transforms() (in module credit.transforms) load_unet_attention() (in module credit.models.unet_attention_modules) load_unsharded_model() (credit.models.checkpoint.TorchFSDPCheckpointIO method) load_unsharded_optimizer() (credit.models.checkpoint.TorchFSDPCheckpointIO method) load_zarr_as_input() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.les_singlestep.LESPredict method) (credit.datasets.wrf_singlestep.WRFPredict method) log_uniform_noise() (in module credit.diffusion) LogCoshLoss (class in credit.losses.logcosh) logger (in module _transforms) (in module credit.datasets) (in module credit.datasets.era5) (in module credit.datasets.era5_multistep) (in module credit.datasets.era5_multistep_batcher) (in module credit.datasets.sequential_multistep) (in module credit.datasets.wrfmultistep) (in module credit.ensemble.bred_vector) (in module credit.forecast) (in module credit.losses) (in module credit.losses.base_losses) (in module credit.losses.downscaling_loss) (in module credit.losses.les_loss) (in module credit.losses.weighted_loss) (in module credit.metrics) (in module credit.models) (in module credit.models.base_model) (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.debugger_model) (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin) (in module credit.models.swin_wrf) (in module credit.models.unet) (in module credit.models.unet_downscaling) (in module credit.models.wxformer.crossformer) (in module credit.models.wxformer.crossformer_diffusion) (in module credit.models.wxformer.crossformer_downscaling) (in module credit.models.wxformer.crossformer_ensemble) (in module credit.models.wxformer.sdl_inference_wrapper) (in module credit.output) (in module credit.output_downscaling) (in module credit.pbs) (in module credit.postblock._postblock) (in module credit.samplers) (in module credit.skebs) (in module credit.trainers) (in module credit.trainers.base_trainer) (in module credit.trainers.ic_optimization) (in module credit.trainers.trainer_downscaling) (in module credit.trainers.trainer_om4_samudra) (in module credit.trainers.trainerERA5) (in module credit.trainers.trainerERA5_Diffusion) (in module credit.trainers.trainerERA5_ensemble) (in module credit.trainers.trainerLES) (in module credit.trainers.trainerWRF) (in module credit.trainers.trainerWRF_multi) (in module credit.transforms) (in module credit.transforms.transforms_global) (in module credit.transforms.transforms_les) (in module credit.transforms.transforms_quantile) (in module credit.transforms.transforms_wrf) (in module credit.transforms_downscaling) (in module credit.verification.ensemble) (in module credit.verification.standard) (in module credit.visualization_tools) (in module rollout_downscaling) (in module rollout_les) (in module rollout_metrics) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) (in module rollout_realtime) (in module rollout_samudra) (in module rollout_to_netcdf) (in module rollout_wrf) (in module trainerERA5_multistep_concat) (in module trainerERA5_multistep_sum) (in module trainerERA5_v1) (in module trainerERA5_v2) logit_scale (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) lon (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) (credit.regrid.Grid attribute) longitude_modes (credit.ensemble.spherical.SphericalRandomField attribute) lonN (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) lons (in module credit.solar) LR_input (credit.data.Sample_dscale attribute) lr_lambda_phase1() (in module credit.scheduler) lr_lambda_phase2() (in module credit.scheduler) M mae() (credit.metrics.LatWeightedMetricsClimatology method) main() (in module calc_global_solar) (in module gefs_init) (in module gfs_init) (in module graph_edges) (in module graph_edges_knn) (in module model_summary) (in module rollout_les) (in module rollout_metrics) (in module rollout_samudra) (in module rollout_to_netcdf) (in module rollout_wrf) (in module scaler) (in module train) (in module train_les) (in module train_multistep) (in module train_wrf) (in module train_wrf_multi) main_cli() (in module rollout_realtime) make_xarray() (in module credit.output) make_xarray_diag() (in module credit.output) manager (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher attribute) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) map_gridline_opt() (in module credit.visualization_tools) MASK_VARS (in module credit.ocean.samudra_constants) max_forecast_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) max_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) mean (_transforms.NormalizeTendency attribute) (credit.transforms_downscaling.Zscore attribute) mean_ds (_transforms.NormalizeState attribute) (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) mean_ds_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) mean_sea_level_pressure() (in module credit.interp) mean_sea_level_pressure_simple() (in module credit.interp) mean_tensors (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) mean_tensors_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) mem_kv (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) merge_linear (credit.models.graph.GraphResidualBlock attribute) message() (credit.models.graph.TransformerConv method) meta_data (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) (rollout_to_netcdf.ForecastProcessor attribute) meta_mlp (credit.models.swin.WindowMultiHeadAttention attribute) metric (rollout_metrics_noisy_model.Objective attribute) (train.Objective attribute) (train_les.Objective attribute) (train_multistep.Objective attribute) (train_wrf.Objective attribute) (train_wrf_multi.Objective attribute) metric_fn (credit.replay_buffer.ReplayBuffer attribute) mid_attn (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) mid_block1 (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) mid_block2 (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) min_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) Minmax (class in credit.transforms_downscaling) mlp (credit.models.swin.SwinTransformerV2CrBlock attribute) (credit.models.unet_diffusion.ResnetBlock attribute) mmax (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.SKEBS attribute) (credit.transforms_downscaling.Minmax attribute) mmin (credit.transforms_downscaling.Minmax attribute) mode (credit.boundary_padding.TensorPadding attribute) (credit.datasets.datamap.DataMap property) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.downscaling_dataset.DownscalingDataset property) (credit.transforms_downscaling.Pad attribute) model (credit.diffusion.GaussianDiffusion attribute) (credit.ensemble.bred_vector.BredVector attribute) (credit.models.checkpoint.FSDPOptimizerWrapper attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) (credit.skebs.BackscatterUnet attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.trainerWRF_multi.Trainer attribute) (credit.trainers.TrainerWRFMulti attribute) (in module credit.models.checkpoint) (trainer_multistep_deprecated.Trainer attribute) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) model_predictions() (credit.diffusion.GaussianDiffusion method) (credit.diffusion.ModifiedGaussianDiffusion method) (credit.models.ModifiedGaussianDiffusion method) model_summary module model_types (in module credit.models) ModelLevelPressures (class in credit.physics_core) ModelPrediction (class in credit.diffusion) (class in credit.models.wxformer.crossformer_diffusion) ModelWrapper (class in credit.models.checkpoint) ModifiedGaussianDiffusion (class in credit.diffusion) (class in credit.models) modulation (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer attribute), [1] module _transforms calc_global_solar credit credit._version credit.animation credit.attend credit.boundary_padding credit.credit_ptype credit.data credit.datasets credit.datasets.count_channels credit.datasets.datamap credit.datasets.downscaling_dataset credit.datasets.era5 credit.datasets.era5_multistep credit.datasets.era5_multistep_batcher credit.datasets.era5_singlestep credit.datasets.les_singlestep credit.datasets.load_dataset_and_dataloader credit.datasets.om4_multistep_batcher credit.datasets.realtime_predict credit.datasets.sequential_multistep credit.datasets.wrf_singlestep credit.datasets.wrfmultistep credit.diffusion credit.diffusion_utils credit.distributed credit.ensemble credit.ensemble.bred_vector credit.ensemble.color credit.ensemble.crps credit.ensemble.gaussian credit.ensemble.spherical credit.ensemble.temporal credit.ensemble.utils credit.forecast credit.gefs credit.grid credit.interp credit.losses credit.losses.almost_fair_crps credit.losses.base_losses credit.losses.covariance credit.losses.downscaling_loss credit.losses.kcrps credit.losses.les_loss credit.losses.logcosh credit.losses.msle credit.losses.power credit.losses.spectral credit.losses.weighted_loss credit.losses.xsigmoid credit.losses.xtanh credit.metadata credit.metrics credit.metrics_downscaling credit.mixed_precision credit.models credit.models.base_model credit.models.camulator credit.models.checkpoint credit.models.crossformer credit.models.debugger_model credit.models.dscale_wrf credit.models.fuxi credit.models.graph credit.models.reset credit.models.swin credit.models.swin_wrf credit.models.unet credit.models.unet_attention_modules credit.models.unet_diffusion credit.models.unet_downscaling credit.models.wxformer credit.models.wxformer.crossformer credit.models.wxformer.crossformer_diffusion credit.models.wxformer.crossformer_downscaling credit.models.wxformer.crossformer_ensemble credit.models.wxformer.sdl_inference_wrapper credit.models.wxformer.stochastic_decomposition_layer credit.nwp credit.ocean credit.ocean.samudra_constants credit.ocean.samudra_data credit.output credit.output_downscaling credit.parser credit.pbs credit.physics_constants credit.physics_core credit.pol_lapdiff_filt credit.postblock credit.postblock._postblock credit.postblock.wet_mask_samudra credit.regrid credit.replay_buffer credit.samplers credit.scheduler credit.seed credit.skebs credit.solar credit.trainers credit.trainers.base_trainer credit.trainers.ic_optimization credit.trainers.trainer_downscaling credit.trainers.trainer_om4_samudra credit.trainers.trainerERA5 credit.trainers.trainerERA5_Diffusion credit.trainers.trainerERA5_ensemble credit.trainers.trainerLES credit.trainers.trainerWRF credit.trainers.trainerWRF_multi credit.trainers.utils credit.transforms credit.transforms.transforms_global credit.transforms.transforms_les credit.transforms.transforms_quantile credit.transforms.transforms_wrf credit.transforms_downscaling credit.verification credit.verification.ensemble credit.verification.standard credit.visualization_tools credit.xr_sampler data_server ensemble_eval gefs_init gfs_init graph_edges graph_edges_knn model_summary rollout_downscaling rollout_les rollout_metrics rollout_metrics_noisy_ic rollout_metrics_noisy_model rollout_realtime rollout_samudra rollout_to_netcdf rollout_wrf scaler train train_les train_multistep train_wrf train_wrf_multi trainer_multistep_deprecated trainerERA5_multistep_concat trainerERA5_multistep_sum trainerERA5_multistep_v1 trainerERA5_v1 trainerERA5_v2 module (credit.models.checkpoint.ModelWrapper attribute) (credit.models.checkpoint.TorchFSDPModel attribute) (credit.trainers.trainerWRF_multi.Trainer attribute) (credit.trainers.TrainerWRFMulti attribute) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) mse() (credit.metrics.LatWeightedMetricsClimatology method) MSLELoss (class in credit.losses.msle) MultiprocessingBatcher (class in credit.datasets.era5_multistep_batcher) MultiprocessingBatcherPrefetch (class in credit.datasets.era5_multistep_batcher) multistep (credit.skebs.SKEBS attribute) MultiStepBatchSamplerSubset (class in credit.samplers) N n_3dvar_levels (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) n_blocks (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) n_levels (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) n_static_variables (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) n_surface_variables (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) n_variables (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) (in module credit.models.graph) nanoseconds_to_year() (in module credit.data) nearest_neighbor_indices() (in module credit.regrid) NearestRegridder (class in credit.regrid) next_n_hour() (in module credit.data) nlat (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterPrescribed attribute) (credit.skebs.BackscatterUnet attribute) (credit.skebs.SKEBS attribute) nlon (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterPrescribed attribute) (credit.skebs.BackscatterUnet attribute) (credit.skebs.SKEBS attribute) no_autocast (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) noise_amplitude (credit.ensemble.bred_vector.BredVector attribute) noise_assignment() (credit.diffusion.GaussianDiffusion method) noise_factor (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer attribute), [1] noise_generator (credit.ensemble.temporal.TemporalNoise attribute) noise_inject1 (credit.models.CrossFormerWithNoise attribute) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) noise_inject2 (credit.models.CrossFormerWithNoise attribute) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) noise_inject3 (credit.models.CrossFormerWithNoise attribute) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) noise_latent_dim (credit.models.CrossFormerWithNoise attribute), [1] (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute), [1] noise_transform (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer attribute), [1] noop (credit.output_downscaling.OutputWriter attribute) norm (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.PatchMerging attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.Block attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) norm1 (credit.models.swin.SwinTransformerV2CrBlock attribute) norm2 (credit.models.swin.SwinTransformerV2CrBlock attribute) norm3 (credit.models.swin.SwinTransformerV2CrBlock attribute) norm_layer (credit.models.graph.GraphResidualBlock attribute) Normalize (class in credit.ocean.samudra_data) normalize (credit.datasets.datamap.DataMap attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.diffusion.GaussianDiffusion attribute) normalize_boundary() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) Normalize_ERA5_and_Forcing (class in credit.transforms) (class in credit.transforms.transforms_global) normalize_numpy_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) normalize_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) normalize_tensor_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) normalize_to_neg_one_to_one() (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) NormalizeLES (class in credit.transforms) (class in credit.transforms.transforms_les) NormalizeState (class in _transforms) NormalizeState_Quantile (class in _transforms) NormalizeState_Quantile_Bridgescalar (class in credit.transforms) (class in credit.transforms.transforms_quantile) NormalizeTendency (class in _transforms) NormalizeWRF (class in credit.transforms) (class in credit.transforms.transforms_wrf) num_boundary_vars (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) num_cycles (credit.ensemble.bred_vector.BredVector attribute) num_epochs (in module credit.scheduler) num_features (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) num_forcing_static (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) num_forecast_steps (credit.datasets.era5.ERA5Dataset attribute) (credit.samplers.DistributedMultiStepBatchSampler attribute) (credit.samplers.MultiStepBatchSamplerSubset attribute) num_heads (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) num_indices_per_rank (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) num_patches (credit.models.swin.PatchEmbed attribute) num_prognostic_vars (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) num_replicas (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) num_start_batches (credit.samplers.MultiStepBatchSamplerSubset attribute) num_timesteps (credit.diffusion.GaussianDiffusion attribute) num_to_groups() (in module credit.diffusion_utils) num_upper_air (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) num_upper_air_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) num_workers (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher attribute) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) numpy_dir (credit.replay_buffer.ReplayBuffer attribute) O Objective (class in rollout_metrics_noisy_model) (class in train) (class in train_les) (class in train_multistep) (class in train_wrf) (class in train_wrf_multi) objective (credit.diffusion.GaussianDiffusion attribute) Ocean_MultiStep_Batcher (class in credit.datasets.om4_multistep_batcher) Ocean_Tensor_Batcher (class in credit.datasets.om4_multistep_batcher) off_diagonal_scale (credit.losses.covariance.CovarianceWeightedMSELoss attribute) offset_noise_strength (credit.diffusion.GaussianDiffusion attribute) omega (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) OMEGA (in module credit.physics_constants) once() (in module credit.attend) one_shot (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) operations (credit.postblock._postblock.PostBlock attribute) (credit.postblock.PostBlock attribute) optim (credit.models.checkpoint.OptimizerWrapper attribute) OptimizerWrapper (class in credit.models.checkpoint) option (in module credit.datasets.era5_multistep_batcher) out_channels (credit.models.graph.TransformerConv attribute) out_chans (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.WRFTransformer attribute) out_size (credit.models.graph.GraphResidualBlock attribute) output (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.downscaling_dataset.DownscalingDataset property) output_channels (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.camulator.UpBlock attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.UpBlock attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_downscaling.UpBlock attribute) output_dir (credit.output_downscaling.OutputWrangler attribute) output_dtype (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) output_frames (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) output_length (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) output_only_channels (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.skebs.SKEBS attribute) OutputWrangler (class in credit.output_downscaling) OutputWriter (class in credit.output_downscaling) outside_file_indices (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) outside_file_year_range (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) P p_losses() (credit.diffusion.GaussianDiffusion method) (credit.diffusion.ModifiedGaussianDiffusion method) (credit.models.ModifiedGaussianDiffusion method) p_mean_variance() (credit.diffusion.GaussianDiffusion method) p_sample() (credit.diffusion.GaussianDiffusion method) p_sample_loop() (credit.diffusion.GaussianDiffusion method) pa_to_hpa() (in module credit.animation) Pad (class in credit.transforms_downscaling) pad (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) (credit.skebs.BackscatterUnet attribute) pad() (credit.boundary_padding.TensorPadding method) (credit.skebs.BackscatterCNN method) pad_lat (credit.skebs.BackscatterCNN attribute) pad_lon (credit.skebs.BackscatterCNN attribute) pad_NS (credit.boundary_padding.TensorPadding attribute) pad_WE (credit.boundary_padding.TensorPadding attribute) padding (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) (credit.models.unet_diffusion.PeriodicConv2d attribute) param_groups (credit.models.checkpoint.OptimizerWrapper property) parameters (credit.models.checkpoint.OptimizerWrapper property) parse_dtype() (in module credit.mixed_precision) parser (in module credit.models.wxformer.sdl_inference_wrapper) passthrough() (in module credit.losses.covariance) patch_embed (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) patch_height (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) patch_size (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.WRFTransformer attribute) patch_width (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) PatchEmbed (class in credit.models.swin) patches_resolution (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) PatchMerging (class in credit.models.swin) PeriodicConv2d (class in credit.models.unet_diffusion) perturb() (credit.ensemble.bred_vector.BredVector method) perturb_channel_idx (credit.ensemble.bred_vector.BredVector attribute) perturbation_method (credit.ensemble.bred_vector.BredVector attribute) perturbation_std (credit.ensemble.temporal.TemporalNoise attribute) phased_lr_lambda() (in module credit.scheduler) physics_hybrid_sigma_level (class in credit.physics_core) physics_linear (credit.models.graph.GraphResidualBlock attribute) physics_pressure_level (class in credit.physics_core) PI (in module credit.physics_constants) (in module credit.postblock._postblock) plev_dim (credit.physics_core.ModelLevelPressures attribute) plot_global_animation() (in module credit.animation) plot_regional_animation() (in module credit.animation) polefilt_lap2d_QV1() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) polefilt_lap2d_V1() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) polefilt_lap2d_V2() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) polfiltT() (in module credit.pol_lapdiff_filt) pool_h (credit.models.unet_attention_modules.CoordinateAttention attribute) pool_w (credit.models.unet_attention_modules.CoordinateAttention attribute) populate() (credit.replay_buffer.ReplayBuffer method) post_conf (credit.ensemble.bred_vector.BredVector attribute) (credit.skebs.SKEBS attribute) PostBlock (class in credit.postblock) (class in credit.postblock._postblock) Power (class in credit.transforms_downscaling) pre_out_dim (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) precip_ind (credit.postblock._postblock.GlobalWaterFixer attribute) (credit.postblock.GlobalWaterFixer attribute) pred_noise (credit.diffusion.ModelPrediction attribute) (credit.models.wxformer.crossformer_diffusion.ModelPrediction attribute) pred_x_start (credit.diffusion.ModelPrediction attribute) (credit.models.wxformer.crossformer_diffusion.ModelPrediction attribute) predict() (in module rollout_downscaling) (in module rollout_les) (in module rollout_metrics) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) (in module rollout_realtime) (in module rollout_samudra) (in module rollout_to_netcdf) (in module rollout_wrf) predict_data_check() (in module credit.parser) Predict_Dataset (class in credit.data) Predict_Dataset_Batcher (class in credit.datasets.era5_multistep_batcher) predict_noise_from_start() (credit.diffusion.GaussianDiffusion method) Predict_Ocean_Batcher (class in credit.datasets.om4_multistep_batcher) predict_start_from_noise() (credit.diffusion.GaussianDiffusion method) predict_start_from_v() (credit.diffusion.GaussianDiffusion method) predict_v() (credit.diffusion.GaussianDiffusion method) prefetch_batches() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) prefetch_factor (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) prefetch_queue (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) prefetch_thread (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) pressure_integral_midpoint() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) pressure_integral_midpoint_sliced() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) pressure_integral_trapz() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) pressure_integral_trapz_sliced() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) pressure_thickness (credit.physics_core.physics_pressure_level attribute) previous_hourly_steps() (in module credit.data) primary_main() (in module train_les) (in module train_wrf) (in module train_wrf_multi) print_once (in module credit.attend) process() (credit.output_downscaling.OutputWrangler method) (credit.trainers.ic_optimization.ForecastProcessor method) (rollout_samudra.ForecastProcessor method) (rollout_to_netcdf.ForecastProcessor method) process_forecast() (in module rollout_realtime) process_member() (in module credit.gefs) process_pressure_interp() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) PROG_VARS_MAP (in module credit.ocean.samudra_constants) prognostic (credit.datasets.datamap.VarDict attribute) prognostic_mean (credit.datasets.om4_multistep_batcher.StandardScaler attribute) prognostic_std (credit.datasets.om4_multistep_batcher.StandardScaler attribute) proj (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.unet_diffusion.Block attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) proj_drop (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) projections (in module credit.animation) ps (credit.models.camulator.UpBlockPS attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) PSDLoss (class in credit.losses.power) ptr (credit.replay_buffer.ReplayBuffer attribute) ptype_classification() (credit.credit_ptype.CreditPostProcessor method) Q q_ind_end (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock._postblock.GlobalMassFixer attribute) (credit.postblock._postblock.GlobalWaterFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalMassFixer attribute) (credit.postblock.GlobalWaterFixer attribute) q_ind_start (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock._postblock.GlobalMassFixer attribute) (credit.postblock._postblock.GlobalWaterFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalMassFixer attribute) (credit.postblock.GlobalWaterFixer attribute) Q_inds (credit.skebs.SKEBS attribute) q_posterior() (credit.diffusion.GaussianDiffusion method) q_sample() (credit.diffusion.GaussianDiffusion method) q_values (credit.replay_buffer.ReplayBuffer attribute) qkv (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) R r_h_ln (credit.models.graph.GateCell attribute) r_x_ln (credit.models.graph.GateCell attribute) RAD_EARTH (in module credit.physics_constants) radius (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) random_or_learned_sinusoidal_cond (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) RandomOrLearnedSinusoidalPosEmb (class in credit.models.unet_diffusion) rank (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.trainerWRF_multi.Trainer attribute) (credit.trainers.TrainerWRFMulti attribute) (in module credit.models.reset) (trainer_multistep_deprecated.Trainer attribute) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) rank_histogram_apply() (in module credit.verification.ensemble) RDGAS (in module credit.physics_constants) read() (credit.datasets.datamap.DataMap method) RealtimePredictDataset (class in credit.datasets.realtime_predict) rearrange() (credit.datasets.downscaling_dataset.DownscalingDataset method) reddening (credit.ensemble.color.ColorNoise attribute) reduction (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) (credit.losses.covariance.CovarianceWeightedMSELoss attribute) (credit.losses.logcosh.LogCoshLoss attribute) (credit.losses.msle.MSLELoss attribute) (credit.losses.spectral.SpectralLoss2D attribute) (credit.losses.xsigmoid.XSigmoidLoss attribute) (credit.losses.xtanh.XTanhLoss attribute) (credit.models.swin.PatchMerging attribute) reduction_function (credit.losses.covariance.CovarianceWeightedMSELoss attribute) reduction_functions (in module credit.losses.covariance) regrid() (in module credit.nwp) regrid_array() (credit.regrid.BilinearRegridder method) (credit.regrid.ConservativeRegridder method) (credit.regrid.NearestRegridder method) (credit.regrid.Regridder method) regrid_dataset() (credit.regrid.Regridder method) regrid_member() (in module credit.gefs) regrid_variable() (in module credit.nwp) Regridder (class in credit.regrid) regridder (in module data_server) reload() (credit.replay_buffer.ReplayBuffer method) relu (credit.skebs.BackscatterUnet attribute) relu1 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) (credit.skebs.SKEBS attribute) relu2 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) relu3 (credit.skebs.BackscatterFCNNWide attribute) relu4 (credit.skebs.BackscatterFCNNWide attribute) remove_string_by_pattern() (in module credit.parser) rename_variables() (in module credit.gefs) rename_vars() (in module credit.ocean.samudra_data) RepeatingIndexSampler (class in credit.datasets.era5_multistep) (class in credit.datasets.wrfmultistep) replace_nested_key() (in module credit.parser) ReplayBuffer (class in credit.replay_buffer) res_conv (credit.models.unet_diffusion.ResnetBlock attribute) rescale() (in module credit.transforms_downscaling) rescale_minmax() (in module credit.datasets.datamap) reset() (credit.trainers.ic_optimization.TimeStepper method) reset_classifier() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) reset_latent_capture() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) reset_model() (in module credit.models.reset) reset_parameters() (credit.models.graph.LayerNorm method) (credit.models.graph.TransformerConv method) reset_to_original() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) reshape_only() (credit.models.base_model.BaseModel method) (credit.models.checkpoint.TorchFSDPModel method) (in module credit.data) residual (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) ResnetBlock (class in credit.models.unet_diffusion) results (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher attribute) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) retain_graph (credit.skebs.SKEBS attribute) return_target (credit.datasets.era5.ERA5Dataset attribute) revert() (credit.datasets.downscaling_dataset.DownscalingDataset method) RHO_WATER (in module credit.physics_constants) right (credit.transforms_downscaling.Pad attribute) rk4() (credit.models.Camulator method) (credit.models.camulator.Camulator method) (credit.models.CrossFormer method) (credit.models.crossformer.CrossFormer method) (credit.models.DownscalingCrossFormer method) (credit.models.DownscalingSegmentationModel method) (credit.models.SegmentationModel method) (credit.models.unet.SegmentationModel method) (credit.models.unet_downscaling.DownscalingSegmentationModel method) (credit.models.WXFormer method) (credit.models.wxformer.crossformer.CrossFormer method) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer method) rk4_integration (credit.models.DownscalingSegmentationModel attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) rmse() (credit.metrics.LatWeightedMetricsClimatology method) rmse_scores (credit.replay_buffer.ReplayBuffer attribute) RMSNorm (class in credit.models.unet_diffusion) rng (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) rollout_downscaling module rollout_forecast() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) rollout_les module rollout_metrics module rollout_metrics_noisy_ic module rollout_metrics_noisy_model module rollout_p (credit.data.Predict_Dataset attribute) rollout_realtime module rollout_samudra module rollout_to_netcdf module rollout_wrf module root_weight (credit.models.graph.TransformerConv attribute) rootpath (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) run_multiscale_control_experiment() (in module credit.models.wxformer.sdl_inference_wrapper) RVGAS (in module credit.physics_constants) S Sample (class in credit.data) (class in credit.datasets.downscaling_dataset) sample() (credit.diffusion.GaussianDiffusion method) (credit.replay_buffer.ReplayBuffer method) Sample_diag (class in credit.data) Sample_dscale (class in credit.data) sample_len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) Sample_LES (class in credit.data) Sample_WRF (class in credit.data) sampler (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) samples_dir (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) sampling_timesteps (credit.diffusion.GaussianDiffusion attribute) save() (credit.replay_buffer.ReplayBuffer method) save_checkpoint() (credit.trainers.base_trainer.BaseTrainer method) save_fsdp_checkpoint() (credit.trainers.base_trainer.BaseTrainer method) save_model() (credit.models.base_model.BaseModel method) save_netcdf_clean() (in module credit.output) save_netcdf_diag() (in module credit.output) save_netcdf_increment() (in module credit.output) (in module credit.trainers.ic_optimization) save_state_dict() (in module credit.models.checkpoint) save_unsharded_model() (credit.models.checkpoint.TorchFSDPCheckpointIO method) save_unsharded_optimizer() (credit.models.checkpoint.TorchFSDPCheckpointIO method) save_vars (credit.credit_ptype.CreditPostProcessor attribute) scale (credit.attend.Attend attribute) (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.unet_diffusion.RMSNorm attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) scale_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) scale_latents_multilevel() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) scale_loss() (credit.models.checkpoint.OptimizerWrapper method) scaler module scaler_3d (_transforms.NormalizeState_Quantile attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_3ds (_transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_df (_transforms.NormalizeState_Quantile attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_file (_transforms.NormalizeState_Quantile attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_surf (_transforms.NormalizeState_Quantile attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_surfs (_transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scalers (in module scaler) scrip_from_latlon_grid() (in module credit.grid) SCSEAttention (class in credit.models.unet_attention_modules) SDLWrapper (class in credit.models.wxformer.sdl_inference_wrapper) seed (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) seed_everything() (in module credit.seed) SegmentationModel (class in credit.models) (class in credit.models.unet) self_condition (credit.diffusion.GaussianDiffusion attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) sequential_attn (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) set_decoder_modulation() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_decoder_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_decoder_style_vector() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_encoder_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_epoch() (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.wrfmultistep.WRFMultiStep method) set_globals() (in module credit.datasets) set_grad_checkpointing() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) set_manual_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) setup() (credit.trainers.Trainer404 method) (credit.trainers.trainer_downscaling.Trainer method) (in module credit.distributed) (in module model_summary) setup_data_loading() (in module credit.datasets) shape (credit.regrid.Grid property) shared_mem_draw_wrapper() (in module credit.visualization_tools) sharp (credit.models.camulator.UpBlock attribute) (credit.models.camulator.UpBlockPS attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) should_not_checkpoint() (in module credit.distributed) shuffle (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) sigmoid (credit.models.unet_attention_modules.CoordinateAttention attribute) (credit.models.unet_attention_modules.ECABlock attribute) (credit.models.unet_attention_modules.LightSpatialAttention attribute) (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.BackscatterCNN attribute) sigmoid_beta_schedule() (in module credit.diffusion) sindex2dates() (credit.datasets.datamap.DataMap method) single_sample_forward() (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss method) (credit.losses.kcrps.KCRPSLoss method) SinusoidalPosEmb (class in credit.models.unet_diffusion) size (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.replay_buffer.ReplayBuffer attribute) size_full (credit.datasets.les_singlestep.LESDataset attribute) size_list (credit.datasets.les_singlestep.LESDataset attribute) SKEBS (class in credit.skebs) skip_periods (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) skip_target (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) smoothness (credit.ensemble.spherical.SphericalNoise attribute) source (credit.regrid.Regridder attribute) source_name (credit.datasets.era5.ERA5Dataset attribute) sp_index (credit.skebs.SKEBS attribute) spatial_attention (credit.models.unet_attention_modules.EfficientMixedAttention attribute) spec2grid() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) spec_coef_is_initialized (credit.skebs.SKEBS attribute) spectral_lambda_reg (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) spectral_wavenum_init (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) SpectralLoss2D (class in credit.losses.spectral) SphericalNoise (class in credit.ensemble.spherical) SphericalRandomField (class in credit.ensemble.spherical) split_and_reshape() (credit.models.base_model.BaseModel method) (in module credit.output) spread_error() (in module credit.verification.ensemble) sst_forcing (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) stages (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) standard_normal_noise() (in module credit.diffusion) StandardScaler (class in credit.datasets.om4_multistep_batcher) start_datetime (credit.datasets.era5.ERA5Dataset attribute) state_dict() (credit.models.checkpoint.OptimizerWrapper method) state_trans (credit.skebs.SKEBS attribute) state_transformer (rollout_samudra.ForecastProcessor attribute) state_vars (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) static_inds (credit.skebs.SKEBS attribute) static_variables (_transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) std (_transforms.NormalizeTendency attribute) std_ds (_transforms.NormalizeState attribute) (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) std_ds_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) std_tensors (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) std_tensors_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) stdev (credit.transforms_downscaling.Zscore attribute) step() (credit.models.checkpoint.OptimizerWrapper method) (credit.scheduler.CosineAnnealingWarmupRestarts method) step_in_cycle (credit.scheduler.CosineAnnealingWarmupRestarts attribute) StochasticDecompositionLayer (class in credit.models.wxformer.stochastic_decomposition_layer) stop_event (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) stop_signal (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) store_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) subset_extent() (credit.credit_ptype.CreditPostProcessor method) subset_patch() (in module credit.data) supported_encoders (in module credit.models.unet) (in module credit.models.unet_downscaling) supported_models (in module credit.models.unet) (in module credit.models.unet_downscaling) (in module credit.skebs) surf_LH_ind (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) surf_LR_ind (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) surf_SH_ind (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) surf_solar_ind (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) surface (in module credit.nwp) surface_channels (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.skebs.SKEBS attribute) surface_files (credit.datasets.realtime_predict.RealtimePredictDataset attribute) surface_pressure (credit.physics_core.physics_hybrid_sigma_level attribute) surface_variables (_transforms.NormalizeState attribute) (_transforms.NormalizeState_Quantile attribute) (_transforms.NormalizeTendency attribute) (_transforms.ToTensor attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) swin_from_yaml() (in module credit.models.swin) SwinTransformerV2Cr (class in credit.models) (class in credit.models.swin) SwinTransformerV2CrBlock (class in credit.models.swin) SwinTransformerV2CrStage (class in credit.models.swin) swinv2net() (in module credit.models.swin) T T_ind_end (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) T_ind_start (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) T_inds (credit.skebs.SKEBS attribute) target (credit.datasets.downscaling_dataset.Sample attribute) (credit.regrid.Regridder attribute) target_ERA5_images (credit.data.Sample attribute) target_shift_size (credit.models.swin.SwinTransformerV2CrBlock attribute) template_path (credit.output_downscaling.OutputWriter attribute) templates (credit.output_downscaling.OutputWrangler attribute) temporal_correlation (credit.ensemble.temporal.TemporalNoise attribute) TemporalNoise (class in credit.ensemble.temporal) TensorMap (class in credit.ocean.samudra_constants) TensorPadding (class in credit.boundary_padding) theta (credit.models.unet_diffusion.SinusoidalPosEmb attribute) time_emb_proj (credit.models.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) time_encode (credit.data.Sample_diag attribute) (credit.data.Sample_dscale attribute) (credit.data.Sample_WRF attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) time_mlp (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) time_steps (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) time_to_emb (credit.models.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) times_b (credit.replay_buffer.TOADataLoader attribute) (trainerERA5_v1.TOADataLoader attribute) TimeStepper (class in credit.trainers.ic_optimization) tnames (credit.datasets.downscaling_dataset.DownscalingDataset attribute) to() (credit.models.graph.GraphResTransfGRU method) (credit.models.GraphResTransfGRU method) to_emb (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) to_out (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) to_qkv (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) to_tensor() (credit.datasets.downscaling_dataset.DownscalingDataset method) TOA (credit.replay_buffer.TOADataLoader attribute) (trainerERA5_v1.TOADataLoader attribute) TOA_OLR_ind (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) TOA_solar_ind (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) TOADataLoader (class in credit.replay_buffer) (class in trainerERA5_v1) top (credit.transforms_downscaling.Pad attribute) TorchFSDPCheckpointIO (class in credit.models.checkpoint) TorchFSDPModel (class in credit.models.checkpoint) total_column_water() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) total_dim (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) total_dry_air_mass() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) total_input_channels (credit.models.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) total_len (credit.datasets.les_singlestep.LESDataset attribute) total_length (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) total_n_vars (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) total_seq_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) ToTensor (class in _transforms) ToTensor_BridgeScaler (class in credit.transforms) (class in credit.transforms.transforms_quantile) ToTensor_ERA5_and_Forcing (class in credit.transforms) (class in credit.transforms.transforms_global) ToTensorLES (class in credit.transforms) (class in credit.transforms.transforms_les) ToTensorWRF (class in credit.transforms) (class in credit.transforms.transforms_wrf) tracer_indices (credit.postblock._postblock.TracerFixer attribute) (credit.postblock.TracerFixer attribute) tracer_thres (credit.postblock._postblock.TracerFixer attribute) (credit.postblock.TracerFixer attribute) tracer_thres_max (credit.postblock._postblock.TracerFixer attribute) (credit.postblock.TracerFixer attribute) TracerFixer (class in credit.postblock) (class in credit.postblock._postblock) train module train() (rollout_metrics_noisy_model.Objective method) (train.Objective method) (train_les.Objective method) (train_multistep.Objective method) (train_wrf.Objective method) (train_wrf_multi.Objective method) train_alpha (credit.skebs.SKEBS attribute) train_les module train_multistep module train_one_epoch() (credit.replay_buffer.Trainer method) (credit.trainers.base_trainer.BaseTrainer method) (credit.trainers.ic_optimization.Trainer method) (credit.trainers.Trainer404 method) (credit.trainers.trainer_downscaling.Trainer method) (credit.trainers.trainer_om4_samudra.Trainer method) (credit.trainers.TrainerEnsemble method) (credit.trainers.TrainerERA5 method) (credit.trainers.trainerERA5.Trainer method) (credit.trainers.TrainerERA5_Diffusion method) (credit.trainers.trainerERA5_Diffusion.Trainer method) (credit.trainers.trainerERA5_ensemble.Trainer method) (credit.trainers.TrainerIC method) (credit.trainers.TrainerLES method) (credit.trainers.trainerLES.Trainer method) (credit.trainers.TrainerSamudra method) (credit.trainers.TrainerWRF method) (credit.trainers.trainerWRF.Trainer method) (credit.trainers.trainerWRF_multi.Trainer method) (credit.trainers.TrainerWRFMulti method) (trainer_multistep_deprecated.Trainer method) (trainerERA5_multistep_concat.Trainer method) (trainerERA5_multistep_sum.Trainer method) (trainerERA5_multistep_v1.Trainer method) (trainerERA5_v1.Trainer method) (trainerERA5_v2.Trainer method) train_wrf module train_wrf_multi module Trainer (class in credit.replay_buffer) (class in credit.trainers.ic_optimization) (class in credit.trainers.trainer_downscaling) (class in credit.trainers.trainer_om4_samudra) (class in credit.trainers.trainerERA5) (class in credit.trainers.trainerERA5_Diffusion) (class in credit.trainers.trainerERA5_ensemble) (class in credit.trainers.trainerLES) (class in credit.trainers.trainerWRF) (class in credit.trainers.trainerWRF_multi) (class in trainer_multistep_deprecated) (class in trainerERA5_multistep_concat) (class in trainerERA5_multistep_sum) (class in trainerERA5_multistep_v1) (class in trainerERA5_v1) (class in trainerERA5_v2) Trainer404 (class in credit.trainers) trainer_multistep_deprecated module trainer_types (in module credit.trainers) TrainerEnsemble (class in credit.trainers) TrainerERA5 (class in credit.trainers) TrainerERA5_Diffusion (class in credit.trainers) trainerERA5_multistep_concat module trainerERA5_multistep_sum module trainerERA5_multistep_v1 module trainerERA5_v1 module trainerERA5_v2 module TrainerIC (class in credit.trainers) TrainerLES (class in credit.trainers) TrainerSamudra (class in credit.trainers) TrainerWRF (class in credit.trainers) TrainerWRFMulti (class in credit.trainers) training_data_check() (in module credit.parser) training_loss (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) transform (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) transform() (_transforms.NormalizeState method) (_transforms.NormalizeState_Quantile method) (_transforms.NormalizeTendency method) (credit.transforms.BridgescalerScaleState method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeLES method) (credit.transforms.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.NormalizeWRF method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_quantile.BridgescalerScaleState method) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.transforms_wrf.NormalizeWRF method) transform_array() (_transforms.NormalizeState method) (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.transforms.BridgescalerScaleState method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeLES method) (credit.transforms.NormalizeWRF method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_quantile.BridgescalerScaleState method) (credit.transforms.transforms_wrf.NormalizeWRF method) transform_data() (credit.credit_ptype.CreditPostProcessor method) transform_dataset() (_transforms.NormalizeState method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) transform_era5_times() (in module scaler) Transformer (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_diffusion) (class in credit.models.wxformer.crossformer_downscaling) transformer (credit.models.graph.GraphResidualBlock attribute) TransformerConv (class in credit.models.graph) transforms (credit.transforms_downscaling.DataTransforms attribute) tropics_only_dissipation (credit.skebs.SKEBS attribute) tsi_data (in module credit.solar) U U_ind_end (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) U_ind_start (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) U_inds (credit.skebs.SKEBS attribute) u_transformer (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) unet_config (in module credit.models.unet_diffusion) UnetDiffusion (class in credit.models) (class in credit.models.unet_diffusion) unnormalize (credit.diffusion.GaussianDiffusion attribute) unnormalize_numpy_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) unnormalize_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) unnormalize_tensor_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) unnormalize_to_zero_to_one() (in module credit.diffusion_utils) unpad() (credit.boundary_padding.TensorPadding method) (credit.skebs.BackscatterCNN method) unscale_grad() (credit.models.checkpoint.OptimizerWrapper method) unstagger_winds() (in module credit.gefs) unused (credit.datasets.datamap.VarDict attribute) UnWeightedMetrics (class in credit.metrics_downscaling) unwrap() (credit.models.checkpoint.ModelWrapper method) (credit.models.checkpoint.OptimizerWrapper method) (credit.models.checkpoint.TorchFSDPModel method) unwrap_model() (credit.models.checkpoint.FSDPOptimizerWrapper method) up (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) up_block1 (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) up_block2 (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) up_block3 (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) up_block4 (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) UpBlock (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_diffusion) (class in credit.models.wxformer.crossformer_downscaling) UpBlockPS (class in credit.models.camulator) (class in credit.models.wxformer.crossformer) update() (credit.replay_buffer.ReplayBuffer method) update_input_size() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.swin.SwinTransformerV2CrBlock method) (credit.models.swin.SwinTransformerV2CrStage method) (credit.models.swin.WindowMultiHeadAttention method) (credit.models.swin.WindowMultiHeadAttentionNoPos method) (credit.models.SwinTransformerV2Cr method) update_on_batch (in module credit.scheduler) update_on_epoch (in module credit.scheduler) update_q_values() (credit.replay_buffer.ReplayBuffer method) update_with_predictions() (credit.replay_buffer.ReplayBuffer method) upper_air (in module credit.nwp) upper_air_pressure (credit.physics_core.physics_pressure_level attribute) ups (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) Upsample() (in module credit.models.unet_diffusion) upsample_v_conv (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.UpBlock attribute) upsample_with_ps (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) use_edge_attr (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) use_interp (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.WRFTransformer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) use_noise (credit.models.fuxi.UTransformer attribute) use_padding (credit.ensemble.spherical.SphericalNoise attribute) (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.UnetDiffusion attribute) (credit.models.WRFTransformer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) use_post_block (credit.ensemble.bred_vector.BredVector attribute) (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.DownscalingSegmentationModel attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.SegmentationModel attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.UnetDiffusion attribute) (credit.models.WRFTransformer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) use_power_loss (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) use_spectral_loss (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) use_spectral_norm (credit.models.Camulator attribute) (credit.models.camulator.Camulator attribute) (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.CrossFormerDiffusion attribute) (credit.models.DownscalingCrossFormer attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.DscaleTransformer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.WRFTransformer attribute) (credit.models.WXFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) use_statics (credit.skebs.SKEBS attribute) UTransformer (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) V V_ind_end (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) V_ind_start (credit.postblock._postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) V_inds (credit.skebs.SKEBS attribute) VALID_FIELD_TYPES (in module credit.datasets.era5) valid_forecast_len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.skebs.SKEBS attribute) valid_history_len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) valid_indices (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) validate() (credit.trainers.base_trainer.BaseTrainer method) (credit.trainers.ic_optimization.Trainer method) (credit.trainers.Trainer404 method) (credit.trainers.trainer_downscaling.Trainer method) (credit.trainers.trainer_om4_samudra.Trainer method) (credit.trainers.TrainerEnsemble method) (credit.trainers.TrainerERA5 method) (credit.trainers.trainerERA5.Trainer method) (credit.trainers.TrainerERA5_Diffusion method) (credit.trainers.trainerERA5_Diffusion.Trainer method) (credit.trainers.trainerERA5_ensemble.Trainer method) (credit.trainers.TrainerIC method) (credit.trainers.TrainerLES method) (credit.trainers.trainerLES.Trainer method) (credit.trainers.TrainerSamudra method) (credit.trainers.TrainerWRF method) (credit.trainers.trainerWRF.Trainer method) (credit.trainers.trainerWRF_multi.Trainer method), [1] (credit.trainers.TrainerWRFMulti method), [1] (trainer_multistep_deprecated.Trainer method) (trainerERA5_multistep_concat.Trainer method) (trainerERA5_multistep_sum.Trainer method) (trainerERA5_multistep_v1.Trainer method) (trainerERA5_v1.Trainer method) (trainerERA5_v2.Trainer method) validate_args() (in module credit.parser) validate_data() (in module credit.ocean.samudra_data) validation (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) var_dict (credit.datasets.era5.ERA5Dataset attribute) var_levels (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) var_weights (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) VarDict (class in credit.datasets.datamap) variable_transforms (in module credit.animation) variable_weights() (in module credit.losses.weighted_loss) variables (_transforms.NormalizeState attribute) (_transforms.NormalizeState_Quantile attribute) (_transforms.NormalizeTendency attribute) (_transforms.ToTensor attribute) (credit.datasets.datamap.DataMap attribute) (credit.output_downscaling.OutputWriter attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.xr_sampler.XRSamplerByYear attribute) VariableTotalLoss2D (class in credit.losses) (class in credit.losses.weighted_loss) variance_scale (credit.ensemble.spherical.SphericalNoise attribute) varname_diagnostic (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_dyn_forcing (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_forcing (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) varname_static (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_surface (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_upper_air (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeLES attribute) (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.ToTensorLES attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) varname_upper_air_outside (credit.transforms.NormalizeWRF attribute) (credit.transforms.ToTensorWRF attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) varnum_diag (credit.ensemble.bred_vector.BredVector attribute) vars (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.metrics_downscaling.UnWeightedMetrics attribute) version (in module credit._version) version_tuple (in module credit._version) vrtdivspec() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) W w_lat (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.metrics_downscaling.UnWeightedMetrics attribute) w_var (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.metrics_downscaling.UnWeightedMetrics attribute) warmup_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) wavenum_init (credit.losses.power.PSDLoss attribute) (credit.losses.spectral.SpectralLoss2D attribute) weight (credit.models.graph.LayerNorm attribute) weighted_sum() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) WeightedRMSE (class in credit.replay_buffer) weights (credit.models.unet_diffusion.RandomOrLearnedSinusoidalPosEmb attribute) wet (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) wet_mask (credit.datasets.om4_multistep_batcher.StandardScaler attribute) wet_surface (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) WetMaskBlock (class in credit.postblock) (class in credit.postblock.wet_mask_samudra) which_forecast (credit.data.Predict_Dataset attribute) window_area (credit.models.swin.SwinTransformerV2CrBlock attribute) window_partition() (in module credit.models.swin) window_reverse() (in module credit.models.swin) window_size (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) WindowMultiHeadAttention (class in credit.models.swin) WindowMultiHeadAttentionNoPos (class in credit.models.swin) worker (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) worker() (in module credit.datasets.era5_multistep) (in module credit.datasets.sequential_multistep) (in module credit.datasets.wrfmultistep) worker_process() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) world_size (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) WRF_file_indices (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) WRF_input (credit.data.Sample_diag attribute) (credit.data.Sample_WRF attribute) WRF_target (credit.data.Sample_diag attribute) (credit.data.Sample_WRF attribute) WRFDataset (class in credit.datasets.wrf_singlestep) WRFMultiStep (class in credit.datasets.wrfmultistep) WRFPredict (class in credit.datasets.wrf_singlestep) WRFTransformer (class in credit.models) (class in credit.models.swin_wrf) write() (credit.output_downscaling.OutputWriter method) write_every (credit.skebs.SKEBS attribute) write_rollout_debug_files (credit.skebs.SKEBS attribute) write_to_netcdf() (credit.credit_ptype.CreditPostProcessor method) write_train_debug_files (credit.skebs.SKEBS attribute) WXFormer (class in credit.models) X xclassdict (credit.transforms_downscaling.DataTransforms attribute) XRSamplerByYear (class in credit.xr_sampler) XSigmoidLoss (class in credit.losses.xsigmoid) XTanhLoss (class in credit.losses.xtanh) Y years (credit.datasets.era5.ERA5Dataset attribute) Z z_h_ln (credit.models.graph.GateCell attribute) z_x_ln (credit.models.graph.GateCell attribute) zero_grad() (credit.models.checkpoint.OptimizerWrapper method) zero_out_levels_top_of_model (credit.skebs.SKEBS attribute) zonal_spectrum() (in module credit.verification.standard) Zscore (class in credit.transforms_downscaling) zstride (credit.datasets.datamap.DataMap attribute) (credit.output_downscaling.OutputWriter attribute)
