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__() (credit.metrics.LatWeightedMetrics method) (credit.metrics.LatWeightedMetricsClimatology method) (credit.metrics.LatWeightedMetricsEnsemble method) (credit.replay_buffer.TOADataLoader method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeState method) (credit.transforms.NormalizeState_Quantile method) (credit.transforms.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.ToTensor method) (credit.transforms.ToTensor_BridgeScaler method) (credit.transforms.ToTensor_ERA5_and_Forcing method) (credit.transforms404.NormalizeState method) (credit.transforms404.ToTensor method) (credit.xr_sampler.XRSamplerByYear method) (trainerERA5_v1.TOADataLoader method) __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.data404.CONUS404Dataset 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.realtime_predict.RealtimePredictDataset method) __hash__() (credit.regrid.Grid method) __iter__() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep.RepeatingIndexSampler 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) __len__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.data.Predict_Dataset method) (credit.data404.CONUS404Dataset 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.load_dataset_and_dataloader.BatchForecastLenDataLoader method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler method) (credit.datasets.realtime_predict.RealtimePredictDataset method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) __post_init__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.data404.CONUS404Dataset method) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) __repr__() (credit.models.graph.TransformerConv method) _align_phase_with() (in module credit.regrid) _alpha (credit.models.graph.TransformerConv attribute) _assert_increasing() (in module credit.regrid) _calc_window_shift() (credit.models.swin.SwinTransformerV2CrBlock method) _conservative_latitude_weights() (in module credit.regrid) _conservative_longitude_weights() (in module credit.regrid) _earth_padding() (credit.boundary_padding.TensorPadding method) _earth_unpad() (credit.boundary_padding.TensorPadding method) _fetch_batch() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) _kernel_crps_implementation() (credit.loss.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) _mean() (credit.regrid.ConservativeRegridder method) _mirror_padding() (credit.boundary_padding.TensorPadding method) _mirror_unpad() (credit.boundary_padding.TensorPadding method) _nanmean() (credit.regrid.ConservativeRegridder method) _nearest_neighbor_2d() (credit.regrid.NearestRegridder method) _periodic_lower_bounds() (in module credit.regrid) _periodic_overlap() (in module credit.regrid) _periodic_upper_bounds() (in module credit.regrid) _process_chunk() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) _relative_positional_encodings() (credit.models.swin.WindowMultiHeadAttention method) _shifted_window_attn() (credit.models.swin.SwinTransformerV2CrBlock method) _to_tuple() (credit.regrid.Grid method) 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) add() (credit.replay_buffer.ReplayBuffer method) add_gaussian_noise() (in module credit.ensemble.gaussian_noise) add_param_group() (credit.models.checkpoint.OptimizerWrapper method) add_pressure_and_geopotntial() (in module credit.nwp) 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) allvars (credit.transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) alpha_init (credit.skebs.SKEBS attribute) annealed_probability() (in module credit.scheduler) apply_spectral_norm() (in module credit.models.crossformer) (in module credit.models.fuxi) (in module credit.models.graph) (in module credit.models.swin) area (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) args (in module gfs_init) Array (in module credit.data) (in module credit.data404) (in module credit.regrid) Attention (class in credit.models.crossformer) attn (credit.models.swin.SwinTransformerV2CrBlock attribute) attn_drop (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) attn_type (credit.models.crossformer.Attention attribute) B b (credit.models.crossformer.LayerNorm attribute) (credit.models.crossformer.UpBlock attribute) (credit.models.fuxi.DownBlock attribute) (credit.models.fuxi.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_max_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) base_path (credit.transforms.NormalizeTendency attribute) (in module gfs_init) 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) batch_indices (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) batch_indices_splits (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) batch_size (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (rollout_to_netcdf.ForecastProcessor attribute) batched_forward (credit.loss.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) BatchForecastLenDataLoader (class in credit.datasets.load_dataset_and_dataloader) BatchForecastLenSampler (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.loss.KCRPSLoss attribute) BilinearRegridder (class in credit.regrid) blocks (credit.models.swin.SwinTransformerV2CrStage attribute) BridgescalerScaleState (class in credit.transforms) buffer_size (credit.replay_buffer.ReplayBuffer attribute) build_file_path() (in module credit.nwp) build_GFS_init() (in module credit.nwp) 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_ensemble) (in module rollout_ensemble_bred_vector) 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.models.crossformer) channels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion 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.skebs.SKEBS attribute) checkpoint_stages (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) circular_pad1d() (in module credit.models.swin) cleanup() (in module credit.replay_buffer) (in module credit.trainers.utils) (in module trainer_multistep_deprecated) climatology (credit.metrics.LatWeightedMetricsClimatology attribute) 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) 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) combine_data() (in module credit.nwp) compute_density() (in module credit.physics_core) compute_hlevs() (credit.physics_core.ModelLevelPressures method) compute_metrics() (in module rollout_ensemble) (in module rollout_ensemble_bred_vector) (in module rollout_metrics) compute_mlev_thickness() (credit.physics_core.ModelLevelPressures method) compute_p() (credit.physics_core.ModelLevelPressures method) 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) concat_for_inplace_ops() (in module credit.skebs) conf (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms404.ToTensor attribute) (in module credit.datasets.era5_singlestep) (rollout_to_netcdf.ForecastProcessor attribute) config (in module credit.xr_sampler) (in module gfs_init) (train.Objective attribute) (train_multistep.Objective attribute) config_file (in module credit.pbs) ConservativeRegridder (class in credit.regrid) CONUS404Dataset (class in credit.data404) conv (credit.models.crossformer.UpBlock attribute) (credit.models.fuxi.DownBlock attribute) (credit.models.fuxi.UpBlock attribute) (credit.skebs.BackscatterCNN attribute) convs (credit.models.crossformer.CrossEmbedLayer attribute) copy_checkpoint() (in module credit.models.checkpoint) cos_lat (credit.skebs.SKEBS attribute) CosineAnnealingWarmupRestarts (class in credit.scheduler) 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) create_diffusion() (in module credit.models.crossformer_diffusion) create_model() (in module credit.models.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.animation module credit.boundary_padding module credit.data module credit.data404 module credit.datasets module credit.datasets.era5_multistep module credit.datasets.era5_multistep_batcher module credit.datasets.era5_singlestep module credit.datasets.load_dataset_and_dataloader module credit.datasets.realtime_predict module credit.datasets.sequential_multistep module credit.distributed module credit.ensemble module credit.ensemble.bred_vector module credit.ensemble.crps module credit.ensemble.gaussian_noise module credit.forecast module credit.interp module credit.loss module credit.loss404 module credit.metadata module credit.metrics module credit.mixed_precision module credit.models module credit.models.base_model module credit.models.checkpoint module credit.models.crossformer module credit.models.crossformer_diffusion module credit.models.crossformer_ensemble module credit.models.debugger_model module credit.models.fuxi module credit.models.graph module credit.models.reset module credit.models.swin module credit.models.unet module credit.models.unet404 module credit.nwp module credit.output module credit.parser module credit.pbs module credit.physics_constants module credit.physics_core module credit.pol_lapdiff_filt module credit.postblock module credit.regrid module credit.replay_buffer module credit.scheduler module credit.seed module credit.skebs module credit.solar module credit.trainers module credit.trainers.base_trainer module credit.trainers.trainer404 module credit.trainers.trainerERA5 module credit.trainers.trainerERA5_ensemble module credit.trainers.utils module credit.transforms module credit.transforms404 module credit.visualization_tools module credit.xr_sampler module credit_grid (in module gfs_init) credit_main_parser() (in module credit.parser) CrossEmbedLayer (class in credit.models.crossformer) CrossFormer (class in credit.models) (class in credit.models.crossformer) crossformer_config (in module credit.models.crossformer_diffusion) CrossFormerDiffusion (class in credit.models.crossformer_diffusion) CrossFormerWithNoise (class in credit.models) (class in credit.models.crossformer_ensemble) cube_embedding (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) CubeEmbedding (class in credit.models.crossformer) (class in credit.models.fuxi) 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.sequential_multistep.DistributedSequentialDataset attribute) current_file_year (credit.xr_sampler.XRSamplerByYear attribute) current_index (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) cycle (credit.scheduler.CosineAnnealingWarmupRestarts attribute) cycle() (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) dataset (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.replay_buffer.ReplayBuffer attribute) dataset_id (in module credit.datasets.load_dataset_and_dataloader) date (in module gfs_init) datetime_index (credit.data.Sample attribute) days_of_year (trainerERA5_v1.TOADataLoader 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.crossformer_ensemble.CrossFormerWithNoise attribute) (credit.models.CrossFormerWithNoise attribute) default() (in module credit.models.crossformer_diffusion) defaults (credit.models.checkpoint.OptimizerWrapper property) depth (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) description (in module model_summary) (in module predict404) (in module rollout_ens_batcher) (in module rollout_ensemble) (in module rollout_ensemble_bred_vector) (in module rollout_metrics) (in module rollout_realtime) (in module rollout_to_netcdf) (in module train) (in module train404) (in module train_multistep) device (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (in module credit.pol_lapdiff_filt) (rollout_to_netcdf.ForecastProcessor attribute) (train.Objective attribute) (train_multistep.Objective attribute) (trainer_multistep_deprecated.Trainer 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.models.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.swin.SwinTransformerV2CrBlock attribute) dim_head (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) dissipation_scaling_coefficient (credit.skebs.SKEBS attribute) dissipation_type (credit.skebs.SKEBS attribute) distributed_model_wrapper() (in module credit.distributed) DistributedSequentialDataset (class in credit.datasets.sequential_multistep) DistributedSequentialDatasetBasic (class in credit.datasets.sequential_multistep) down (credit.models.fuxi.UTransformer attribute) DownBlock (class in credit.models.fuxi) downscale (credit.models.swin.SwinTransformerV2CrStage attribute) dpb (credit.models.crossformer.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.models.crossformer.Attention attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.graph.TransformerConv attribute) (credit.models.GraphResTransfGRU attribute) ds_read_and_subset() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) dtype (credit.replay_buffer.ReplayBuffer attribute) DynamicPositionBias (class in credit.models.crossformer) E edge_dim (credit.models.graph.TransformerConv attribute) edge_path (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) embed_dim (credit.models.crossformer.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) enc_norm (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) encoder (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) encoder_noise (credit.models.crossformer_ensemble.CrossFormerWithNoise attribute), [1] (credit.models.CrossFormerWithNoise attribute), [1] encoder_noise_factor (credit.models.crossformer_ensemble.CrossFormerWithNoise attribute) (credit.models.CrossFormerWithNoise attribute) ensemble_size (rollout_to_netcdf.ForecastProcessor attribute) ensure_numpy_datetime() (in module credit.data) eps (credit.models.crossformer.LayerNorm attribute) (credit.models.graph.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) evapor_ind (credit.postblock.GlobalWaterFixer attribute) exists() (in module credit.models.crossformer_diffusion) extract() (in module credit.models.crossformer_diffusion) extract_month_day_hour() (in module credit.data) F fc (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi 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.crossformer) figure_panel_planner() (in module credit.visualization_tools) filename (in module credit.datasets.era5_multistep) filename_diagnostic (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) filename_dyn_forcing (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset 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.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset 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.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) filename_surface (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) filenames (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.xr_sampler.XRSamplerByYear attribute) find_common_indices() (in module credit.data) 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) finish (credit.data404.CONUS404Dataset attribute) first_cycle_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) fit() (credit.trainers.base_trainer.BaseTrainer method) (credit.trainers.Trainer404 method) (credit.trainers.trainer404.Trainer 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.ToTensor_ERA5_and_Forcing attribute) flag_dyn_forcing (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) flag_forcing (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) flag_static (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) flag_surface (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) flag_upper_air (credit.transforms.ToTensor_ERA5_and_Forcing attribute) flatten() (in module credit.data404) flatten_list() (in module credit.data) for_len (credit.transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms404.ToTensor 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.data404.CONUS404Dataset 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.load_dataset_and_dataloader.BatchForecastLenDataLoader attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.skebs.SKEBS 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) forecast_step_counts (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) forecast_times (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_timestep (credit.datasets.realtime_predict.RealtimePredictDataset attribute) ForecastProcessor (class in rollout_to_netcdf) format_data() (in module credit.nwp) format_datetime() (in module credit.nwp) forward() (credit.loss.KCRPSLoss method) (credit.loss.LogCoshLoss method) (credit.loss.MSLELoss method) (credit.loss.PSDLoss method) (credit.loss.SpectralLoss2D method) (credit.loss.VariableTotalLoss2D method) (credit.loss.XSigmoidLoss method) (credit.loss.XTanhLoss method) (credit.loss404.LogCoshLoss method) (credit.loss404.MSLELoss method) (credit.loss404.PSDLoss method) (credit.loss404.SpectralLoss2D method) (credit.loss404.VariableTotalLoss2D method) (credit.loss404.XSigmoidLoss method) (credit.loss404.XTanhLoss 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.crossformer_diffusion.CrossFormerDiffusion method) (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion method) (credit.models.crossformer_ensemble.CrossFormerWithNoise method) (credit.models.crossformer_ensemble.PixelNoiseInjection method), [1] (credit.models.CrossFormerWithNoise method) (credit.models.debugger_model.DebuggerModel method) (credit.models.DebuggerModel 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.SegmentationModel method) (credit.models.SegmentationModel404 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.SwinTransformerV2Cr method) (credit.models.unet.SegmentationModel method) (credit.models.unet404.SegmentationModel404 method) (credit.postblock.GlobalEnergyFixer method) (credit.postblock.GlobalMassFixer method) (credit.postblock.GlobalWaterFixer method) (credit.postblock.PostBlock method) (credit.postblock.TracerFixer 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.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.SegmentationModel attribute) (credit.models.SegmentationModel404 attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet404.SegmentationModel404 attribute) (credit.skebs.SKEBS attribute) freeze (credit.models.crossformer_ensemble.CrossFormerWithNoise attribute) (credit.models.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.crossformer.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) gdas_base_path (in module gfs_init) 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) 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() (credit.data404.CONUS404Dataset method) get_file_extension() (in module credit.models.checkpoint) get_forward_data() (in module credit.data) (in module credit.data404) get_lr() (credit.scheduler.CosineAnnealingWarmupRestarts method) get_pad2d() (in module credit.models.fuxi) get_pad3d() (in module credit.models.fuxi) get_projection() (in module credit.visualization_tools) get_psd() (credit.loss.PSDLoss method) (credit.loss404.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_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) 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_init (in module gfs_init) gfs_map (in module credit.nwp) GlobalEnergyFixer (class in credit.postblock) GlobalMassFixer (class in credit.postblock) GlobalWaterFixer (class in credit.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_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) havg (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) head (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) heads (credit.models.crossformer.Attention attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.graph.TransformerConv attribute) (credit.models.GraphResTransfGRU attribute) hidden_size (credit.models.graph.GateCell attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) hist_len (credit.transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms.ToTensor_ERA5_and_Forcing attribute) (credit.transforms404.ToTensor 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.data404.CONUS404Dataset 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) (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion 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) I identity() (in module credit.models.crossformer_diffusion) IMAGE_ATTR_NAMES (in module credit.data) (in module credit.data404) image_height (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) (in module credit.models.crossformer) (in module credit.models.debugger_model) (in module credit.models.fuxi) (in module credit.models.swin) image_width (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) img_size (credit.models.crossformer.CubeEmbedding 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.SwinTransformerV2Cr 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) 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) init_lr() (credit.scheduler.CosineAnnealingWarmupRestarts method) 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) initialize_batch() (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_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_channels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) input_only_channels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.skebs.SKEBS attribute) input_resolution (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) input_shape (credit.replay_buffer.ReplayBuffer attribute) input_size (credit.models.graph.GraphResidualBlock attribute) integral (credit.physics_core.physics_hybrid_sigma_level attribute) interp_geopotential_to_pressure_levels() (in module credit.interp) interp_hybrid_to_height_agl() (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_to_model_level() (in module credit.nwp) inverse_transform() (credit.transforms.BridgescalerScaleState method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeState method) (credit.transforms.NormalizeState_Quantile method) (credit.transforms.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.NormalizeTendency method) (credit.transforms404.NormalizeState method) inverse_transform_input() (credit.transforms.Normalize_ERA5_and_Forcing method) 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.loss) keep_dataset_vars() (in module credit.data) kgkg_to_gkg() (in module credit.animation) L lat (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) (credit.regrid.Grid attribute) lat_weights (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) (credit.replay_buffer.WeightedRMSE attribute) latititude_weights() (in module credit.loss404) latitude_weights() (in module credit.loss) latlons (rollout_to_netcdf.ForecastProcessor attribute) latN (credit.transforms.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.fuxi.UTransformer attribute) LayerNorm (class in credit.models.crossformer) (class in credit.models.graph) layers (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.DynamicPositionBias attribute) (credit.models.crossformer.FeedForward attribute) (credit.models.crossformer.Transformer attribute) lazymerge() (in module credit.data404) lead_time_periods (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (rollout_to_netcdf.ForecastProcessor attribute) len (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) level_map (in module credit.nwp) levels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer 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.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.NormalizeState attribute) (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.ToTensor_BridgeScaler attribute) LH_ICE (in module credit.physics_constants) LH_WATER (in module credit.physics_constants) 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) 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 train404) 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_graph() (credit.models.graph.GraphResTransfGRU method) (credit.models.GraphResTransfGRU method) load_inputs() (credit.replay_buffer.ReplayBuffer method) load_loss() (in module credit.loss) (in module credit.loss404) load_metadata() (in module credit.output) load_model() (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 train404) (in module train_multistep) load_premade_encoder_model() (in module credit.models.unet) (in module credit.models.unet404) (in module credit.skebs) 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_transforms() (in module credit.transforms) 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) LogCoshLoss (class in credit.loss) (class in credit.loss404) logger (in module credit.datasets) (in module credit.datasets.era5_multistep) (in module credit.datasets.era5_multistep_batcher) (in module credit.datasets.sequential_multistep) (in module credit.ensemble.bred_vector) (in module credit.forecast) (in module credit.loss) (in module credit.loss404) (in module credit.metrics) (in module credit.models) (in module credit.models.base_model) (in module credit.models.crossformer) (in module credit.models.crossformer_ensemble) (in module credit.models.debugger_model) (in module credit.models.fuxi) (in module credit.models.swin) (in module credit.models.unet) (in module credit.models.unet404) (in module credit.output) (in module credit.pbs) (in module credit.postblock) (in module credit.skebs) (in module credit.trainers) (in module credit.trainers.base_trainer) (in module credit.trainers.trainerERA5) (in module credit.trainers.trainerERA5_ensemble) (in module credit.transforms) (in module credit.transforms404) (in module credit.visualization_tools) (in module predict404) (in module rollout_ens_batcher) (in module rollout_ensemble) (in module rollout_ensemble_bred_vector) (in module rollout_metrics) (in module rollout_realtime) (in module rollout_to_netcdf) (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) lonN (credit.transforms.ToTensor_BridgeScaler attribute) lons (in module credit.solar) 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 graph_edges) (in module graph_edges_knn) (in module model_summary) (in module scaler) (in module train) (in module train404) (in module train_multistep) make_xarray() (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) 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) max_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) mean (credit.transforms.NormalizeTendency attribute) mean_ds (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeState attribute) (credit.transforms404.NormalizeState attribute) mean_sea_level_pressure() (in module credit.interp) mean_tensors (credit.transforms.Normalize_ERA5_and_Forcing attribute) merge_linear (credit.models.graph.GraphResidualBlock attribute) message() (credit.models.graph.TransformerConv method) meta_data (rollout_to_netcdf.ForecastProcessor attribute) meta_mlp (credit.models.swin.WindowMultiHeadAttention attribute) metric (train.Objective attribute) (train_multistep.Objective attribute) metric_fn (credit.replay_buffer.ReplayBuffer attribute) min_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) mlp (credit.models.swin.SwinTransformerV2CrBlock attribute) mmax (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.SKEBS attribute) mode (credit.boundary_padding.TensorPadding attribute) model (credit.models.checkpoint.FSDPOptimizerWrapper attribute) (credit.models.SegmentationModel attribute) (credit.models.SegmentationModel404 attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet404.SegmentationModel404 attribute) (credit.skebs.BackscatterUnet attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (in module credit.models.checkpoint) (trainer_multistep_deprecated.Trainer attribute) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) model_and_memory_summary() (in module train404) model_level_indices (in module gfs_init) model_levels (in module gfs_init) model_predictions() (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion method) model_summary module model_types (in module credit.models) ModelLevelPressures (class in credit.physics_core) ModelPrediction (class in credit.models.crossformer_diffusion) ModelWrapper (class in credit.models.checkpoint) ModifiedGaussianDiffusion (class in credit.models.crossformer_diffusion) modulation (credit.models.crossformer_ensemble.PixelNoiseInjection attribute), [1] module calc_global_solar credit credit.animation credit.boundary_padding credit.data credit.data404 credit.datasets credit.datasets.era5_multistep credit.datasets.era5_multistep_batcher credit.datasets.era5_singlestep credit.datasets.load_dataset_and_dataloader credit.datasets.realtime_predict credit.datasets.sequential_multistep credit.distributed credit.ensemble credit.ensemble.bred_vector credit.ensemble.crps credit.ensemble.gaussian_noise credit.forecast credit.interp credit.loss credit.loss404 credit.metadata credit.metrics credit.mixed_precision credit.models credit.models.base_model credit.models.checkpoint credit.models.crossformer credit.models.crossformer_diffusion credit.models.crossformer_ensemble credit.models.debugger_model credit.models.fuxi credit.models.graph credit.models.reset credit.models.swin credit.models.unet credit.models.unet404 credit.nwp credit.output credit.parser credit.pbs credit.physics_constants credit.physics_core credit.pol_lapdiff_filt credit.postblock credit.regrid credit.replay_buffer credit.scheduler credit.seed credit.skebs credit.solar credit.trainers credit.trainers.base_trainer credit.trainers.trainer404 credit.trainers.trainerERA5 credit.trainers.trainerERA5_ensemble credit.trainers.utils credit.transforms credit.transforms404 credit.visualization_tools credit.xr_sampler gfs_init graph_edges graph_edges_knn model_summary predict404 rollout_ens_batcher rollout_ensemble rollout_ensemble_bred_vector rollout_metrics rollout_realtime rollout_to_netcdf scaler train train404 train_multistep 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) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) mse() (credit.metrics.LatWeightedMetricsClimatology method) MSLELoss (class in credit.loss) (class in credit.loss404) MultiprocessingBatcher (class in credit.datasets.era5_multistep_batcher) MultiprocessingBatcherPrefetch (class in credit.datasets.era5_multistep_batcher) multistep (credit.skebs.SKEBS attribute) N n_3dvar_levels (credit.transforms.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) 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) 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) 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) noise_factor (credit.models.crossformer_ensemble.PixelNoiseInjection attribute), [1] noise_inject1 (credit.models.crossformer_ensemble.CrossFormerWithNoise attribute) (credit.models.CrossFormerWithNoise attribute) noise_inject2 (credit.models.crossformer_ensemble.CrossFormerWithNoise attribute) (credit.models.CrossFormerWithNoise attribute) noise_inject3 (credit.models.crossformer_ensemble.CrossFormerWithNoise attribute) (credit.models.CrossFormerWithNoise attribute) noise_latent_dim (credit.models.crossformer_ensemble.CrossFormerWithNoise attribute), [1] (credit.models.CrossFormerWithNoise attribute), [1] noise_transform (credit.models.crossformer_ensemble.PixelNoiseInjection attribute), [1] norm (credit.models.crossformer.Attention attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.PatchMerging 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_ERA5_and_Forcing (class in credit.transforms) normalize_to_neg_one_to_one() (in module credit.models.crossformer_diffusion) NormalizeState (class in credit.transforms) (class in credit.transforms404) NormalizeState_Quantile (class in credit.transforms) NormalizeState_Quantile_Bridgescalar (class in credit.transforms) NormalizeTendency (class in credit.transforms) now_date (in module gfs_init) 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) num_heads (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) num_indices_per_rank (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) num_patches (credit.models.swin.PatchEmbed attribute) num_replicas (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) num_upper_air (credit.transforms.Normalize_ERA5_and_Forcing 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 train) (class in train404) (class in train_multistep) omega (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) OMEGA (in module credit.physics_constants) one_shot (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data404.CONUS404Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.transforms.ToTensor_BridgeScaler attribute) operations (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.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) out_size (credit.models.graph.GraphResidualBlock attribute) output_channels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.UpBlock attribute) output_dtype (credit.transforms.ToTensor_ERA5_and_Forcing attribute) output_only_channels (credit.models.debugger_model.DebuggerModel attribute) (credit.models.DebuggerModel attribute) (credit.skebs.SKEBS attribute) P p_losses() (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion method) p_mean_variance() (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion method) p_sample() (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion method) p_sample_loop() (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion method) pad (credit.models.fuxi.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.fuxi.UTransformer attribute) param_groups (credit.models.checkpoint.OptimizerWrapper property) parameters (credit.models.checkpoint.OptimizerWrapper property) parse_dtype() (in module credit.mixed_precision) parser (in module gfs_init) patch_embed (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) patch_height (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) patch_size (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) patch_width (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) PatchEmbed (class in credit.models.swin) patches_resolution (credit.models.crossformer.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) PatchMerging (class in credit.models.swin) 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) PixelNoiseInjection (class in credit.models.crossformer_ensemble) 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) populate() (credit.replay_buffer.ReplayBuffer method) post_conf (credit.skebs.SKEBS attribute) PostBlock (class in credit.postblock) precip_ind (credit.postblock.GlobalWaterFixer attribute) pred_noise (credit.models.crossformer_diffusion.ModelPrediction attribute) pred_x_start (credit.models.crossformer_diffusion.ModelPrediction attribute) predict() (in module predict404) (in module rollout_ens_batcher) (in module rollout_ensemble) (in module rollout_ensemble_bred_vector) (in module rollout_metrics) (in module rollout_realtime) (in module rollout_to_netcdf) predict404 module predict_data_check() (in module credit.parser) Predict_Dataset (class in credit.data) Predict_Dataset_Batcher (class in credit.datasets.era5_multistep_batcher) 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) process() (rollout_to_netcdf.ForecastProcessor method) process_forecast() (in module rollout_realtime) proj (credit.models.crossformer.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) proj_drop (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) projections (in module credit.animation) PSDLoss (class in credit.loss) (class in credit.loss404) ptr (credit.replay_buffer.ReplayBuffer attribute) Q q_ind_end (credit.postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalMassFixer attribute) (credit.postblock.GlobalWaterFixer attribute) q_ind_start (credit.postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalMassFixer attribute) (credit.postblock.GlobalWaterFixer attribute) Q_inds (credit.skebs.SKEBS attribute) 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) 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.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (in module credit.models.reset) (trainer_multistep_deprecated.Trainer attribute) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) RDGAS (in module credit.physics_constants) RealtimePredictDataset (class in credit.datasets.realtime_predict) reduction (credit.loss.LogCoshLoss attribute) (credit.loss.MSLELoss attribute) (credit.loss.SpectralLoss2D attribute) (credit.loss.XSigmoidLoss attribute) (credit.loss.XTanhLoss attribute) (credit.loss404.LogCoshLoss attribute) (credit.loss404.MSLELoss attribute) (credit.loss404.SpectralLoss2D attribute) (credit.loss404.XSigmoidLoss attribute) (credit.loss404.XTanhLoss attribute) (credit.models.swin.PatchMerging attribute) 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) Regridder (class in credit.regrid) 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) RepeatingIndexSampler (class in credit.datasets.era5_multistep) ReplayBuffer (class in credit.replay_buffer) reset_classifier() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) reset_model() (in module credit.models.reset) reset_parameters() (credit.models.graph.LayerNorm method) (credit.models.graph.TransformerConv 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) results (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher attribute) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) retain_graph (credit.skebs.SKEBS attribute) RHO_WATER (in module credit.physics_constants) rk4() (credit.models.CrossFormer method) (credit.models.crossformer.CrossFormer method) (credit.models.SegmentationModel method) (credit.models.unet.SegmentationModel method) rk4_integration (credit.models.SegmentationModel attribute) (credit.models.unet.SegmentationModel attribute) rmse() (credit.metrics.LatWeightedMetricsClimatology method) rmse_scores (credit.replay_buffer.ReplayBuffer attribute) 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.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) rollout_ens_batcher module rollout_ensemble module rollout_ensemble_bred_vector module rollout_metrics module rollout_p (credit.data.Predict_Dataset attribute) rollout_realtime module rollout_to_netcdf module root_weight (credit.models.graph.TransformerConv attribute) RVGAS (in module credit.physics_constants) S Sample (class in credit.data) (class in credit.data404) sample() (credit.models.crossformer_diffusion.ModifiedGaussianDiffusion method) (credit.replay_buffer.ReplayBuffer method) sampler (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher 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_increment() (in module credit.output) save_state_dict() (in module credit.models.checkpoint) save_unsharded_model() (credit.models.checkpoint.TorchFSDPCheckpointIO method) save_unsharded_optimizer() (credit.models.checkpoint.TorchFSDPCheckpointIO method) scale (credit.models.crossformer.Attention attribute) scale_loss() (credit.models.checkpoint.OptimizerWrapper method) scaler module scaler_3d (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) scaler_3ds (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) scaler_df (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) scaler_file (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) scaler_surf (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) scaler_surfs (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) scalers (in module scaler) seed (credit.data404.CONUS404Dataset 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.realtime_predict.RealtimePredictDataset attribute) seed_everything() (in module credit.seed) SegmentationModel (class in credit.models) (class in credit.models.unet) SegmentationModel404 (class in credit.models) (class in credit.models.unet404) self_condition (credit.models.crossformer_diffusion.CrossFormerDiffusion attribute) sequential_attn (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) 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.sequential_multistep.DistributedSequentialDataset method) set_globals() (in module credit.datasets) set_grad_checkpointing() (credit.models.swin.SwinTransformerV2Cr method) (credit.models.SwinTransformerV2Cr method) setup() (in module credit.distributed) (in module model_summary) (in module train404) setup_data_loading() (in module credit.datasets) shape (credit.regrid.Grid property) shared_mem_draw_wrapper() (in module credit.visualization_tools) 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.sequential_multistep.DistributedSequentialDataset attribute) sigmoid (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.BackscatterCNN attribute) single_sample_forward() (credit.loss.KCRPSLoss method) size (credit.replay_buffer.ReplayBuffer 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.data404.CONUS404Dataset 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) slice_x (credit.transforms404.ToTensor attribute) slice_y (credit.transforms404.ToTensor attribute) source (credit.regrid.Regridder attribute) sp_index (credit.skebs.SKEBS attribute) spec2grid() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) spec_coef_is_initialized (credit.skebs.SKEBS attribute) SpectralLoss2D (class in credit.loss) (class in credit.loss404) split_and_reshape() (credit.models.base_model.BaseModel method) (in module credit.output) 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) start (credit.data404.CONUS404Dataset attribute) state_dict() (credit.models.checkpoint.OptimizerWrapper method) state_trans (credit.skebs.SKEBS attribute) state_vars (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) static_inds (credit.skebs.SKEBS attribute) static_variables (credit.models.SegmentationModel404 attribute) (credit.models.unet404.SegmentationModel404 attribute) (credit.transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms404.ToTensor attribute) std (credit.transforms.NormalizeTendency attribute) std_ds (credit.transforms.Normalize_ERA5_and_Forcing attribute) (credit.transforms.NormalizeState attribute) (credit.transforms404.NormalizeState attribute) std_tensors (credit.transforms.Normalize_ERA5_and_Forcing attribute) step() (credit.models.checkpoint.OptimizerWrapper method) (credit.scheduler.CosineAnnealingWarmupRestarts method) step_in_cycle (credit.scheduler.CosineAnnealingWarmupRestarts attribute) stop_event (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) stop_signal (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) supported_encoders (in module credit.models.unet) (in module credit.models.unet404) supported_models (in module credit.models.unet) (in module credit.models.unet404) (in module credit.skebs) surf_LH_ind (credit.postblock.GlobalEnergyFixer attribute) surf_LR_ind (credit.postblock.GlobalEnergyFixer attribute) surf_SH_ind (credit.postblock.GlobalEnergyFixer attribute) surf_solar_ind (credit.postblock.GlobalEnergyFixer attribute) surface (in module credit.nwp) surface_channels (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer 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.skebs.SKEBS attribute) surface_files (credit.datasets.realtime_predict.RealtimePredictDataset attribute) surface_pressure (credit.physics_core.physics_hybrid_sigma_level attribute) surface_variables (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState attribute) (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.NormalizeTendency attribute) (credit.transforms.ToTensor attribute) (credit.transforms.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.GlobalEnergyFixer attribute) T_ind_start (credit.postblock.GlobalEnergyFixer attribute) T_inds (credit.skebs.SKEBS attribute) target (credit.regrid.Regridder attribute) target_ERA5_images (credit.data.Sample attribute) target_shift_size (credit.models.swin.SwinTransformerV2CrBlock attribute) TensorPadding (class in credit.boundary_padding) testC4loader() (in module credit.data404) time_mlp (credit.models.crossformer_diffusion.CrossFormerDiffusion attribute) time_steps (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) times_b (credit.replay_buffer.TOADataLoader attribute) (trainerERA5_v1.TOADataLoader attribute) to() (credit.models.graph.GraphResTransfGRU method) (credit.models.GraphResTransfGRU method) to_out (credit.models.crossformer.Attention attribute) to_qkv (credit.models.crossformer.Attention attribute) TOA (credit.replay_buffer.TOADataLoader attribute) (trainerERA5_v1.TOADataLoader attribute) TOA_OLR_ind (credit.postblock.GlobalEnergyFixer attribute) TOA_solar_ind (credit.postblock.GlobalEnergyFixer attribute) TOADataLoader (class in credit.replay_buffer) (class in trainerERA5_v1) 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_dry_air_mass() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) total_length (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep 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.sequential_multistep.DistributedSequentialDataset attribute) ToTensor (class in credit.transforms) (class in credit.transforms404) ToTensor_BridgeScaler (class in credit.transforms) ToTensor_ERA5_and_Forcing (class in credit.transforms) tracer_indices (credit.postblock.TracerFixer attribute) tracer_thres (credit.postblock.TracerFixer attribute) TracerFixer (class in credit.postblock) train module train() (train.Objective method) (train404.Objective method) (train_multistep.Objective method) train404 module train_alpha (credit.skebs.SKEBS attribute) train_multistep module train_one_epoch() (credit.replay_buffer.Trainer method) (credit.trainers.base_trainer.BaseTrainer method) (credit.trainers.Trainer404 method) (credit.trainers.trainer404.Trainer method) (credit.trainers.TrainerEnsemble method) (credit.trainers.TrainerERA5 method) (credit.trainers.trainerERA5.Trainer method) (credit.trainers.trainerERA5_ensemble.Trainer 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) Trainer (class in credit.replay_buffer) (class in credit.trainers.trainer404) (class in credit.trainers.trainerERA5) (class in credit.trainers.trainerERA5_ensemble) (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_multistep_concat module trainerERA5_multistep_sum module trainerERA5_multistep_v1 module trainerERA5_v1 module trainerERA5_v2 module training_data_check() (in module credit.parser) training_loss (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) transform (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.data404.CONUS404Dataset 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) transform() (credit.transforms.BridgescalerScaleState method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeState method) (credit.transforms.NormalizeState_Quantile method) (credit.transforms.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.NormalizeTendency method) (credit.transforms404.NormalizeState method) transform_array() (credit.transforms.BridgescalerScaleState method) (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeState method) transform_dataset() (credit.transforms.Normalize_ERA5_and_Forcing method) (credit.transforms.NormalizeState method) transform_era5_times() (in module scaler) Transformer (class in credit.models.crossformer) transformer (credit.models.graph.GraphResidualBlock attribute) TransformerConv (class in credit.models.graph) tropics_only_dissipation (credit.skebs.SKEBS attribute) tsi_data (in module credit.solar) U U_ind_end (credit.postblock.GlobalEnergyFixer attribute) U_ind_start (credit.postblock.GlobalEnergyFixer attribute) U_inds (credit.skebs.SKEBS attribute) u_transformer (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) unpad() (credit.boundary_padding.TensorPadding method) (credit.skebs.BackscatterCNN method) unscale_grad() (credit.models.checkpoint.OptimizerWrapper method) 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.fuxi.UTransformer attribute) up_block1 (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) up_block2 (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) up_block3 (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) up_block4 (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) UpBlock (class in credit.models.crossformer) (class in credit.models.fuxi) 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) use_edge_attr (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.GraphResTransfGRU attribute) use_interp (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) use_padding (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.Fuxi attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) use_post_block (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer 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.swin.SwinTransformerV2Cr attribute) (credit.models.SwinTransformerV2Cr attribute) (credit.models.unet.SegmentationModel attribute) use_power_loss (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) use_spectral_loss (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) use_spectral_norm (credit.models.CrossFormer attribute) (credit.models.crossformer.CrossFormer 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.SwinTransformerV2Cr attribute) use_statics (credit.skebs.SKEBS attribute) UTransformer (class in credit.models.fuxi) V V_ind_end (credit.postblock.GlobalEnergyFixer attribute) V_ind_start (credit.postblock.GlobalEnergyFixer attribute) V_inds (credit.skebs.SKEBS attribute) valid_forecast_len (credit.skebs.SKEBS attribute) validate() (credit.trainers.base_trainer.BaseTrainer method) (credit.trainers.Trainer404 method) (credit.trainers.trainer404.Trainer method) (credit.trainers.TrainerEnsemble method) (credit.trainers.TrainerERA5 method) (credit.trainers.trainerERA5.Trainer method) (credit.trainers.trainerERA5_ensemble.Trainer 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) validation (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) var_levels (credit.transforms.BridgescalerScaleState attribute) var_weights (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) variable_transforms (in module credit.animation) variable_weights() (in module credit.loss) (in module credit.loss404) variables (credit.models.SegmentationModel404 attribute) (credit.models.unet404.SegmentationModel404 attribute) (credit.transforms.BridgescalerScaleState attribute) (credit.transforms.NormalizeState attribute) (credit.transforms.NormalizeState_Quantile attribute) (credit.transforms.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.NormalizeTendency attribute) (credit.transforms.ToTensor attribute) (credit.transforms.ToTensor_BridgeScaler attribute) (credit.transforms404.NormalizeState attribute) (credit.transforms404.ToTensor attribute) (credit.xr_sampler.XRSamplerByYear attribute) (in module gfs_init) VariableTotalLoss2D (class in credit.loss) (class in credit.loss404) 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.ToTensor_ERA5_and_Forcing attribute) varnames (credit.data404.CONUS404Dataset attribute) vars (credit.loss.VariableTotalLoss2D attribute) (credit.loss404.VariableTotalLoss2D attribute) (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) 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) w_var (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) warmup_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) wavenum_init (credit.loss.PSDLoss attribute) (credit.loss.SpectralLoss2D attribute) (credit.loss404.PSDLoss attribute) (credit.loss404.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) 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.crossformer.Attention attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) (credit.models.SwinTransformerV2Cr 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) worker() (in module credit.datasets.era5_multistep) (in module credit.datasets.sequential_multistep) 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.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) write_every (credit.skebs.SKEBS attribute) write_rollout_debug_files (credit.skebs.SKEBS attribute) write_train_debug_files (credit.skebs.SKEBS attribute) X x (credit.data404.Sample attribute) XRSamplerByYear (class in credit.xr_sampler) XSigmoidLoss (class in credit.loss) (class in credit.loss404) XTanhLoss (class in credit.loss) (class in credit.loss404) Y y (credit.data404.Sample attribute) Z z_h_ln (credit.models.graph.GateCell attribute) z_x_ln (credit.models.graph.GateCell attribute) zarrpath (credit.data404.CONUS404Dataset attribute) zero_grad() (credit.models.checkpoint.OptimizerWrapper method) zero_out_levels_top_of_model (credit.skebs.SKEBS attribute)
