Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Y | Z _ __call__() (_transforms.NormalizeState method) (_transforms.NormalizeState_Quantile method) (_transforms.ToTensor method) (credit.ensemble.bred_vector.BredVector method) (credit.ensemble.color.ColorNoise method) (credit.ensemble.gaussian.GaussianNoise method) (credit.ensemble.spherical.SphericalNoise method) (credit.ensemble.temporal.TemporalNoise method) (credit.metrics.LatWeightedMetrics method) (credit.metrics.LatWeightedMetricsClimatology method) (credit.metrics.LatWeightedMetricsEnsemble method) (credit.metrics_downscaling.UnWeightedMetrics method) (credit.replay_buffer.TOADataLoader method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_les.ToTensorLES method) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler method) (credit.transforms.transforms_wrf.NormalizeWRF method) (credit.transforms.transforms_wrf.ToTensorWRF method) (credit.transforms_downscaling.Clip method) (credit.transforms_downscaling.DataTransforms method) (credit.transforms_downscaling.Expand method) (credit.transforms_downscaling.Identity method) (credit.transforms_downscaling.Minmax method) (credit.transforms_downscaling.Pad method) (credit.transforms_downscaling.Power method) (credit.transforms_downscaling.Zscore method) (credit.xr_sampler.XRSamplerByYear method) (trainerERA5_v1.TOADataLoader method) __commit_id__ (in module credit._version) __del__() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) __enter__() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) __eq__() (credit.regrid.Grid method) __exit__() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) __getattr__() (in module credit.datasets) (in module credit.models) (in module credit.transforms) __getitem__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) (credit.datasets.datamap.DataMap method) (credit.datasets.downscaling_dataset.DownscalingDataset method) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.era5_singlestep.ERA5_and_Forcing_SingleStep method) (credit.datasets.les_singlestep.LESDataset method) (credit.datasets.multi_source.MultiSourceDataset method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.realtime_predict.RealtimePredictDataset method) (credit.datasets.wrf_singlestep.WRFDataset method) (credit.datasets.wrfmultistep.WRFMultiStep method) __hash__() (credit.regrid.Grid method) __iter__() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep.RepeatingIndexSampler method) (credit.datasets.les_singlestep.LESPredict method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.sequential_multistep.DistributedSequentialDatasetBasic method) (credit.datasets.wrf_singlestep.WRFPredict method) (credit.datasets.wrfmultistep.RepeatingIndexSampler method) (credit.samplers.DistributedMultiStepBatchSampler method) (credit.samplers.MultiStepBatchSamplerSubset method) (credit.trainers.ic_optimization.TimeStepper method) __len__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.data.Predict_Dataset method) (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) (credit.datasets.datamap.DataMap method) (credit.datasets.downscaling_dataset.DownscalingDataset method) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep.RepeatingIndexSampler method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.les_singlestep.LESDataset method) (credit.datasets.les_singlestep.LESPredict method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader method) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler method) (credit.datasets.multi_source.MultiSourceDataset method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) (credit.datasets.realtime_predict.RealtimePredictDataset method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.wrf_singlestep.WRFDataset method) (credit.datasets.wrf_singlestep.WRFPredict method) (credit.datasets.wrfmultistep.RepeatingIndexSampler method) (credit.datasets.wrfmultistep.WRFMultiStep method) (credit.samplers.DistributedMultiStepBatchSampler method) (credit.samplers.MultiStepBatchSamplerSubset method) __next__() (credit.trainers.ic_optimization.TimeStepper method) __post_init__() (credit.data.ERA5_and_Forcing_Dataset method) (credit.data.ERA5_Dataset_Distributed method) (credit.datasets.datamap.DataMap method) (credit.datasets.downscaling_dataset.DownscalingDataset method) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.les_singlestep.LESDataset method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.wrf_singlestep.WRFDataset method) (credit.datasets.wrfmultistep.WRFMultiStep method) (credit.output_downscaling.OutputWrangler method) (credit.output_downscaling.OutputWriter method) __repr__() (credit.ensemble.color.ColorNoise method) (credit.ensemble.gaussian.GaussianNoise method) (credit.models.graph.TransformerConv method) __version__ (in module credit._version) __version_tuple__ (in module credit._version) _active (credit.trainers.ic_optimization.TimeStepper attribute) _add_pressure_and_geopotential() (in module credit.nwp) _agent() (in module credit.cli) (in module credit.cli._ask) _agent_bash() (in module credit.cli) (in module credit.cli._common) _AGENT_BASH_BLOCKLIST (in module credit.cli) (in module credit.cli._common) _agent_list_files() (in module credit.cli) (in module credit.cli._common) _agent_read_file() (in module credit.cli) (in module credit.cli._common) _AGENT_SYSTEM_PROMPT (in module credit.cli) (in module credit.cli._ask) _AGENT_TOOL_DEFS (in module credit.cli) (in module credit.cli._ask) _align_coords() (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) _align_phase_with() (in module credit.regrid) _alpha (credit.models.graph.TransformerConv attribute) _apply_extent() (in module credit.datasets.mrms) _ask() (in module credit.cli) (in module credit.cli._ask) _ask_anthropic() (in module credit.cli._ask) _ask_gemini() (in module credit.cli._ask) _ask_groq() (in module credit.cli._ask) _ask_openai() (in module credit.cli._ask) _ask_openrouter() (in module credit.cli._ask) _assert_increasing() (in module credit.regrid) _available_ram_gib() (in module credit.trainers.preflight) _boundary_data (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) _BRIDGESCALER_AVAILABLE (in module credit.preblock) _build_bridgescaler_jsons() (in module credit.cli._convert) _build_channel_map() (in module credit.cli) (in module credit.cli._plot) _build_denorm_stats() (in module credit.cli) (in module credit.cli._plot) _build_file_path() (in module credit.nwp) _build_master_clock() (credit.datasets.multi_source.MultiSourceDataset method) _build_nat_entry_map() (in module credit.datasets.hrrr) _build_output_denorm() (in module rollout_realtime_gen2) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) _build_parser() (in module credit.cli) (in module credit.cli._parser) _build_pbs_script() (in module credit.cli) (in module credit.cli._submit) _build_postblock_section() (in module credit.postblock) _build_preblock_section() (in module credit.preblock) _build_prs_entry_map() (in module credit.datasets.hrrr) _build_realtime_pbs_script() (in module credit.cli) (in module credit.cli._submit) _build_rollout_pbs_script() (in module credit.cli) (in module credit.cli._submit) _build_spatial_slices() (in module credit.datasets.goes) _build_timestamps() (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) _build_valid_indices() (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) _calc_window_shift() (credit.models.swin.SwinTransformerV2CrBlock method) _capture_enabled (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _channel_sort_key() (in module credit.preblock.concat) _check_in_data_config() (credit.datasets.base_dataset.BaseDataset method) _CLASS_SOURCES (in module credit.datasets) (in module credit.models) _collect_GOES_file_path() (credit.datasets.goes.GOESDataset method) _collect_noise_layers() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) _collect_run_context() (in module credit.cli) (in module credit.cli._ask) _combine_data() (in module credit.nwp) _compute_chain() (in module credit.cli) (in module credit.cli._submit) _compute_tisr() (in module credit.datasets.tisr) _conservative_latitude_weights() (in module credit.regrid) _conservative_longitude_weights() (in module credit.regrid) _convert() (in module credit.cli) (in module credit.cli._convert) _convert_windows_to_indices() (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) _COORD_CANDIDATES (in module credit.grid) _copy_batch() (credit.preblock.base.BasePreblock method) _corners_curvilinear() (in module credit.grid) _corners_rectilinear() (in module credit.grid) _create_correlated_noise() (credit.ensemble.color.ColorNoise method) _CREDIT_SYSTEM_PROMPT (in module credit.cli) (in module credit.cli._ask) _current_latents (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _current_timestep_map (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _DATASET_DRIVEN (in module credit.datasets.channel_layout) _DIAGNOSTIC (in module credit.datasets.channel_layout) _DIM (in module credit.cli._ask) _dispatch_tool() (in module credit.cli) (in module credit.cli._common) _do_special_eval_on_variable() (in module ensemble_eval) _do_standard_eval_on_variable() (in module ensemble_eval) _do_submit_realtime() (in module credit.cli) (in module credit.cli._submit) _do_submit_rollout() (in module credit.cli) (in module credit.cli._submit) _domain_group (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) _domain_group_idx (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) _domain_rank (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) _download_one() (in module credit.datasets.hrrr_download) _DownloadTask (class in credit.datasets.hrrr_download) _dp_group (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) _dp_rank (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) _earth_padding() (credit.boundary_padding.TensorPadding method) _earth_unpad() (credit.boundary_padding.TensorPadding method) _era5_tsi_data() (in module credit.datasets.tisr) _extract_field() (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) (credit.datasets.era5.ARCOERA5Dataset method) (credit.datasets.era5.WeatherBench2ERA5Dataset method) (credit.datasets.goes.GOESDataset method) (credit.datasets.hrrr.HRRRDataset method) (credit.datasets.local.LocalDataset method) (credit.datasets.mrms.MRMSDataset method) (credit.datasets.tisr.TISRDataset method) _extract_from_idx() (credit.datasets.hrrr.HRRRDataset method) _fetch_batch() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) _fetch_bytes_local() (in module credit.datasets.hrrr) _fetch_idx() (in module credit.datasets.hrrr) _fetch_message() (in module credit.datasets.hrrr) _fetch_one_batch() (in module credit.trainers.preflight) _find_coord_pair() (in module credit.grid) _find_file() (in module credit.datasets._utils) _find_nearest_latlon() (in module credit.datasets.goes) _find_subhf_entry() (in module credit.datasets.hrrr) _find_torchrun() (in module credit.cli) (in module credit.cli._common) _FlowSeqDumper (class in credit.cli._convert) _format_data() (in module credit.nwp) _format_datetime() (in module credit.nwp) _forward_with_latent_control() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) _fs (credit.datasets.era5.ARCOERA5Dataset attribute) (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) (credit.datasets.goes.GOESDataset attribute) (credit.datasets.mrms.MRMSDataset attribute) _get_batch_samples() (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) _get_cosine_zenith_angle() (in module credit.datasets.tisr) _get_field_name() (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) _get_file_source() (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) (credit.datasets.goes.GOESDataset method) (credit.datasets.mrms.MRMSDataset method) (credit.datasets.tisr.TISRDataset method) _get_gfs_maps() (in module credit.nwp) _get_hour_angle() (in module credit.datasets.tisr) _get_instantaneous_toa_tisr() (in module credit.datasets.tisr) _get_integrated_toa_tisr() (in module credit.datasets.tisr) _get_j2000_days() (in module credit.datasets.tisr) _get_latlon_grid() (in module credit.datasets.tisr) _get_ocean_sample() (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) _get_orbital_parameters() (in module credit.datasets.tisr) _get_session() (credit.datasets.hrrr.HRRRDataset method) _get_solar_time() (in module credit.datasets.tisr) _get_spatial_slice() (credit.datasets.hrrr.HRRRDataset method) _get_tsi() (in module credit.datasets.tisr) _get_W() (credit.preblock.regrid.Regridder method) (credit.preblock.Regridder method) _GRIB_TABLE_URL (in module credit.datasets.mrms_download) _HaloExchangeFunction (class in credit.domain_parallel.halo_exchange) _HRRR_HTTPS_BASE (in module credit.datasets.hrrr) _hrrr_local_path() (in module credit.datasets.hrrr) _hrrr_s3_uri() (in module credit.datasets.hrrr) _HRRR_V3_CUTOFF (in module credit.datasets.hrrr) _http_session (credit.datasets.hrrr.HRRRDataset attribute) _HTTP_TIMEOUT (in module credit.datasets.hrrr) _idx_cache (credit.datasets.hrrr.HRRRDataset attribute) _in_source_config() (credit.datasets.base_dataset.BaseDataset method) _infer_period_freq() (in module credit.datasets._utils) _init() (in module credit.cli) (in module credit.cli._convert) _init_fs() (credit.datasets.era5.ARCOERA5Dataset method) (credit.datasets.era5.WeatherBench2ERA5Dataset method) _initialize() (credit.ocean.samudra_constants.TensorMap method) (credit.ocean.samudra_data.Normalize method) _inject_flat_schema() (in module rollout_realtime_gen2) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) _inject_tracer_inds() (in module rollout_realtime_gen2) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) _instance (credit.ocean.samudra_constants.TensorMap attribute) (credit.ocean.samudra_data.Normalize attribute) _is_ncar_system() (in module credit.cli) (in module credit.cli._common) _is_spectral_norm_buffer() (credit.trainers.base_trainer.EMATracker static method) _kernel_crps() (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss method) _kernel_crps_implementation() (credit.losses.kcrps.KCRPSLoss method) _latitude_cell_bounds() (in module credit.regrid) _latitude_overlap() (in module credit.regrid) _LAZY_EXPORTS (in module credit.transforms) _load_dt() (credit.datasets.base_dataset.BaseDataset method) _load_end_datetime() (credit.datasets.base_dataset.BaseDataset method) _load_gfs_data() (in module credit.nwp) _load_gfs_variable() (in module credit.nwp) _load_idx_local() (in module credit.datasets.hrrr) _load_local_var() (credit.datasets.goes.GOESDataset method) (credit.datasets.mrms.MRMSDataset method) _load_model() (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) _load_model_entry() (in module credit.models) _load_model_state_dict() (credit.trainers.base_trainer.BaseTrainer static method) _load_num_forecast_steps() (credit.datasets.base_dataset.BaseDataset method) _load_pbs_config() (in module credit.cli) (in module credit.cli._submit) _load_remote_var() (credit.datasets.goes.GOESDataset method) (credit.datasets.mrms.MRMSDataset method) _load_sample() (credit.datasets.base_dataset.BaseDataset method) _load_start_datetime() (credit.datasets.base_dataset.BaseDataset method) _log_batch_progress() (credit.trainers.base_trainer.BaseTrainer method) _longitude_overlap() (in module credit.regrid) _make_attention_mask() (credit.models.swin.SwinTransformerV2CrBlock method) _make_pair_wise_relative_positions() (credit.models.swin.WindowMultiHeadAttention method) _MANAGER (in module credit.domain_parallel.manager) _map_files() (in module credit.datasets._utils) _match_spatial() (credit.models.swin_wrf.WRFTransformer method) _MAX_REMOTE_WORKERS (in module credit.datasets.hrrr) _mean (credit.preblock.ERA5Normalizer attribute) (credit.preblock.norm.ERA5Normalizer attribute) _mean() (credit.regrid.ConservativeRegridder method) _metrics() (in module credit.cli) (in module credit.cli._plot) _mirror_padding() (credit.boundary_padding.TensorPadding method) _mirror_unpad() (credit.boundary_padding.TensorPadding method) _mode (credit.datasets.downscaling_dataset.DownscalingDataset attribute) _MODEL_PREDICTED (in module credit.datasets.channel_layout) _MODEL_REGISTRY (in module credit.models) _model_state_dict() (credit.trainers.base_trainer.BaseTrainer static method) _nanmean() (credit.regrid.ConservativeRegridder method) _nearest_neighbor_2d() (credit.regrid.NearestRegridder method) _needs_halo_conv2d() (in module credit.domain_parallel.convert) _needs_halo_conv3d() (in module credit.domain_parallel.convert) _needs_halo_conv_transpose2d() (in module credit.domain_parallel.convert) _needs_halo_conv_transpose3d() (in module credit.domain_parallel.convert) _noise_layers (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _normalize_tensor() (credit.preblock.ERA5Normalizer method) (credit.preblock.norm.ERA5Normalizer method) _OPENROUTER_MODEL (in module credit.cli._ask) _original_factors (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _output (credit.datasets.downscaling_dataset.DownscalingDataset attribute) _parse_idx() (in module credit.datasets.hrrr) _PBS_DEFAULTS (in module credit.cli) (in module credit.cli._common) _periodic_lower_bounds() (in module credit.regrid) _periodic_overlap() (in module credit.regrid) _periodic_upper_bounds() (in module credit.regrid) _persist_cache (credit.datasets.base_dataset.BaseDataset attribute) _plot() (in module credit.cli) (in module credit.cli._plot) _populate_dp_3d_idx() (credit.ocean.samudra_constants.TensorMap method) _populate_var_3d_idx() (credit.ocean.samudra_constants.TensorMap method) _PREDICTABLE_FIELD_TYPES (in module credit.preblock.concat) _print_ensemble_rollout_plan() (in module credit.cli) (in module credit.cli._submit) _print_job_plan() (in module credit.cli) (in module credit.cli._submit) _process_chunk() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) _prognostic_data (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) _prognostic_mean_np (credit.datasets.om4_multistep_batcher.StandardScaler attribute) _prognostic_std_np (credit.datasets.om4_multistep_batcher.StandardScaler attribute) _prompt() (in module credit.cli) (in module credit.cli._common) _prompt_bool() (in module credit.cli) (in module credit.cli._common) _PROVIDER_INSTALL (in module credit.cli._ask) _PROVIDER_RUNNERS (in module credit.cli._ask) _ProviderError, [1] _PROVIDERS (in module credit.cli._ask) _qsub() (in module credit.cli) (in module credit.cli._submit) _realtime() (in module credit.cli) (in module credit.cli._submit) _register_field() (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) (credit.datasets.hrrr.HRRRDataset method) _regrid() (credit.preblock.regrid.Regridder method) (credit.preblock.Regridder method) (in module credit.nwp) _regrid_variable() (in module credit.nwp) _relative_positional_encodings() (credit.models.swin.WindowMultiHeadAttention method) _replace_module() (in module credit.domain_parallel.convert) _repo_root() (in module credit.cli) (in module credit.cli._common) _RESET (in module credit.cli._ask) _resolve_nat_levels() (in module credit.datasets.hrrr) _resolve_pbs_opts() (in module credit.cli) (in module credit.cli._submit) _resolve_persist_timestamp() (credit.datasets.base_dataset.BaseDataset method) _resolve_pressure_levels() (in module credit.datasets.hrrr) _rollout() (in module credit.cli) (in module credit.cli._submit) _rollout_ensemble() (in module credit.cli) (in module credit.cli._submit) _run_preblock_group() (in module credit.preblock) _S3_BUCKET (in module credit.datasets.hrrr) _s3_uri_to_https() (in module credit.datasets.hrrr) _sample_to_batch() (in module rollout_realtime_gen2) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) _save_checkpoint() (credit.trainers.base_trainer.BaseTrainer method) _save_worker() (in module rollout_realtime_gen2) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) _setup_arrangement() (credit.datasets.downscaling_dataset.DownscalingDataset method) _setup_datasets() (credit.datasets.downscaling_dataset.DownscalingDataset method) _setup_logging() (in module credit.cli) (in module credit.cli._common) _shifted_window_attn() (credit.models.swin.SwinTransformerV2CrBlock method) _SOURCE_REGISTRY (in module credit.datasets.multi_source) _spatial_slice (credit.datasets.hrrr.HRRRDataset attribute) _SPATIAL_TIME_DIMS (in module credit.cli._convert) _start_s3_fs() (in module credit.datasets._utils) _std (credit.preblock.ERA5Normalizer attribute) (credit.preblock.norm.ERA5Normalizer attribute) _stored_latents (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) _STRFTIME_TO_FREQ (in module credit.datasets._utils) _STRFTIME_TO_REGEX (in module credit.datasets._utils) _strftime_to_regex() (in module credit.datasets._utils) _submit() (in module credit.cli) (in module credit.cli._submit) _SummaryWriter (in module credit.trainers.base_trainer) _to_cftime() (in module credit.datasets._utils) _to_float32() (in module credit.datasets.hrrr) _to_tensor() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) _to_tuple() (credit.regrid.Grid method) _TORCH_DTYPE (in module credit.datasets.tisr) _train() (in module credit.cli) (in module credit.cli._submit) _TRAINER_REGISTRY (in module credit.trainers) _TRANSFORM_REGISTRY (in module credit.transforms) _transforms module _VALID_SECTIONS (in module credit.postblock) (in module credit.preblock) _validate_product_request() (in module credit.datasets.hrrr) _vertical_interpolation() (in module credit.nwp) _W (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) _W_device (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) _WB2_ERA5_BASE (in module credit.datasets.era5) _WB2_ERA5_DEFAULT_LEVELS (in module credit.datasets.era5) _WB2_ERA5_STORE_PATHS (in module credit.datasets.era5) _wet_mask_np (credit.datasets.om4_multistep_batcher.StandardScaler attribute) _write_reload_config() (in module credit.cli) (in module credit.cli._convert) _write_scrip() (in module credit.grid) _yaml_dump() (in module credit.cli._convert) _zarr_dims() (in module credit.cli._convert) A AbstractBaseDataset (class in credit.datasets.base_dataset) acc() (credit.metrics.LatWeightedMetricsClimatology method) acc_vars (credit.metrics.LatWeightedMetricsClimatology attribute) accum_log() (in module credit.replay_buffer) (in module credit.trainers.utils) (in module trainer_multistep_deprecated) act (credit.models.unet_attention_modules.CoordinateAttention attribute) (credit.models.unet_diffusion.Block attribute) add() (credit.replay_buffer.ReplayBuffer method) add_param_group() (credit.models.checkpoint.OptimizerWrapper method) add_spatially_correlated_noise() (in module credit.ensemble.utils) adjust_start_times() (in module credit.ensemble.bred_vector) affine (credit.domain_parallel.DomainParallelGroupNorm attribute) (credit.domain_parallel.layers.DomainParallelGroupNorm attribute) all_files (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) all_start_indices (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) allvars (_transforms.ToTensor attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) AlmostFairKCRPSLoss (class in credit.losses.almost_fair_crps) alpha (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) alpha_init (credit.skebs.SKEBS attribute) amp (credit.trainers.base_trainer.BaseTrainer attribute) amplitude (credit.ensemble.color.ColorNoise attribute) (credit.ensemble.gaussian.GaussianNoise attribute) (credit.ensemble.spherical.SphericalNoise attribute) annealed_probability() (in module credit.scheduler) app (in module data_server) apply_noise_perturbation_step() (in module credit.ensemble.color) apply_postblocks() (in module credit.postblock) apply_preblocks() (in module credit.preblock) apply_spectral_norm() (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.graph) (in module credit.models.swin) (in module credit.models.swin_wrf) (in module credit.models.wxformer.crossformer) (in module credit.models.wxformer.crossformer_downscaling) ARCOERA5Dataset (class in credit.datasets.era5) area (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) arrangement (credit.datasets.downscaling_dataset.DownscalingDataset attribute) Array (in module credit.data) (in module credit.datasets.downscaling_dataset) (in module credit.regrid) assemble_rollout_batch() (in module credit.trainers.rollout_utils) Attend (class in credit.attend) attend (credit.models.unet_diffusion.Attention attribute) Attention (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.unet_diffusion) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) attention (credit.models.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer.UpBlock attribute) AttentionConfig (class in credit.attend) attn (credit.models.swin.SwinTransformerV2CrBlock attribute) attn_drop (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) attn_dropout (credit.attend.Attend attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) attn_type (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) average_div_rot_spectrum() (in module credit.verification.standard) average_zonal_spectrum() (in module credit.verification.standard) avg_pool (credit.models.unet_attention_modules.ECABlock attribute) B b (credit.models.camulator.LayerNorm attribute) (credit.models.camulator.UpBlock attribute) (credit.models.camulator.UpBlockPS attribute) (credit.models.crossformer.LayerNorm attribute) (credit.models.crossformer.UpBlock attribute) (credit.models.dscale_wrf.DownBlock attribute) (credit.models.dscale_wrf.UpBlock attribute) (credit.models.fuxi.DownBlock attribute) (credit.models.fuxi.UpBlock attribute) (credit.models.swin_wrf.DownBlock attribute) (credit.models.swin_wrf.UpBlock attribute) (credit.models.wxformer.crossformer.LayerNorm attribute) (credit.models.wxformer.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) (credit.models.wxformer.crossformer_downscaling.LayerNorm attribute) (credit.models.wxformer.crossformer_downscaling.UpBlock attribute) backprop_on_timestep (credit.trainers.trainer_gen2.TrainerERA5Gen2 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.domain_parallel.halo_exchange._HaloExchangeFunction static method) (credit.models.checkpoint.OptimizerWrapper method) (credit.trainers.trainerERA5_ensemble.Gather static method) backward_by_grad() (credit.models.checkpoint.OptimizerWrapper method) base_input_channels (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) base_losses() (in module credit.losses) (in module credit.losses.base_losses) base_max_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) base_output_channels (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) base_path (_transforms.NormalizeTendency attribute) (credit.datasets.hrrr.HRRRDataset attribute) BaseDataset (class in credit.datasets.base_dataset) BaseModel (class in credit.models.base_model) BasePostblock (class in credit.postblock.base) BasePreblock (class in credit.preblock.base) BaseTrainer (class in credit.trainers.base_trainer) batch_call_count (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) batch_indices (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) batch_indices_splits (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) batch_normalize (credit.losses.covariance.CovarianceWeightedMSELoss attribute) batch_size (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.samplers.DistributedMultiStepBatchSampler attribute) (credit.samplers.MultiStepBatchSamplerSubset attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) batched_forward (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) (credit.losses.kcrps.KCRPSLoss attribute) batches_per_epoch (credit.trainers.base_trainer.BaseTrainer attribute) batches_per_epoch() (credit.datasets.era5_multistep.RepeatingIndexSampler method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.wrfmultistep.RepeatingIndexSampler method) BatchForecastLenDataLoader (class in credit.datasets.load_dataset_and_dataloader) BatchForecastLenSampler (class in credit.datasets.load_dataset_and_dataloader) BatchForecastLenSamplerSamudra (class in credit.datasets.load_dataset_and_dataloader) bchw_to_bhwc() (in module credit.models.swin) beta (credit.models.graph.TransformerConv attribute) bhwc_to_bchw() (in module credit.models.swin) bias (credit.domain_parallel.DomainParallelConv2d property) (credit.domain_parallel.DomainParallelPeriodicConv2d property) (credit.domain_parallel.layers.DomainParallelConv2d property) (credit.domain_parallel.layers.DomainParallelPeriodicConv2d property) (credit.models.graph.LayerNorm attribute) biased (credit.losses.kcrps.KCRPSLoss attribute) BilinearRegridder (class in credit.regrid) binned_spread_skill() (in module credit.verification.ensemble) Block (class in credit.models.unet_diffusion) block1 (credit.models.unet_diffusion.ResnetBlock attribute) block2 (credit.models.unet_diffusion.ResnetBlock attribute) blocks (credit.models.swin.SwinTransformerV2CrStage attribute) bn1 (credit.models.unet_attention_modules.CoordinateAttention attribute) bottom (credit.transforms_downscaling.Pad attribute) BOUND_VARS_MAP (in module credit.ocean.samudra_constants) boundary (credit.datasets.datamap.VarDict attribute) boundary_input (credit.data.Sample_WRF attribute) boundary_mean (credit.datasets.om4_multistep_batcher.StandardScaler attribute) boundary_std (credit.datasets.om4_multistep_batcher.StandardScaler attribute) BredVector (class in credit.ensemble.bred_vector) BridgescalerScaleState (class in credit.transforms.transforms_quantile) BridgeScalerTransformer (class in credit.postblock) (class in credit.postblock.scaler) (class in credit.preblock.scaler) buffer_size (credit.replay_buffer.ReplayBuffer attribute) build_channel_layout() (in module credit.datasets.channel_layout) build_GFS_init() (in module credit.nwp) build_postblocks() (in module credit.postblock) build_preblocks() (in module credit.preblock) by (credit.transforms_downscaling.Expand attribute) C calc_edges() (in module graph_edges) calc_global_solar module calculate_crps_per_channel() (in module credit.ensemble.crps) calculate_ensemble_metrics() (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) calculate_mslp_and_append() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) Camulator (class in credit.models.camulator) cartopy_panel2() (in module credit.visualization_tools) cartopy_panel4() (in module credit.visualization_tools) cartopy_panel6() (in module credit.visualization_tools) cartopy_single_panel() (in module credit.visualization_tools) cast_tuple() (in module credit.diffusion_utils) (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.wxformer.crossformer) (in module credit.models.wxformer.crossformer_downscaling) channel_attention (credit.models.unet_attention_modules.EfficientMixedAttention attribute) channels (credit.diffusion.GaussianDiffusion attribute) (credit.diffusion.ModifiedGaussianDiffusion attribute) (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.skebs.SKEBS attribute) channels_out (credit.models.unet_diffusion.UnetDiffusion attribute) check_dataloader_startup() (in module credit.trainers.preflight) check_model_gpu_memory() (in module credit.trainers.preflight) check_rollout_files() (in module ensemble_eval) checkpoint_stages (credit.models.swin.SwinTransformerV2Cr attribute) chunk_size (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] circular_pad1d() (in module credit.models.swin) clamp (credit.ensemble.bred_vector.BredVector attribute) clamp_max (credit.ensemble.bred_vector.BredVector attribute) clamp_min (credit.ensemble.bred_vector.BredVector attribute) cleanup() (in module credit.replay_buffer) (in module credit.trainers.utils) (in module trainer_multistep_deprecated) clear_stored_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) climatology (credit.metrics.LatWeightedMetricsClimatology attribute) Clip (class in credit.transforms_downscaling) clip_grad_by_norm() (credit.models.checkpoint.OptimizerWrapper method) clip_grad_by_value() (credit.models.checkpoint.OptimizerWrapper method) clip_parameters() (credit.skebs.SKEBS method) clone_dataset() (in module credit.ensemble.bred_vector) cmap_combine() (in module credit.visualization_tools) cmax (credit.transforms_downscaling.Clip attribute) cmin (credit.transforms_downscaling.Clip attribute) coef_a (credit.physics_core.physics_hybrid_sigma_level attribute), [1] coef_b (credit.physics_core.physics_hybrid_sigma_level attribute), [1] collate_fn() (in module credit.datasets.load_dataset_and_dataloader) colorbar_opt() (in module credit.visualization_tools) ColorNoise (class in credit.ensemble.color) combine_microphysics_terms() (in module credit.gefs) combine_tiles() (in module credit.gefs) commit_id (in module credit._version) compute_anomalies() (in module credit.ocean.samudra_data) compute_density() (in module credit.physics_core) compute_hlevs() (credit.physics_core.ModelLevelPressures method) compute_metrics() (in module rollout_metrics) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) compute_mlev_thickness() (credit.physics_core.ModelLevelPressures method) compute_p() (credit.physics_core.ModelLevelPressures method) compute_spread_skill_metric() (in module rollout_metrics_noisy_model) compute_virtual_temperature() (in module credit.physics_core) concat (credit.models.graph.TransformerConv attribute) concat_and_reshape() (credit.models.base_model.BaseModel method) (credit.models.checkpoint.TorchFSDPModel method) (credit.replay_buffer.ReplayBuffer method) (in module credit.data) concat_fix() (in module credit.postblock.gen1) concat_for_inplace_ops() (in module credit.skebs) ConcatToTensor (class in credit.preblock) (class in credit.preblock.concat) condition (credit.diffusion.GaussianDiffusion attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) conditional_dimensions (credit.models.unet_diffusion.UnetDiffusion attribute) conf (_transforms.ToTensor attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) (in module credit.datasets.era5_singlestep) (in module credit.datasets.om4_multistep_batcher) (rollout_samudra.ForecastProcessor attribute) config (in module credit.xr_sampler) (rollout_metrics_noisy_model.Objective attribute) (train.Objective attribute) (train_gen1.Objective attribute) (train_les.Objective attribute) (train_multistep.Objective attribute) (train_wrf.Objective attribute) (train_wrf_multi.Objective attribute) config_file (in module credit.pbs) ConservativeRegridder (class in credit.regrid) construct_metadata() (in module credit.ocean.samudra_constants) conv (credit.domain_parallel.DomainParallelConv2d attribute) (credit.domain_parallel.DomainParallelConv3d attribute) (credit.domain_parallel.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.DomainParallelConvTranspose3d attribute) (credit.domain_parallel.DomainParallelPeriodicConv2d attribute) (credit.domain_parallel.layers.DomainParallelConv2d attribute) (credit.domain_parallel.layers.DomainParallelConv3d attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose3d attribute) (credit.domain_parallel.layers.DomainParallelPeriodicConv2d attribute) (credit.models.camulator.UpBlock attribute) (credit.models.camulator.UpBlockPS attribute) (credit.models.dscale_wrf.DownBlock attribute) (credit.models.dscale_wrf.UpBlock attribute) (credit.models.fuxi.DownBlock attribute) (credit.models.fuxi.UpBlock attribute) (credit.models.swin_wrf.DownBlock attribute) (credit.models.swin_wrf.UpBlock attribute) (credit.models.unet_attention_modules.ECABlock attribute) (credit.models.unet_attention_modules.LightSpatialAttention attribute) (credit.models.unet_diffusion.PeriodicConv2d attribute) (credit.models.wxformer.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) (credit.models.wxformer.crossformer_downscaling.UpBlock attribute) (credit.skebs.BackscatterCNN attribute) conv1 (credit.models.unet_attention_modules.CoordinateAttention attribute) conv4up (credit.models.camulator.Camulator attribute) conv_h (credit.models.unet_attention_modules.CoordinateAttention attribute) conv_w (credit.models.unet_attention_modules.CoordinateAttention attribute) convert_image_to_fn() (in module credit.diffusion_utils) convert_longitude() (credit.credit_ptype.CreditPostProcessor method) convert_tensor_out_to_dict() (in module credit.ocean.samudra_data) convert_to_domain_parallel() (in module credit.domain_parallel) (in module credit.domain_parallel.convert) convs (credit.models.camulator.CrossEmbedLayer attribute) (credit.models.crossformer.CrossEmbedLayer attribute) (credit.models.wxformer.crossformer.CrossEmbedLayer attribute) (credit.models.wxformer.crossformer_downscaling.CrossEmbedLayer attribute) CoordinateAttention (class in credit.models.unet_attention_modules) copy_checkpoint() (in module credit.models.checkpoint) correlated (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) cos_lat (credit.skebs.SKEBS attribute) cosine_beta_schedule() (in module credit.diffusion) CosineAnnealingWarmupRestarts (class in credit.scheduler) count_channels() (in module credit.datasets.count_channels) CovarianceWeightedMSELoss (class in credit.losses.covariance) CP_DRY (in module credit.physics_constants) CP_ICE (in module credit.physics_constants) CP_LIQUID (in module credit.physics_constants) CP_VAPOR (in module credit.physics_constants) cpu_config (credit.attend.Attend attribute) create_diffusion() (in module credit.models.unet_diffusion) (in module credit.models.wxformer.crossformer_diffusion) create_model() (in module credit.models.unet_diffusion) (in module credit.models.wxformer.crossformer_diffusion) create_pressure_grid() (in module credit.interp) create_reduced_pressure_grid() (in module credit.interp) create_sigmoid_ramp_function() (in module credit.pol_lapdiff_filt) credit module credit._version module credit.animation module credit.attend module credit.boundary_padding module credit.cli module credit.cli._ask module credit.cli._common module credit.cli._convert module credit.cli._parser module credit.cli._plot module credit.cli._submit module credit.credit_ptype module credit.data module credit.datasets module credit.datasets._utils module credit.datasets.base_dataset module credit.datasets.channel_layout module credit.datasets.count_channels module credit.datasets.datamap module credit.datasets.downscaling_dataset module credit.datasets.era5 module credit.datasets.era5_multistep module credit.datasets.era5_multistep_batcher module credit.datasets.era5_singlestep module credit.datasets.goes module credit.datasets.hrrr module credit.datasets.hrrr_download module credit.datasets.les_singlestep module credit.datasets.load_dataset_and_dataloader module credit.datasets.local module credit.datasets.mrms module credit.datasets.mrms_download module credit.datasets.multi_source module credit.datasets.om4_multistep_batcher module credit.datasets.realtime_predict module credit.datasets.sequential_multistep module credit.datasets.tisr module credit.datasets.wrf_singlestep module credit.datasets.wrfmultistep module credit.diffusion module credit.diffusion_utils module credit.distributed module credit.domain_parallel module credit.domain_parallel.convert module credit.domain_parallel.halo_exchange module credit.domain_parallel.layers module credit.domain_parallel.manager module credit.domain_parallel.sharding module credit.ensemble module credit.ensemble.bred_vector module credit.ensemble.color module credit.ensemble.crps module credit.ensemble.gaussian module credit.ensemble.spherical module credit.ensemble.temporal module credit.ensemble.utils module credit.forecast module credit.gefs module credit.grid module credit.interp module credit.losses module credit.losses.almost_fair_crps module credit.losses.base_losses module credit.losses.covariance module credit.losses.downscaling_loss module credit.losses.kcrps module credit.losses.les_loss module credit.losses.logcosh module credit.losses.msle module credit.losses.power module credit.losses.spectral module credit.losses.weighted_loss module credit.losses.xsigmoid module credit.losses.xtanh module credit.metadata module credit.metrics module credit.metrics_downscaling module credit.mixed_precision module credit.models module credit.models.base_model module credit.models.camulator module credit.models.checkpoint module credit.models.crossformer module credit.models.debugger_model module credit.models.dscale_wrf module credit.models.fuxi module credit.models.graph module credit.models.reset module credit.models.swin module credit.models.swin_wrf module credit.models.unet module credit.models.unet_attention_modules module credit.models.unet_diffusion module credit.models.unet_downscaling module credit.models.wxformer module credit.models.wxformer.crossformer module credit.models.wxformer.crossformer_diffusion module credit.models.wxformer.crossformer_downscaling module credit.models.wxformer.crossformer_ensemble module credit.models.wxformer.sdl_inference_wrapper module credit.models.wxformer.stochastic_decomposition_layer module credit.nwp module credit.ocean module credit.ocean.samudra_constants module credit.ocean.samudra_data module credit.output module credit.output_downscaling module credit.parser module credit.pbs module credit.physics_constants module credit.physics_core module credit.pol_lapdiff_filt module credit.postblock module credit.postblock.base module credit.postblock.gen1 module credit.postblock.geopotential module credit.postblock.reconstruct module credit.postblock.scaler module credit.postblock.wet_mask_samudra module credit.preblock module credit.preblock.base module credit.preblock.concat module credit.preblock.log module credit.preblock.norm module credit.preblock.regrid module credit.preblock.scaler module credit.preblock.sqrt module credit.regrid module credit.replay_buffer module credit.samplers module credit.scheduler module credit.seed module credit.skebs module credit.solar module credit.trainers module credit.trainers.base_trainer module credit.trainers.ic_optimization module credit.trainers.preflight module credit.trainers.rollout_utils module credit.trainers.trainer_downscaling module credit.trainers.trainer_gen2 module credit.trainers.trainer_om4_samudra module credit.trainers.trainerERA5_Diffusion module credit.trainers.trainerERA5_ensemble module credit.trainers.trainerERA5gen1 module credit.trainers.trainerLES module credit.trainers.trainerWRF module credit.trainers.trainerWRF_multi module credit.trainers.utils module credit.transforms module credit.transforms.transforms_global module credit.transforms.transforms_les module credit.transforms.transforms_quantile module credit.transforms.transforms_wrf module credit.transforms_downscaling module credit.verification module credit.verification.ensemble module credit.verification.standard module credit.visualization_tools module credit.xr_sampler module credit_main_parser() (in module credit.parser) CreditPostProcessor (class in credit.credit_ptype) criterion (credit.diffusion.ModifiedGaussianDiffusion attribute) CrossEmbedLayer (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) CrossFormer (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) crossformer_config (in module credit.models.wxformer.crossformer_diffusion) CrossFormerDiffusion (class in credit.models.wxformer.crossformer_diffusion) CrossFormerWithNoise (class in credit.models.wxformer.crossformer_ensemble) cse (credit.models.unet_attention_modules.SCSEAttention attribute) cube_embedding (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) cube_embedding_inside (credit.models.swin_wrf.WRFTransformer attribute) cube_embedding_outside (credit.models.swin_wrf.WRFTransformer attribute) CubeEmbedding (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) cuda_config (credit.attend.Attend attribute) cur_cycle_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) curr_source_cfg (credit.datasets.base_dataset.BaseDataset attribute) curr_source_name (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) current_epoch (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) current_file_year (credit.xr_sampler.XRSamplerByYear attribute) current_index (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) cycle (credit.scheduler.CosineAnnealingWarmupRestarts attribute) cycle() (in module credit.diffusion_utils) (in module credit.replay_buffer) (in module credit.trainers.utils) (in module trainer_multistep_deprecated) cycle_mult (credit.scheduler.CosineAnnealingWarmupRestarts attribute) cycle_pattern() (credit.skebs.SKEBS method) D data_keys (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] data_lookup (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) data_parallel_group (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) data_parallel_size (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) data_path (in module data_server) data_server module data_types (credit.preblock.log.LogTransform attribute) (credit.preblock.LogTransform attribute) (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) (credit.preblock.sqrt.SqrtTransform attribute) (credit.preblock.SqrtTransform attribute) DataMap (class in credit.datasets.datamap) dataset (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.output_downscaling.OutputWrangler attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.samplers.MultiStepBatchSamplerSubset attribute) (credit.trainers.ic_optimization.TimeStepper attribute) dataset_id (in module credit.datasets.load_dataset_and_dataloader) dataset_name (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] dataset_type (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.era5.ARCOERA5Dataset attribute) (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) (credit.datasets.goes.GOESDataset attribute) (credit.datasets.hrrr.HRRRDataset attribute), [1] (credit.datasets.local.LocalDataset attribute) (credit.datasets.mrms.MRMSDataset attribute) (credit.datasets.tisr.TISRDataset attribute) datasets (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.multi_source.MultiSourceDataset attribute), [1] DataTransforms (class in credit.transforms_downscaling) date2tindex() (credit.datasets.datamap.DataMap method) datetime_index (credit.data.Sample attribute) (credit.data.Sample_diag attribute) (credit.data.Sample_dscale attribute) (credit.data.Sample_LES attribute) (credit.data.Sample_WRF attribute) datetimes (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) (credit.datasets.goes.GOESDataset attribute) (credit.datasets.hrrr.HRRRDataset attribute) (credit.datasets.multi_source.MultiSourceDataset attribute), [1] days_of_year (trainerERA5_v1.TOADataLoader attribute) ddim_sample() (credit.diffusion.GaussianDiffusion method) ddim_sampling_eta (credit.diffusion.GaussianDiffusion attribute) debug_save_loc (credit.skebs.SKEBS attribute) DebuggerModel (class in credit.models.debugger_model) dec_norm (credit.models.graph.GraphResTransfGRU attribute) decay (credit.trainers.base_trainer.EMATracker attribute) decoder (credit.models.graph.GraphResTransfGRU attribute) decoder_noise_factor (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) default() (in module credit.attend) (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) default_metadata (in module credit.ocean.samudra_constants) defaults (credit.models.checkpoint.OptimizerWrapper property) depth (credit.models.swin.SwinTransformerV2Cr attribute) DEPTH_I_LEVELS (in module credit.ocean.samudra_constants) DEPTH_LEVELS (in module credit.ocean.samudra_constants) description (in module ensemble_eval) (in module model_summary) (in module rollout_downscaling) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) (in module train) (in module train_multistep) device (credit.datasets.tisr.TISRDataset attribute) (credit.diffusion.GaussianDiffusion property) (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (in module credit.pol_lapdiff_filt) (rollout_metrics_noisy_model.Objective attribute) (rollout_samudra.ForecastProcessor attribute) (train.Objective attribute) (train_gen1.Objective attribute) (train_les.Objective attribute) (train_multistep.Objective attribute) (train_wrf.Objective attribute) (train_wrf_multi.Objective attribute) (trainer_multistep_deprecated.Trainer attribute) device_compatible_to() (in module credit.data) dewpoint_temp() (credit.credit_ptype.CreditPostProcessor method) diagnostic (credit.datasets.datamap.VarDict attribute) diff_lap2d_filt() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) Diffusion_and_Pole_Filter (class in credit.pol_lapdiff_filt) dim (credit.datasets.datamap.DataMap attribute) (credit.domain_parallel.halo_exchange.HaloExchange attribute) (credit.domain_parallel.HaloExchange attribute) (credit.models.swin.SwinTransformerV2CrBlock attribute) (credit.models.unet_diffusion.SinusoidalPosEmb attribute) (credit.output_downscaling.OutputWriter attribute) dim_head (credit.models.graph.GraphResTransfGRU attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) direction (credit.trainers.base_trainer.BaseTrainer attribute) disable_latent_capture() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) dissipation_scaling_coefficient (credit.skebs.SKEBS attribute) dissipation_type (credit.skebs.SKEBS attribute) distributed (credit.trainers.base_trainer.BaseTrainer attribute) distributed_model_wrapper() (in module credit.distributed) DistributedMultiStepBatchSampler (class in credit.samplers) DistributedSequentialDataset (class in credit.datasets.sequential_multistep) DistributedSequentialDatasetBasic (class in credit.datasets.sequential_multistep) div_rot_spectrum() (in module credit.verification.standard) divisible_by() (in module credit.diffusion_utils) do_eval() (in module ensemble_eval) domain_group (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) domain_parallel_size (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) domain_rank (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) domain_world_size (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) DomainParallelConv2d (class in credit.domain_parallel) (class in credit.domain_parallel.layers) DomainParallelConv3d (class in credit.domain_parallel) (class in credit.domain_parallel.layers) DomainParallelConvTranspose2d (class in credit.domain_parallel) (class in credit.domain_parallel.layers) DomainParallelConvTranspose3d (class in credit.domain_parallel) (class in credit.domain_parallel.layers) DomainParallelGroupNorm (class in credit.domain_parallel) (class in credit.domain_parallel.layers) DomainParallelInterpolate (class in credit.domain_parallel) (class in credit.domain_parallel.layers) DomainParallelManager (class in credit.domain_parallel) (class in credit.domain_parallel.manager) DomainParallelPeriodicConv2d (class in credit.domain_parallel) (class in credit.domain_parallel.layers) down (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) DownBlock (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) download_gefs_run() (in module credit.gefs) download_hrrr() (in module credit.datasets.hrrr_download) download_mrms() (in module credit.datasets.mrms_download) downs (credit.models.unet_diffusion.UnetDiffusion attribute) Downsample() (in module credit.models.unet_diffusion) downsample_factor (credit.models.unet_diffusion.UnetDiffusion property) downscale (credit.models.swin.SwinTransformerV2CrStage attribute) DownscalingCrossFormer (class in credit.models.wxformer.crossformer_downscaling) DownscalingDataset (class in credit.datasets.downscaling_dataset) DownscalingLoss (class in credit.losses) (class in credit.losses.downscaling_loss) DownscalingSegmentationModel (class in credit.models.unet_downscaling) dp_rank (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) dp_world_size (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) dpb (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) draw_variables() (in module credit.visualization_tools) drop_path1 (credit.models.swin.SwinTransformerV2CrBlock attribute) drop_path2 (credit.models.swin.SwinTransformerV2CrBlock attribute) drop_var_from_dataset() (in module credit.data) dropout (credit.attend.Attend attribute) (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.graph.TransformerConv attribute) (credit.models.unet_diffusion.Block attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) ds_read_and_subset() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) DscaleTransformer (class in credit.models.dscale_wrf) dst_shape (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) dt (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) (credit.samplers.MultiStepBatchSamplerSubset attribute) dtype (credit.replay_buffer.ReplayBuffer attribute) dyn_forcing_files (credit.datasets.realtime_predict.RealtimePredictDataset attribute) DynamicPositionBias (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) E ECABlock (class in credit.models.unet_attention_modules) edge_dim (credit.models.graph.TransformerConv attribute) edge_path (credit.models.graph.GraphResTransfGRU attribute) EfficientMixedAttention (class in credit.models.unet_attention_modules) EMATracker (class in credit.trainers.base_trainer) embed_dim (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) enable_flash (credit.attend.AttentionConfig attribute) enable_latent_capture() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) enable_math (credit.attend.AttentionConfig attribute) enable_mem_efficient (credit.attend.AttentionConfig attribute) enc_norm (credit.models.graph.GraphResTransfGRU attribute) encode_datetime64() (in module credit.data) encoder (credit.models.graph.GraphResTransfGRU attribute) encoder_noise (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute), [1] encoder_noise_factor (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) end_datetime (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) ensemble_eval module ensemble_perturb (credit.ensemble.bred_vector.BredVector attribute) ensemble_size (credit.metrics_downscaling.UnWeightedMetrics attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) ensure_numpy_datetime() (in module credit.data) epochs (credit.trainers.base_trainer.BaseTrainer attribute) eps (credit.domain_parallel.DomainParallelGroupNorm attribute) (credit.domain_parallel.layers.DomainParallelGroupNorm attribute) (credit.models.camulator.LayerNorm attribute) (credit.models.crossformer.LayerNorm attribute) (credit.models.graph.LayerNorm attribute) (credit.models.wxformer.crossformer.LayerNorm attribute) (credit.models.wxformer.crossformer_downscaling.LayerNorm attribute) (credit.preblock.log.LogTransform attribute) (credit.preblock.LogTransform 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) ERA5Normalizer (class in credit.preblock) (class in credit.preblock.norm) estimate_dataloader_memory_gib() (in module credit.trainers.preflight) evaluate() (in module ensemble_eval) evapor_ind (credit.postblock.gen1.GlobalWaterFixer attribute) (credit.postblock.GlobalWaterFixer attribute) exists() (in module credit.attend) (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) Expand (class in credit.transforms_downscaling) exponent (credit.transforms_downscaling.Power attribute) extent (credit.datasets.goes.GOESDataset attribute) (credit.datasets.hrrr.HRRRDataset attribute) (credit.datasets.mrms.MRMSDataset attribute) extract() (in module credit.diffusion) (in module credit.models.wxformer.crossformer_diffusion) extract_month_day_hour() (in module credit.data) extract_variable_levels() (credit.credit_ptype.CreditPostProcessor method) extract_wet_mask() (in module credit.ocean.samudra_data) F fast_state_interp_loop() (in module credit.interp) fc (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) fc1 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) fc2 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) fc3 (credit.skebs.BackscatterFCNNWide attribute) fc4 (credit.skebs.BackscatterFCNNWide attribute) feat_size (credit.models.swin.SwinTransformerV2CrBlock attribute) (credit.models.swin.SwinTransformerV2CrStage attribute) feature_info (credit.models.swin.SwinTransformerV2Cr attribute) FeedForward (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) ff_dropout (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) FIELD_TYPE_RANK (in module credit.datasets.channel_layout) figure_panel_planner() (in module credit.visualization_tools) file_dict (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) (credit.datasets.goes.GOESDataset attribute) filename (in module credit.datasets.era5_multistep) filename_diagnostic (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) filename_dyn_forcing (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) filename_forcing (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) filename_static (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) filename_surface (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) filenames (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.xr_sampler.XRSamplerByYear attribute) film (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.swin_wrf.WRFTransformer attribute) filter_ds() (in module credit.data) final_conv (credit.models.unet_diffusion.UnetDiffusion attribute) final_res_block (credit.models.unet_diffusion.UnetDiffusion attribute) find_common_indices() (in module credit.data) find_key() (in module credit.datasets.count_channels) find_key_for_number() (in module credit.data) find_nearest() (in module credit.pol_lapdiff_filt) find_start_stop_indices() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.wrf_singlestep.WRFPredict method) first_cycle_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) first_date (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) fit() (credit.trainers.base_trainer.BaseTrainer method) (trainer_multistep_deprecated.Trainer method) (trainerERA5_multistep_v1.Trainer method) fit_deprecated() (trainerERA5_v2.Trainer method) fit_era5_scaler_times() (in module scaler) fit_scaled_era5_time_residuals() (in module scaler) flag_diagnostic (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_dyn_forcing (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_energy_conserve (credit.ensemble.bred_vector.BredVector attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) flag_forcing (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_mass_conserve (credit.ensemble.bred_vector.BredVector attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) flag_static (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_surface (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_surface_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) flag_upper_air (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) flag_water_conserve (credit.ensemble.bred_vector.BredVector attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) flash (credit.attend.Attend attribute) flash_attn() (credit.attend.Attend method) flatten_list() (in module credit.data) flip_axis (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) flip_vertical (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] for_len (_transforms.ToTensor attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) for_len_outside (credit.transforms.transforms_wrf.ToTensorWRF attribute) forecast_details (in module credit.forecast) forecast_end_time (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_hour (credit.datasets.hrrr.HRRRDataset attribute) (credit.replay_buffer.ReplayBuffer attribute) forecast_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenDataLoader attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSamplerSamudra attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) (credit.skebs.SKEBS attribute) (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) forecast_len_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) forecast_period (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) forecast_start_time (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_step_count (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) forecast_step_counts (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) forecast_times (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_timestep (credit.datasets.realtime_predict.RealtimePredictDataset attribute) forecast_windows (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) ForecastProcessor (class in credit.trainers.ic_optimization) (class in rollout_samudra) forward() (credit.attend.Attend method) (credit.diffusion.GaussianDiffusion method) (credit.diffusion.ModifiedGaussianDiffusion method) (credit.domain_parallel.DomainParallelConv2d method) (credit.domain_parallel.DomainParallelConv3d method) (credit.domain_parallel.DomainParallelConvTranspose2d method) (credit.domain_parallel.DomainParallelConvTranspose3d method) (credit.domain_parallel.DomainParallelGroupNorm method) (credit.domain_parallel.DomainParallelInterpolate method) (credit.domain_parallel.DomainParallelPeriodicConv2d method) (credit.domain_parallel.halo_exchange._HaloExchangeFunction static method) (credit.domain_parallel.halo_exchange.HaloExchange method) (credit.domain_parallel.HaloExchange method) (credit.domain_parallel.layers.DomainParallelConv2d method) (credit.domain_parallel.layers.DomainParallelConv3d method) (credit.domain_parallel.layers.DomainParallelConvTranspose2d method) (credit.domain_parallel.layers.DomainParallelConvTranspose3d method) (credit.domain_parallel.layers.DomainParallelGroupNorm method) (credit.domain_parallel.layers.DomainParallelInterpolate method) (credit.domain_parallel.layers.DomainParallelPeriodicConv2d method) (credit.ensemble.spherical.SphericalRandomField method) (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss method) (credit.losses.covariance.CovarianceWeightedMSELoss method) (credit.losses.downscaling_loss.DownscalingLoss method) (credit.losses.DownscalingLoss method) (credit.losses.kcrps.KCRPSLoss method) (credit.losses.les_loss.LESLoss2D method) (credit.losses.logcosh.LogCoshLoss method) (credit.losses.msle.MSLELoss method) (credit.losses.power.PSDLoss method) (credit.losses.spectral.SpectralLoss2D method) (credit.losses.VariableTotalLoss2D method) (credit.losses.weighted_loss.VariableTotalLoss2D method) (credit.losses.xsigmoid.XSigmoidLoss method) (credit.losses.xtanh.XTanhLoss method) (credit.models.camulator.Attention method) (credit.models.camulator.Camulator method) (credit.models.camulator.CrossEmbedLayer method) (credit.models.camulator.CubeEmbedding method) (credit.models.camulator.DynamicPositionBias method) (credit.models.camulator.FeedForward method) (credit.models.camulator.LayerNorm method) (credit.models.camulator.Transformer method) (credit.models.camulator.UpBlock method) (credit.models.camulator.UpBlockPS method) (credit.models.checkpoint.ModelWrapper method) (credit.models.crossformer.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.debugger_model.DebuggerModel method) (credit.models.dscale_wrf.CubeEmbedding method) (credit.models.dscale_wrf.DownBlock method) (credit.models.dscale_wrf.DscaleTransformer method) (credit.models.dscale_wrf.UpBlock method) (credit.models.dscale_wrf.UTransformer method) (credit.models.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.swin.PatchEmbed method) (credit.models.swin.PatchMerging method) (credit.models.swin.SwinTransformerV2Cr method) (credit.models.swin.SwinTransformerV2CrBlock method) (credit.models.swin.SwinTransformerV2CrStage method) (credit.models.swin.WindowMultiHeadAttention method) (credit.models.swin.WindowMultiHeadAttentionNoPos method) (credit.models.swin_wrf.CubeEmbedding method) (credit.models.swin_wrf.DownBlock method) (credit.models.swin_wrf.UpBlock method) (credit.models.swin_wrf.UTransformer method) (credit.models.swin_wrf.WRFTransformer method) (credit.models.unet.SegmentationModel method) (credit.models.unet_attention_modules.CoordinateAttention method) (credit.models.unet_attention_modules.ECABlock method) (credit.models.unet_attention_modules.EfficientMixedAttention method) (credit.models.unet_attention_modules.LightSpatialAttention method) (credit.models.unet_attention_modules.SCSEAttention method) (credit.models.unet_diffusion.Attention method) (credit.models.unet_diffusion.Block method) (credit.models.unet_diffusion.LinearAttention method) (credit.models.unet_diffusion.PeriodicConv2d method) (credit.models.unet_diffusion.RandomOrLearnedSinusoidalPosEmb method) (credit.models.unet_diffusion.ResnetBlock method) (credit.models.unet_diffusion.RMSNorm method) (credit.models.unet_diffusion.SinusoidalPosEmb method) (credit.models.unet_diffusion.UnetDiffusion method) (credit.models.unet_downscaling.DownscalingSegmentationModel method) (credit.models.wxformer.crossformer.Attention method) (credit.models.wxformer.crossformer.CrossEmbedLayer method) (credit.models.wxformer.crossformer.CrossFormer method) (credit.models.wxformer.crossformer.CubeEmbedding method) (credit.models.wxformer.crossformer.DynamicPositionBias method) (credit.models.wxformer.crossformer.FeedForward method) (credit.models.wxformer.crossformer.LayerNorm method) (credit.models.wxformer.crossformer.Transformer method) (credit.models.wxformer.crossformer.UpBlock method) (credit.models.wxformer.crossformer.UpBlockPS method) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion method) (credit.models.wxformer.crossformer_diffusion.Transformer method) (credit.models.wxformer.crossformer_diffusion.UpBlock method) (credit.models.wxformer.crossformer_downscaling.Attention method) (credit.models.wxformer.crossformer_downscaling.CrossEmbedLayer method) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding method) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer method) (credit.models.wxformer.crossformer_downscaling.DynamicPositionBias method) (credit.models.wxformer.crossformer_downscaling.FeedForward method) (credit.models.wxformer.crossformer_downscaling.LayerNorm method) (credit.models.wxformer.crossformer_downscaling.Transformer method) (credit.models.wxformer.crossformer_downscaling.UpBlock method) (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise method) (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer method), [1] (credit.postblock.base.BasePostblock method) (credit.postblock.BridgeScalerTransformer method) (credit.postblock.gen1.GlobalEnergyFixer method) (credit.postblock.gen1.GlobalEnergyFixerUpDown method) (credit.postblock.gen1.GlobalMassFixer method) (credit.postblock.gen1.GlobalWaterFixer method) (credit.postblock.gen1.PostBlock method) (credit.postblock.gen1.TracerFixer method) (credit.postblock.geopotential.GeopotentialDiagnostic method) (credit.postblock.GeopotentialDiagnostic method) (credit.postblock.GlobalEnergyFixer method) (credit.postblock.GlobalMassFixer method) (credit.postblock.GlobalWaterFixer method) (credit.postblock.Reconstruct method) (credit.postblock.reconstruct.Reconstruct method) (credit.postblock.scaler.BridgeScalerTransformer method) (credit.postblock.TracerFixer method) (credit.postblock.wet_mask_samudra.WetMaskBlock method) (credit.postblock.WetMaskBlock method) (credit.preblock.base.BasePreblock method) (credit.preblock.concat.ConcatToTensor method) (credit.preblock.ConcatToTensor method) (credit.preblock.ERA5Normalizer method) (credit.preblock.log.LogTransform method) (credit.preblock.LogTransform method) (credit.preblock.norm.ERA5Normalizer method) (credit.preblock.regrid.Regridder method) (credit.preblock.Regridder method) (credit.preblock.scaler.BridgeScalerTransformer method) (credit.preblock.sqrt.SqrtTransform method) (credit.preblock.SqrtTransform 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) forward_head() (credit.models.swin.SwinTransformerV2Cr method) frames (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (credit.skebs.SKEBS attribute) freeze (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) freeze_pattern_weights (credit.skebs.SKEBS attribute) from_config() (credit.postblock.base.BasePostblock class method) (credit.preblock.base.BasePreblock class method) from_degrees() (credit.regrid.Grid class method) FSDPOptimizerWrapper (class in credit.models.checkpoint) full_pos_embed (credit.models.swin.SwinTransformerV2Cr attribute) full_state_pressure_interpolation() (in module credit.interp) Fuxi (class in credit.models.fuxi) G g (credit.models.camulator.LayerNorm attribute) (credit.models.crossformer.LayerNorm attribute) (credit.models.unet_diffusion.RMSNorm attribute) (credit.models.wxformer.crossformer.LayerNorm attribute) (credit.models.wxformer.crossformer_downscaling.LayerNorm attribute) gamma (credit.scheduler.CosineAnnealingWarmupRestarts attribute) GateCell (class in credit.models.graph) gated_unit (credit.models.graph.GraphResTransfGRU attribute) Gather (class in credit.trainers.trainerERA5_ensemble) gather_tensor() (in module credit.domain_parallel) (in module credit.domain_parallel.sharding) (in module credit.trainers.trainerERA5_ensemble) gaussian_grid (in module data_server) GaussianDiffusion (class in credit.diffusion) GaussianNoise (class in credit.ensemble.gaussian) gefs_init module generate_bred_vectors() (in module credit.ensemble.bred_vector) generate_bred_vectors_cycle() (in module credit.ensemble.bred_vector) generate_datetime() (in module credit.data) generate_forecasts() (in module credit.forecast) generate_integer_list_around() (in module credit.data) generate_interpolation_sequence() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) generate_scaled_ensemble() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) geopotential() (in module credit.postblock.geopotential) geopotential_from_model_vars() (in module credit.interp) GeopotentialDiagnostic (class in credit.postblock) (class in credit.postblock.geopotential) get_classifier() (credit.models.swin.SwinTransformerV2Cr method) get_climatology() (credit.metrics.LatWeightedMetricsClimatology method) get_colormap() (in module credit.visualization_tools) get_colormap_extend() (in module credit.visualization_tools) get_data() (in module ensemble_eval) get_domain_parallel_manager() (in module credit.domain_parallel) (in module credit.domain_parallel.manager) get_file_extension() (in module credit.models.checkpoint) get_forward_data() (in module credit.data) get_inference_field() (in module data_server) get_inference_steps() (in module credit.ocean.samudra_data) get_instance() (credit.ocean.samudra_constants.TensorMap class method) (credit.ocean.samudra_data.Normalize class method) get_lr() (credit.scheduler.CosineAnnealingWarmupRestarts method) (credit.scheduler.LinearWarmupCosineScheduler method) get_meta_file_path() (in module credit.metadata) get_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) get_norm_unnorm_dicts() (in module credit.ocean.samudra_data) get_num_cpus() (in module credit.diffusion_utils) (in module credit.pbs) get_pad2d() (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin_wrf) get_pad3d() (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin_wrf) get_projection() (in module credit.visualization_tools) get_psd() (credit.losses.power.PSDLoss method) get_rank_info() (in module credit.distributed) get_solar_index() (in module credit.solar) get_solar_radiation_loc() (in module credit.solar) get_stored_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) get_time_from (credit.datasets.downscaling_dataset.DownscalingDataset attribute) get_time_variable() (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) get_toa_radiation() (in module credit.solar) get_tsi() (in module credit.solar) get_variable_range() (in module credit.visualization_tools) get_variable_range_with_rounding() (in module credit.visualization_tools) getdata() (credit.datasets.downscaling_dataset.DownscalingDataset method) getgrad() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) getuv() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) gfs_init module gfs_map (in module credit.nwp) glob (credit.datasets.datamap.DataMap attribute) global_levels (credit.datasets.hrrr.HRRRDataset attribute) GlobalEnergyFixer (class in credit.postblock) (class in credit.postblock.gen1) GlobalEnergyFixerUpDown (class in credit.postblock.gen1) GlobalMassFixer (class in credit.postblock) (class in credit.postblock.gen1) GlobalWaterFixer (class in credit.postblock) (class in credit.postblock.gen1) goes_position (credit.datasets.goes.GOESDataset attribute) GOESDataset (class in credit.datasets.goes) gp_to_height_dam() (in module credit.animation) grad_checkpointing (credit.models.swin.SwinTransformerV2CrStage attribute) grad_max_norm (credit.trainers.base_trainer.BaseTrainer attribute) graph_blocks (credit.models.graph.GraphResTransfGRU attribute) graph_edges module graph_edges_knn module graph_norm_layers (credit.models.graph.GraphResTransfGRU attribute) GraphResidualBlock (class in credit.models.graph) GraphResTransfGRU (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_predictions() (credit.credit_ptype.CreditPostProcessor method) grid_size (credit.models.swin.PatchEmbed attribute) group_matcher() (credit.models.swin.SwinTransformerV2Cr method) H h_h_ln (credit.models.graph.GateCell attribute) h_x_ln (credit.models.graph.GateCell attribute) halo_exchange (credit.domain_parallel.DomainParallelConv2d attribute) (credit.domain_parallel.DomainParallelConv3d attribute) (credit.domain_parallel.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.DomainParallelConvTranspose3d attribute) (credit.domain_parallel.DomainParallelInterpolate attribute) (credit.domain_parallel.DomainParallelPeriodicConv2d attribute) (credit.domain_parallel.layers.DomainParallelConv2d attribute) (credit.domain_parallel.layers.DomainParallelConv3d attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose3d attribute) (credit.domain_parallel.layers.DomainParallelInterpolate attribute) (credit.domain_parallel.layers.DomainParallelPeriodicConv2d attribute) halo_width (credit.domain_parallel.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.DomainParallelConvTranspose3d attribute) (credit.domain_parallel.halo_exchange.HaloExchange attribute) (credit.domain_parallel.HaloExchange attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose3d attribute) HaloExchange (class in credit.domain_parallel) (class in credit.domain_parallel.halo_exchange) hamp (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) handle_signal() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) has_int_squareroot() (in module credit.diffusion_utils) havg (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) head (credit.models.swin.SwinTransformerV2Cr attribute) heads (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.graph.TransformerConv attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) hemispheric_rescale (credit.ensemble.bred_vector.BredVector attribute) (credit.ensemble.temporal.TemporalNoise attribute) hemispheric_rescale() (in module credit.ensemble.utils) hidden_size (credit.models.graph.GateCell attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) hist_len (_transforms.ToTensor attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) hist_len_outside (credit.transforms.transforms_wrf.ToTensorWRF attribute) historical_ERA5_images (credit.data.Sample attribute) history_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) (credit.diffusion.ModifiedGaussianDiffusion attribute) history_len_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) histroy_len (credit.models.graph.GraphResTransfGRU attribute) hour_to_nanoseconds() (in module credit.data) hours_of_day (trainerERA5_v1.TOADataLoader attribute) HR_input (credit.data.Sample_dscale attribute) HR_target (credit.data.Sample_dscale attribute) HRRRDataset (class in credit.datasets.hrrr) I ic_preblocks (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) Identity (class in credit.transforms_downscaling) identity() (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) IMAGE_ATTR_NAMES (in module credit.data) image_height (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.debugger_model) (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin) (in module credit.models.wxformer.crossformer) image_size (credit.diffusion.GaussianDiffusion attribute) image_width (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (in module credit.models.wxformer.crossformer_downscaling) img_size (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) immiscible (credit.diffusion.GaussianDiffusion attribute) in_channels (credit.models.graph.LayerNorm attribute) (credit.models.graph.TransformerConv attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) (credit.skebs.BackscatterUnet attribute) in_chans (credit.models.swin.SwinTransformerV2Cr attribute) in_features (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) index (credit.replay_buffer.ReplayBuffer attribute) index_subset (credit.samplers.MultiStepBatchSamplerSubset attribute) indices() (credit.regrid.NearestRegridder method) indpol (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) init_datetime (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) init_instance() (credit.ocean.samudra_constants.TensorMap class method) (credit.ocean.samudra_data.Normalize class method) init_lr() (credit.scheduler.CosineAnnealingWarmupRestarts method) init_register_all_fields() (credit.datasets.base_dataset.AbstractBaseDataset method) (credit.datasets.base_dataset.BaseDataset method) init_times (credit.samplers.MultiStepBatchSamplerSubset attribute) init_values (credit.models.swin.SwinTransformerV2CrBlock attribute) init_weights() (credit.models.swin.SwinTransformerV2CrBlock method) (in module credit.models.swin) initial_index (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) initialize_batch() (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher method) initialize_domain_parallel() (in module credit.domain_parallel) (in module credit.domain_parallel.manager) 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) inject_flat_var_keys() (in module credit.trainers.utils) inject_postblock_info() (in module credit.trainers.utils) input (credit.datasets.downscaling_dataset.Sample attribute) input_channels (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) input_length (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) input_only_channels (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.skebs.SKEBS attribute) input_resolution (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) input_shape (credit.replay_buffer.ReplayBuffer attribute) input_size (credit.models.graph.GraphResidualBlock attribute) input_static_dim (credit.ensemble.bred_vector.BredVector attribute) integral (credit.physics_core.physics_hybrid_sigma_level attribute) integration_steps (credit.ensemble.bred_vector.BredVector attribute) interp (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) interp_geopotential_to_pressure_levels() (in module credit.interp) interp_hybrid_to_height_agl() (in module credit.interp) interp_hybrid_to_hybrid_levels() (in module credit.interp) interp_hybrid_to_pressure_levels() (in module credit.interp) interp_pressure_to_hybrid_levels() (in module credit.interp) interp_temperature_to_pressure_levels() (in module credit.interp) interpolate() (credit.diffusion.GaussianDiffusion method) interpolate_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) interpolate_vertical_levels() (in module credit.gefs) inverse_sht (credit.ensemble.spherical.SphericalRandomField attribute) inverse_transform() (_transforms.NormalizeState method) (_transforms.NormalizeState_Quantile method) (_transforms.NormalizeTendency method) (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_quantile.BridgescalerScaleState method) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.transforms_wrf.NormalizeWRF method) inverse_transform_dataset() (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) inverse_transform_input() (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) is_ddim_sampling (credit.diffusion.GaussianDiffusion attribute) is_dtensor_checkpoint() (in module credit.models.checkpoint) is_first_domain_rank (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) is_fully_initialized (credit.physics_core.ModelLevelPressures attribute) is_last_domain_rank (credit.domain_parallel.DomainParallelManager property) (credit.domain_parallel.manager.DomainParallelManager property) is_safetensor_checkpoint() (in module credit.models.checkpoint) is_safetensors_available() (in module credit.models.checkpoint) is_training (credit.skebs.SKEBS attribute) iteration (credit.skebs.SKEBS attribute) iteration_stop (credit.skebs.SKEBS attribute) K k_to_c() (in module credit.animation) k_to_f() (in module credit.animation) KCRPSLoss (class in credit.losses.kcrps) keep_dataset_vars() (in module credit.data) key (credit.postblock.BridgeScalerTransformer attribute) (credit.postblock.scaler.BridgeScalerTransformer attribute) (credit.postblock.wet_mask_samudra.WetMaskBlock attribute) (credit.postblock.WetMaskBlock attribute) kgkg_to_gkg() (in module credit.animation) L last_date (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) lat (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) (credit.regrid.Grid attribute) lat_weights (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) (credit.replay_buffer.WeightedRMSE attribute) latitude_modes (credit.ensemble.spherical.SphericalRandomField attribute) latitude_slices (in module credit.verification.ensemble) latitude_weights() (in module credit.losses.weighted_loss) latlon2d_dir (credit.datasets.goes.GOESDataset attribute) latlon_grid (in module data_server) latlon_grid_path (credit.datasets.tisr.TISRDataset attribute) latlons (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) latN (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) LatWeightedMetrics (class in credit.metrics) LatWeightedMetricsClimatology (class in credit.metrics) LatWeightedMetricsEnsemble (class in credit.metrics) launch_script() (in module credit.pbs) launch_script_mpi() (in module credit.pbs) launch_script_torchrun() (in module credit.pbs) layer (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) LayerNorm (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.graph) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_downscaling) layers (credit.models.camulator.Camulator attribute) (credit.models.camulator.DynamicPositionBias attribute) (credit.models.camulator.FeedForward attribute) (credit.models.camulator.Transformer attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.DynamicPositionBias attribute) (credit.models.crossformer.FeedForward attribute) (credit.models.crossformer.Transformer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.DynamicPositionBias attribute) (credit.models.wxformer.crossformer.FeedForward attribute) (credit.models.wxformer.crossformer.Transformer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.Transformer attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_downscaling.DynamicPositionBias attribute) (credit.models.wxformer.crossformer_downscaling.FeedForward attribute) (credit.models.wxformer.crossformer_downscaling.Transformer attribute) lead_time_periods (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) left (credit.transforms_downscaling.Pad attribute) len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSampler attribute) (credit.datasets.load_dataset_and_dataloader.BatchForecastLenSamplerSamudra attribute) length_scale (credit.ensemble.spherical.SphericalNoise attribute) LES_file_indices (credit.datasets.les_singlestep.LESDataset attribute) LES_input (credit.data.Sample_LES attribute) LES_target (credit.data.Sample_LES attribute) LESDataset (class in credit.datasets.les_singlestep) LESLoss2D (class in credit.losses.les_loss) LESPredict (class in credit.datasets.les_singlestep) level_coord (credit.datasets.era5.ARCOERA5Dataset attribute) (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) (credit.datasets.local.LocalDataset attribute) level_info_file (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] level_map (in module credit.nwp) levels (_transforms.NormalizeState attribute) (_transforms.NormalizeState_Quantile attribute) (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) (credit.datasets.local.LocalDataset attribute) (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.postblock.geopotential.GeopotentialDiagnostic attribute) (credit.postblock.GeopotentialDiagnostic attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) (credit.skebs.BackscatterUnet attribute) (credit.skebs.SKEBS attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) levels_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) LH_ICE (in module credit.physics_constants) LH_WATER (in module credit.physics_constants) LightSpatialAttention (class in credit.models.unet_attention_modules) lin_key (credit.models.graph.TransformerConv attribute) lin_query (credit.models.graph.TransformerConv attribute) lin_value (credit.models.graph.TransformerConv attribute) linear (credit.models.debugger_model.DebuggerModel attribute) linear_beta_schedule() (in module credit.diffusion) LinearAttention (class in credit.models.unet_diffusion) LinearWarmupCosineScheduler (class in credit.scheduler) list_diag_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_dyn_forcing_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_stored_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) list_surf_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_surf_ds_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_upper_ds (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) list_upper_ds_outside (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) lmax (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.SKEBS attribute) load_custom_model_modules() (in module credit.models) load_data() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) load_dataloader() (in module credit.datasets.load_dataset_and_dataloader) (in module credit.trainers.utils) load_dataset() (in module credit.datasets.load_dataset_and_dataloader) (in module credit.trainers.utils) load_dataset_and_sampler() (in module train_les) (in module train_wrf) (in module train_wrf_multi) load_dataset_and_sampler_zscore_only() (in module train) load_forecasts() (in module credit.forecast) load_fsdp_or_checkpoint_policy() (in module credit.models) load_graph() (credit.models.graph.GraphResTransfGRU method) load_inputs() (credit.replay_buffer.ReplayBuffer method) load_loss() (credit.diffusion.ModifiedGaussianDiffusion method) (in module credit.losses) load_member_tiles() (in module credit.gefs) load_metadata() (in module credit.output) load_model() (credit.credit_ptype.CreditPostProcessor method) (credit.models.base_model.BaseModel class method) (in module credit.models) load_model_name() (credit.models.base_model.BaseModel class method) (in module credit.models) load_model_state() (in module credit.models.checkpoint) load_model_states_and_optimizer() (in module credit.trainers.utils) (in module train) (in module train_gen1) (in module train_les) (in module train_multistep) (in module train_wrf) (in module train_wrf_multi) load_premade_encoder_model() (in module credit.models.unet) (in module credit.models.unet_downscaling) (in module credit.skebs) load_scaler() (credit.credit_ptype.CreditPostProcessor method) load_scheduler() (in module credit.scheduler) load_state_dict() (credit.models.checkpoint.OptimizerWrapper method) (credit.trainers.base_trainer.EMATracker method) (in module credit.models.checkpoint) load_state_dict_error_handler() (in module credit.models.checkpoint) load_trainer() (in module credit.trainers) load_transform() (in module credit.datasets.om4_multistep_batcher) load_transforms() (in module credit.transforms) load_unet_attention() (in module credit.models.unet_attention_modules) load_unsharded_model() (credit.models.checkpoint.TorchFSDPCheckpointIO method) load_unsharded_optimizer() (credit.models.checkpoint.TorchFSDPCheckpointIO method) load_weights (credit.trainers.base_trainer.BaseTrainer attribute) load_zarr_as_input() (credit.data.Predict_Dataset method) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher method) (credit.datasets.les_singlestep.LESPredict method) (credit.datasets.wrf_singlestep.WRFPredict method) local_path (credit.datasets.hrrr_download._DownloadTask attribute) LocalDataset (class in credit.datasets.local) log_uniform_noise() (in module credit.diffusion) LogCoshLoss (class in credit.losses.logcosh) logger (in module _transforms) (in module credit.cli._ask) (in module credit.cli._common) (in module credit.cli._plot) (in module credit.cli._submit) (in module credit.datasets) (in module credit.datasets.era5_multistep) (in module credit.datasets.era5_multistep_batcher) (in module credit.datasets.hrrr) (in module credit.datasets.hrrr_download) (in module credit.datasets.mrms_download) (in module credit.datasets.multi_source) (in module credit.datasets.sequential_multistep) (in module credit.datasets.tisr) (in module credit.datasets.wrfmultistep) (in module credit.domain_parallel.convert) (in module credit.domain_parallel.manager) (in module credit.ensemble.bred_vector) (in module credit.forecast) (in module credit.losses) (in module credit.losses.base_losses) (in module credit.losses.downscaling_loss) (in module credit.losses.les_loss) (in module credit.losses.weighted_loss) (in module credit.metrics) (in module credit.models) (in module credit.models.base_model) (in module credit.models.camulator) (in module credit.models.crossformer) (in module credit.models.debugger_model) (in module credit.models.dscale_wrf) (in module credit.models.fuxi) (in module credit.models.swin) (in module credit.models.swin_wrf) (in module credit.models.unet) (in module credit.models.unet_downscaling) (in module credit.models.wxformer.crossformer) (in module credit.models.wxformer.crossformer_diffusion) (in module credit.models.wxformer.crossformer_downscaling) (in module credit.models.wxformer.crossformer_ensemble) (in module credit.models.wxformer.sdl_inference_wrapper) (in module credit.output) (in module credit.output_downscaling) (in module credit.pbs) (in module credit.postblock.gen1) (in module credit.preblock.norm) (in module credit.samplers) (in module credit.skebs) (in module credit.trainers) (in module credit.trainers.base_trainer) (in module credit.trainers.ic_optimization) (in module credit.trainers.preflight) (in module credit.trainers.rollout_utils) (in module credit.trainers.trainer_downscaling) (in module credit.trainers.trainer_gen2) (in module credit.trainers.trainer_om4_samudra) (in module credit.trainers.trainerERA5_Diffusion) (in module credit.trainers.trainerERA5_ensemble) (in module credit.trainers.trainerERA5gen1) (in module credit.trainers.trainerLES) (in module credit.trainers.trainerWRF) (in module credit.trainers.trainerWRF_multi) (in module credit.transforms) (in module credit.transforms.transforms_global) (in module credit.transforms.transforms_les) (in module credit.transforms.transforms_quantile) (in module credit.transforms.transforms_wrf) (in module credit.transforms_downscaling) (in module credit.verification.ensemble) (in module credit.verification.standard) (in module credit.visualization_tools) (in module rollout_downscaling) (in module rollout_les) (in module rollout_metrics) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) (in module rollout_realtime) (in module rollout_realtime_gen2) (in module rollout_samudra) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) (in module rollout_wrf) (in module trainerERA5_multistep_concat) (in module trainerERA5_multistep_sum) (in module trainerERA5_v1) (in module trainerERA5_v2) logit_scale (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) LogTransform (class in credit.preblock) (class in credit.preblock.log) lon (credit.physics_core.physics_hybrid_sigma_level attribute), [1] (credit.physics_core.physics_pressure_level attribute) (credit.regrid.Grid attribute) longitude_modes (credit.ensemble.spherical.SphericalRandomField attribute) lonN (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) lons (in module credit.solar) LR_input (credit.data.Sample_dscale attribute) lr_lambda_phase1() (in module credit.scheduler) lr_lambda_phase2() (in module credit.scheduler) M mae() (credit.metrics.LatWeightedMetricsClimatology method) main() (in module calc_global_solar) (in module credit.cli) (in module credit.cli._parser) (in module gefs_init) (in module gfs_init) (in module graph_edges) (in module graph_edges_knn) (in module model_summary) (in module rollout_les) (in module rollout_metrics) (in module rollout_realtime_gen2) (in module rollout_samudra) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) (in module rollout_wrf) (in module scaler) (in module train) (in module train_gen1) (in module train_les) (in module train_multistep) (in module train_wrf) (in module train_wrf_multi) main_cli() (in module rollout_realtime) (in module train_gen1) (in module train_gen2) make_single_source_subconfig() (in module credit.datasets.multi_source) make_xarray() (in module credit.output) make_xarray_diag() (in module credit.output) manager (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher attribute) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) map_gridline_opt() (in module credit.visualization_tools) MASK_VARS (in module credit.ocean.samudra_constants) max_forecast_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) max_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) mean (_transforms.NormalizeTendency attribute) (credit.transforms_downscaling.Zscore attribute) mean_ds (_transforms.NormalizeState attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) mean_ds_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) mean_sea_level_pressure() (in module credit.interp) mean_sea_level_pressure_simple() (in module credit.interp) mean_tensors (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) mean_tensors_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) mem_kv (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) merge_linear (credit.models.graph.GraphResidualBlock attribute) message() (credit.models.graph.TransformerConv method) meta_data (credit.trainers.ic_optimization.ForecastProcessor attribute) (rollout_samudra.ForecastProcessor attribute) meta_mlp (credit.models.swin.WindowMultiHeadAttention attribute) method (credit.postblock.BridgeScalerTransformer attribute) (credit.postblock.scaler.BridgeScalerTransformer attribute) (credit.preblock.scaler.BridgeScalerTransformer attribute) metric (rollout_metrics_noisy_model.Objective attribute) (train.Objective attribute) (train_gen1.Objective attribute) (train_les.Objective attribute) (train_multistep.Objective attribute) (train_wrf.Objective attribute) (train_wrf_multi.Objective attribute) metric_fn (credit.replay_buffer.ReplayBuffer attribute) mid_attn (credit.models.unet_diffusion.UnetDiffusion attribute) mid_block1 (credit.models.unet_diffusion.UnetDiffusion attribute) mid_block2 (credit.models.unet_diffusion.UnetDiffusion attribute) min_lr (credit.scheduler.CosineAnnealingWarmupRestarts attribute) (credit.scheduler.LinearWarmupCosineScheduler attribute) Minmax (class in credit.transforms_downscaling) mlp (credit.models.swin.SwinTransformerV2CrBlock attribute) (credit.models.unet_diffusion.ResnetBlock attribute) mmax (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.SKEBS attribute) (credit.transforms_downscaling.Minmax attribute) mmin (credit.transforms_downscaling.Minmax attribute) mod_level_store (credit.datasets.era5.ARCOERA5Dataset attribute) mode (credit.boundary_padding.TensorPadding attribute) (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) (credit.datasets.datamap.DataMap property) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.downscaling_dataset.DownscalingDataset property) (credit.datasets.era5.ARCOERA5Dataset attribute) (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) (credit.datasets.hrrr.HRRRDataset attribute) (credit.datasets.local.LocalDataset attribute) (credit.domain_parallel.DomainParallelInterpolate attribute) (credit.domain_parallel.layers.DomainParallelInterpolate attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.transforms_downscaling.Pad attribute) model (credit.diffusion.GaussianDiffusion attribute) (credit.ensemble.bred_vector.BredVector attribute) (credit.models.checkpoint.FSDPOptimizerWrapper attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper attribute) (credit.skebs.BackscatterUnet attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.trainerWRF_multi.TrainerWRFMulti attribute) (in module credit.models.checkpoint) (trainer_multistep_deprecated.Trainer attribute) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) model_a_half_var (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] model_b_half_var (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] model_lev_era5_path (credit.datasets.era5.ARCOERA5Dataset attribute) model_lev_vars (credit.datasets.era5.ARCOERA5Dataset attribute) model_predictions() (credit.diffusion.GaussianDiffusion method) (credit.diffusion.ModifiedGaussianDiffusion method) model_summary module ModelLevelPressures (class in credit.physics_core) ModelPrediction (class in credit.diffusion) (class in credit.models.wxformer.crossformer_diffusion) ModelWrapper (class in credit.models.checkpoint) ModifiedGaussianDiffusion (class in credit.diffusion) modulation (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer attribute), [1] module _transforms calc_global_solar credit credit._version credit.animation credit.attend credit.boundary_padding credit.cli credit.cli._ask credit.cli._common credit.cli._convert credit.cli._parser credit.cli._plot credit.cli._submit credit.credit_ptype credit.data credit.datasets credit.datasets._utils credit.datasets.base_dataset credit.datasets.channel_layout credit.datasets.count_channels credit.datasets.datamap credit.datasets.downscaling_dataset credit.datasets.era5 credit.datasets.era5_multistep credit.datasets.era5_multistep_batcher credit.datasets.era5_singlestep credit.datasets.goes credit.datasets.hrrr credit.datasets.hrrr_download credit.datasets.les_singlestep credit.datasets.load_dataset_and_dataloader credit.datasets.local credit.datasets.mrms credit.datasets.mrms_download credit.datasets.multi_source credit.datasets.om4_multistep_batcher credit.datasets.realtime_predict credit.datasets.sequential_multistep credit.datasets.tisr credit.datasets.wrf_singlestep credit.datasets.wrfmultistep credit.diffusion credit.diffusion_utils credit.distributed credit.domain_parallel credit.domain_parallel.convert credit.domain_parallel.halo_exchange credit.domain_parallel.layers credit.domain_parallel.manager credit.domain_parallel.sharding credit.ensemble credit.ensemble.bred_vector credit.ensemble.color credit.ensemble.crps credit.ensemble.gaussian credit.ensemble.spherical credit.ensemble.temporal credit.ensemble.utils credit.forecast credit.gefs credit.grid credit.interp credit.losses credit.losses.almost_fair_crps credit.losses.base_losses credit.losses.covariance credit.losses.downscaling_loss credit.losses.kcrps credit.losses.les_loss credit.losses.logcosh credit.losses.msle credit.losses.power credit.losses.spectral credit.losses.weighted_loss credit.losses.xsigmoid credit.losses.xtanh credit.metadata credit.metrics credit.metrics_downscaling credit.mixed_precision credit.models credit.models.base_model credit.models.camulator credit.models.checkpoint credit.models.crossformer credit.models.debugger_model credit.models.dscale_wrf credit.models.fuxi credit.models.graph credit.models.reset credit.models.swin credit.models.swin_wrf credit.models.unet credit.models.unet_attention_modules credit.models.unet_diffusion credit.models.unet_downscaling credit.models.wxformer credit.models.wxformer.crossformer credit.models.wxformer.crossformer_diffusion credit.models.wxformer.crossformer_downscaling credit.models.wxformer.crossformer_ensemble credit.models.wxformer.sdl_inference_wrapper credit.models.wxformer.stochastic_decomposition_layer credit.nwp credit.ocean credit.ocean.samudra_constants credit.ocean.samudra_data credit.output credit.output_downscaling credit.parser credit.pbs credit.physics_constants credit.physics_core credit.pol_lapdiff_filt credit.postblock credit.postblock.base credit.postblock.gen1 credit.postblock.geopotential credit.postblock.reconstruct credit.postblock.scaler credit.postblock.wet_mask_samudra credit.preblock credit.preblock.base credit.preblock.concat credit.preblock.log credit.preblock.norm credit.preblock.regrid credit.preblock.scaler credit.preblock.sqrt credit.regrid credit.replay_buffer credit.samplers credit.scheduler credit.seed credit.skebs credit.solar credit.trainers credit.trainers.base_trainer credit.trainers.ic_optimization credit.trainers.preflight credit.trainers.rollout_utils credit.trainers.trainer_downscaling credit.trainers.trainer_gen2 credit.trainers.trainer_om4_samudra credit.trainers.trainerERA5_Diffusion credit.trainers.trainerERA5_ensemble credit.trainers.trainerERA5gen1 credit.trainers.trainerLES credit.trainers.trainerWRF credit.trainers.trainerWRF_multi credit.trainers.utils credit.transforms credit.transforms.transforms_global credit.transforms.transforms_les credit.transforms.transforms_quantile credit.transforms.transforms_wrf credit.transforms_downscaling credit.verification credit.verification.ensemble credit.verification.standard credit.visualization_tools credit.xr_sampler data_server ensemble_eval gefs_init gfs_init graph_edges graph_edges_knn model_summary rollout_downscaling rollout_les rollout_metrics rollout_metrics_noisy_ic rollout_metrics_noisy_model rollout_realtime rollout_realtime_gen2 rollout_samudra rollout_to_netcdf rollout_to_netcdf_gen2 rollout_wrf scaler train train_gen1 train_gen2 train_les train_multistep train_wrf train_wrf_multi trainer_multistep_deprecated trainerERA5_multistep_concat trainerERA5_multistep_sum trainerERA5_multistep_v1 trainerERA5_v1 trainerERA5_v2 module (credit.models.checkpoint.ModelWrapper attribute) (credit.models.checkpoint.TorchFSDPModel attribute) (credit.trainers.trainerWRF_multi.TrainerWRFMulti attribute) (trainerERA5_multistep_concat.Trainer attribute) (trainerERA5_multistep_sum.Trainer attribute) MRMSDataset (class in credit.datasets.mrms) mse() (credit.metrics.LatWeightedMetricsClimatology method) MSLELoss (class in credit.losses.msle) MultiprocessingBatcher (class in credit.datasets.era5_multistep_batcher) MultiprocessingBatcherPrefetch (class in credit.datasets.era5_multistep_batcher) MultiSourceDataset (class in credit.datasets.multi_source) multistep (credit.skebs.SKEBS attribute) MultiStepBatchSamplerSubset (class in credit.samplers) N n_3dvar_levels (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) n_a (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) n_b (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) n_blocks (credit.models.graph.GraphResTransfGRU attribute) n_levels (credit.models.graph.GraphResTransfGRU attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) n_static_variables (credit.models.graph.GraphResTransfGRU attribute) n_surface_variables (credit.models.graph.GraphResTransfGRU attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) n_variables (credit.models.graph.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) neighbor_ranks() (credit.domain_parallel.DomainParallelManager method) (credit.domain_parallel.manager.DomainParallelManager method) next_n_hour() (in module credit.data) nlat (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterPrescribed attribute) (credit.skebs.BackscatterUnet attribute) (credit.skebs.SKEBS attribute) nlon (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.BackscatterCNN attribute) (credit.skebs.BackscatterPrescribed attribute) (credit.skebs.BackscatterUnet attribute) (credit.skebs.SKEBS attribute) no_autocast (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) noise_amplitude (credit.ensemble.bred_vector.BredVector attribute) noise_assignment() (credit.diffusion.GaussianDiffusion method) noise_factor (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer attribute), [1] noise_generator (credit.ensemble.temporal.TemporalNoise attribute) noise_inject1 (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) noise_inject2 (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) noise_inject3 (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute) noise_latent_dim (credit.models.wxformer.crossformer_ensemble.CrossFormerWithNoise attribute), [1] noise_transform (credit.models.wxformer.stochastic_decomposition_layer.StochasticDecompositionLayer attribute), [1] noop (credit.output_downscaling.OutputWriter attribute) norm (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.PatchMerging attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.Block attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) norm1 (credit.models.swin.SwinTransformerV2CrBlock attribute) norm2 (credit.models.swin.SwinTransformerV2CrBlock attribute) norm3 (credit.models.swin.SwinTransformerV2CrBlock attribute) norm_layer (credit.models.graph.GraphResidualBlock attribute) Normalize (class in credit.ocean.samudra_data) normalize (credit.datasets.datamap.DataMap attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.diffusion.GaussianDiffusion attribute) normalize_boundary() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) Normalize_ERA5_and_Forcing (class in credit.transforms.transforms_global) normalize_numpy_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) normalize_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) normalize_tensor_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) normalize_to_neg_one_to_one() (in module credit.diffusion_utils) (in module credit.models.wxformer.crossformer_diffusion) NormalizeLES (class in credit.transforms.transforms_les) NormalizeState (class in _transforms) NormalizeState_Quantile (class in _transforms) NormalizeState_Quantile_Bridgescalar (class in credit.transforms.transforms_quantile) NormalizeTendency (class in _transforms) NormalizeWRF (class in credit.transforms.transforms_wrf) num_boundary_vars (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) num_channels (credit.domain_parallel.DomainParallelGroupNorm attribute) (credit.domain_parallel.layers.DomainParallelGroupNorm attribute) num_cycles (credit.ensemble.bred_vector.BredVector attribute) num_epoch (credit.trainers.base_trainer.BaseTrainer attribute) num_epochs (in module credit.scheduler) num_features (credit.models.swin.SwinTransformerV2Cr attribute) num_fetch_workers (credit.datasets.hrrr.HRRRDataset attribute) num_forcing_static (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) num_forecast_steps (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) (credit.samplers.DistributedMultiStepBatchSampler attribute) (credit.samplers.MultiStepBatchSamplerSubset attribute) num_groups (credit.domain_parallel.DomainParallelGroupNorm attribute) (credit.domain_parallel.layers.DomainParallelGroupNorm attribute) num_heads (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) num_indices_per_rank (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) num_integration_steps (credit.datasets.tisr.TISRDataset attribute) num_patches (credit.models.swin.PatchEmbed attribute) num_prognostic_vars (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) num_replicas (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) num_start_batches (credit.samplers.MultiStepBatchSamplerSubset attribute) num_timesteps (credit.diffusion.GaussianDiffusion attribute) num_to_groups() (in module credit.diffusion_utils) num_upper_air (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) num_upper_air_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) num_workers (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher attribute) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) numpy_dir (credit.replay_buffer.ReplayBuffer attribute) O Objective (class in rollout_metrics_noisy_model) (class in train) (class in train_gen1) (class in train_les) (class in train_multistep) (class in train_wrf) (class in train_wrf_multi) objective (credit.diffusion.GaussianDiffusion attribute) Ocean_MultiStep_Batcher (class in credit.datasets.om4_multistep_batcher) Ocean_Tensor_Batcher (class in credit.datasets.om4_multistep_batcher) off_diagonal_scale (credit.losses.covariance.CovarianceWeightedMSELoss attribute) offset_noise_strength (credit.diffusion.GaussianDiffusion attribute) omega (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) OMEGA (in module credit.physics_constants) once() (in module credit.attend) one_shot (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) operations (credit.postblock.gen1.PostBlock attribute) opt_energy (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) opt_mass (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) opt_water (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) optim (credit.models.checkpoint.OptimizerWrapper attribute) OptimizerWrapper (class in credit.models.checkpoint) option (in module credit.datasets.era5_multistep_batcher) out_channels (credit.models.graph.TransformerConv attribute) out_chans (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) out_size (credit.models.graph.GraphResidualBlock attribute) output (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.downscaling_dataset.DownscalingDataset property) output_channels (credit.models.camulator.Camulator attribute) (credit.models.camulator.UpBlock attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.UpBlock attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.UpBlock attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) (credit.models.wxformer.crossformer_downscaling.UpBlock attribute) output_dir (credit.output_downscaling.OutputWrangler attribute) output_dtype (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) output_frames (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) output_length (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) output_name (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] output_only_channels (credit.models.debugger_model.DebuggerModel attribute) (credit.skebs.SKEBS attribute) OutputWrangler (class in credit.output_downscaling) OutputWriter (class in credit.output_downscaling) outside_file_indices (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) outside_file_year_range (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) overwrite (credit.datasets.hrrr_download._DownloadTask attribute) P p_losses() (credit.diffusion.GaussianDiffusion method) (credit.diffusion.ModifiedGaussianDiffusion method) p_mean_variance() (credit.diffusion.GaussianDiffusion method) p_sample() (credit.diffusion.GaussianDiffusion method) p_sample_loop() (credit.diffusion.GaussianDiffusion method) pa_to_hpa() (in module credit.animation) Pad (class in credit.transforms_downscaling) pad (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) (credit.skebs.BackscatterUnet attribute) pad() (credit.boundary_padding.TensorPadding method) (credit.skebs.BackscatterCNN method) pad_lat (credit.skebs.BackscatterCNN attribute) pad_lon (credit.skebs.BackscatterCNN attribute) pad_NS (credit.boundary_padding.TensorPadding attribute) pad_WE (credit.boundary_padding.TensorPadding attribute) padding (credit.domain_parallel.DomainParallelPeriodicConv2d attribute) (credit.domain_parallel.layers.DomainParallelPeriodicConv2d attribute) (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) (credit.models.unet_diffusion.PeriodicConv2d attribute) param_groups (credit.models.checkpoint.OptimizerWrapper property) parameters (credit.models.checkpoint.OptimizerWrapper property) parse_dtype() (in module credit.mixed_precision) parser (in module credit.datasets.hrrr_download) (in module credit.datasets.mrms_download) (in module credit.grid) (in module credit.models.wxformer.sdl_inference_wrapper) passthrough() (in module credit.losses.covariance) patch_embed (credit.models.swin.SwinTransformerV2Cr attribute) patch_height (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) patch_size (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) patch_width (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) PatchEmbed (class in credit.models.swin) patches_resolution (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) PatchMerging (class in credit.models.swin) PeriodicConv2d (class in credit.models.unet_diffusion) perturb() (credit.ensemble.bred_vector.BredVector method) perturb_channel_idx (credit.ensemble.bred_vector.BredVector attribute) perturbation_method (credit.ensemble.bred_vector.BredVector attribute) perturbation_std (credit.ensemble.temporal.TemporalNoise attribute) phased_lr_lambda() (in module credit.scheduler) physics_hybrid_sigma_level (class in credit.physics_core) physics_linear (credit.models.graph.GraphResidualBlock attribute) physics_pressure_level (class in credit.physics_core) PI (in module credit.physics_constants) (in module credit.postblock.gen1) plev_dim (credit.physics_core.ModelLevelPressures attribute) plot_global_animation() (in module credit.animation) plot_regional_animation() (in module credit.animation) polefilt_lap2d_QV1() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) polefilt_lap2d_V1() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) polefilt_lap2d_V2() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) polfiltT() (in module credit.pol_lapdiff_filt) pool_h (credit.models.unet_attention_modules.CoordinateAttention attribute) pool_w (credit.models.unet_attention_modules.CoordinateAttention attribute) populate() (credit.replay_buffer.ReplayBuffer method) post_conf (credit.ensemble.bred_vector.BredVector attribute) (credit.skebs.SKEBS attribute) PostBlock (class in credit.postblock.gen1) POSTBLOCK_REGISTRY (in module credit.postblock) Power (class in credit.transforms_downscaling) pre_out_dim (credit.models.unet_diffusion.UnetDiffusion attribute) PREBLOCK_REGISTRY (in module credit.preblock) precip_ind (credit.postblock.gen1.GlobalWaterFixer attribute) (credit.postblock.GlobalWaterFixer attribute) pred_noise (credit.diffusion.ModelPrediction attribute) (credit.models.wxformer.crossformer_diffusion.ModelPrediction attribute) pred_x_start (credit.diffusion.ModelPrediction attribute) (credit.models.wxformer.crossformer_diffusion.ModelPrediction attribute) predict() (in module rollout_downscaling) (in module rollout_les) (in module rollout_metrics) (in module rollout_metrics_noisy_ic) (in module rollout_metrics_noisy_model) (in module rollout_realtime) (in module rollout_samudra) (in module rollout_to_netcdf) (in module rollout_to_netcdf_gen2) (in module rollout_wrf) predict_data_check() (in module credit.parser) Predict_Dataset (class in credit.data) Predict_Dataset_Batcher (class in credit.datasets.era5_multistep_batcher) predict_noise_from_start() (credit.diffusion.GaussianDiffusion method) Predict_Ocean_Batcher (class in credit.datasets.om4_multistep_batcher) predict_start_from_noise() (credit.diffusion.GaussianDiffusion method) predict_start_from_v() (credit.diffusion.GaussianDiffusion method) predict_v() (credit.diffusion.GaussianDiffusion method) prefetch_batches() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) prefetch_factor (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) prefetch_queue (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) prefetch_thread (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) pres_level_store (credit.datasets.era5.ARCOERA5Dataset 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_lev_era5_path (credit.datasets.era5.ARCOERA5Dataset attribute) pressure_on_interfaces() (in module credit.postblock.geopotential) pressure_thickness (credit.physics_core.physics_pressure_level attribute) previous_hourly_steps() (in module credit.data) primary_main() (in module train_les) (in module train_wrf) (in module train_wrf_multi) print_once (in module credit.attend) process() (credit.output_downscaling.OutputWrangler method) (credit.trainers.ic_optimization.ForecastProcessor method) (rollout_samudra.ForecastProcessor method) process_forecast() (in module rollout_realtime) process_member() (in module credit.gefs) process_pressure_interp() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) product (credit.datasets.goes.GOESDataset attribute) (credit.datasets.hrrr.HRRRDataset attribute), [1] PROG_VARS_MAP (in module credit.ocean.samudra_constants) prognostic (credit.datasets.datamap.VarDict attribute) prognostic_mean (credit.datasets.om4_multistep_batcher.StandardScaler attribute) prognostic_std (credit.datasets.om4_multistep_batcher.StandardScaler attribute) proj (credit.models.camulator.CubeEmbedding attribute) (credit.models.crossformer.CubeEmbedding attribute) (credit.models.dscale_wrf.CubeEmbedding attribute) (credit.models.fuxi.CubeEmbedding attribute) (credit.models.swin.PatchEmbed attribute) (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) (credit.models.swin_wrf.CubeEmbedding attribute) (credit.models.unet_diffusion.Block attribute) (credit.models.wxformer.crossformer.CubeEmbedding attribute) (credit.models.wxformer.crossformer_downscaling.CubeEmbedding attribute) proj_drop (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) projections (in module credit.animation) ps (credit.models.camulator.UpBlockPS attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) PSDLoss (class in credit.losses.power) ptr (credit.replay_buffer.ReplayBuffer attribute) ptype_classification() (credit.credit_ptype.CreditPostProcessor method) Q q_ind_end (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.gen1.GlobalMassFixer attribute) (credit.postblock.gen1.GlobalWaterFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalMassFixer attribute) (credit.postblock.GlobalWaterFixer attribute) q_ind_start (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.gen1.GlobalMassFixer attribute) (credit.postblock.gen1.GlobalWaterFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) (credit.postblock.GlobalMassFixer attribute) (credit.postblock.GlobalWaterFixer attribute) Q_inds (credit.skebs.SKEBS attribute) q_posterior() (credit.diffusion.GaussianDiffusion method) q_sample() (credit.diffusion.GaussianDiffusion method) q_values (credit.replay_buffer.ReplayBuffer attribute) qc_path (credit.datasets.goes.GOESDataset attribute) qkv (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) R r_h_ln (credit.models.graph.GateCell attribute) r_x_ln (credit.models.graph.GateCell attribute) RAD_EARTH (in module credit.physics_constants) radius (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) random_or_learned_sinusoidal_cond (credit.models.unet_diffusion.UnetDiffusion attribute) RandomOrLearnedSinusoidalPosEmb (class in credit.models.unet_diffusion) rank (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) (credit.replay_buffer.ReplayBuffer attribute) (credit.trainers.base_trainer.BaseTrainer attribute) (credit.trainers.trainerWRF_multi.TrainerWRFMulti 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) read() (credit.datasets.datamap.DataMap method) read_scaler (in module _transforms) (in module credit.transforms.transforms_quantile) RealtimePredictDataset (class in credit.datasets.realtime_predict) rearrange() (credit.datasets.downscaling_dataset.DownscalingDataset method) Reconstruct (class in credit.postblock) (class in credit.postblock.reconstruct) reddening (credit.ensemble.color.ColorNoise attribute) reduction (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss attribute) (credit.losses.covariance.CovarianceWeightedMSELoss attribute) (credit.losses.logcosh.LogCoshLoss attribute) (credit.losses.msle.MSLELoss attribute) (credit.losses.spectral.SpectralLoss2D attribute) (credit.losses.xsigmoid.XSigmoidLoss attribute) (credit.losses.xtanh.XTanhLoss attribute) (credit.models.swin.PatchMerging attribute) reduction_function (credit.losses.covariance.CovarianceWeightedMSELoss attribute) reduction_functions (in module credit.losses.covariance) region (credit.datasets.mrms.MRMSDataset attribute) register_model() (in module credit.models) regrid_array() (credit.regrid.BilinearRegridder method) (credit.regrid.ConservativeRegridder method) (credit.regrid.NearestRegridder method) (credit.regrid.Regridder method) regrid_dataset() (credit.regrid.Regridder method) regrid_member() (in module credit.gefs) Regridder (class in credit.preblock) (class in credit.preblock.regrid) (class in credit.regrid) regridder (in module data_server) reload() (credit.replay_buffer.ReplayBuffer method) relu (credit.skebs.BackscatterUnet attribute) relu1 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) (credit.skebs.SKEBS attribute) relu2 (credit.skebs.BackscatterFCNN attribute) (credit.skebs.BackscatterFCNNWide attribute) relu3 (credit.skebs.BackscatterFCNNWide attribute) relu4 (credit.skebs.BackscatterFCNNWide attribute) remove_string_by_pattern() (in module credit.parser) rename_variables() (in module credit.gefs) rename_vars() (in module credit.ocean.samudra_data) RepeatingIndexSampler (class in credit.datasets.era5_multistep) (class in credit.datasets.wrfmultistep) replace_nested_key() (in module credit.parser) ReplayBuffer (class in credit.replay_buffer) represent_sequence() (credit.cli._convert._FlowSeqDumper method) res_conv (credit.models.unet_diffusion.ResnetBlock attribute) rescale() (in module credit.transforms_downscaling) rescale_minmax() (in module credit.datasets.datamap) reset() (credit.trainers.ic_optimization.TimeStepper method) reset_classifier() (credit.models.swin.SwinTransformerV2Cr method) reset_latent_capture() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) reset_model() (in module credit.models.reset) reset_parameters() (credit.models.graph.LayerNorm method) (credit.models.graph.TransformerConv method) reset_to_original() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) reshape_only() (credit.models.base_model.BaseModel method) (credit.models.checkpoint.TorchFSDPModel method) (in module credit.data) reshape_to_xy (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) residual (credit.models.swin.SwinTransformerV2Cr attribute) ResnetBlock (class in credit.models.unet_diffusion) resolution (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) results (credit.datasets.era5_multistep_batcher.MultiprocessingBatcher attribute) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) retain_graph (credit.skebs.SKEBS attribute) (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) return_target (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) revert() (credit.datasets.downscaling_dataset.DownscalingDataset method) RHO_WATER (in module credit.physics_constants) right (credit.transforms_downscaling.Pad attribute) rk4() (credit.models.camulator.Camulator method) (credit.models.crossformer.CrossFormer method) (credit.models.unet.SegmentationModel method) (credit.models.unet_downscaling.DownscalingSegmentationModel method) (credit.models.wxformer.crossformer.CrossFormer method) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer method) rk4_integration (credit.models.unet.SegmentationModel attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) rmse() (credit.metrics.LatWeightedMetricsClimatology method) rmse_scores (credit.replay_buffer.ReplayBuffer attribute) RMSNorm (class in credit.models.unet_diffusion) rng (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) rollout_downscaling module rollout_forecast() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) rollout_les module rollout_metrics module rollout_metrics_noisy_ic module rollout_metrics_noisy_model module rollout_p (credit.data.Predict_Dataset attribute) rollout_postblocks (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) rollout_realtime module rollout_realtime_gen2 module rollout_samudra module rollout_to_netcdf module rollout_to_netcdf_gen2 module rollout_wrf module root_weight (credit.models.graph.TransformerConv attribute) rootpath (credit.datasets.datamap.DataMap attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) route_to_dataset_class() (in module credit.datasets.multi_source) run_forecast() (in module rollout_realtime_gen2) run_gfs_init() (in module rollout_realtime_gen2) run_multiscale_control_experiment() (in module credit.models.wxformer.sdl_inference_wrapper) RVGAS (in module credit.physics_constants) S s3_uri (credit.datasets.hrrr_download._DownloadTask attribute) Sample (class in credit.data) (class in credit.datasets.downscaling_dataset) sample() (credit.diffusion.GaussianDiffusion method) (credit.replay_buffer.ReplayBuffer method) Sample_diag (class in credit.data) Sample_dscale (class in credit.data) sample_len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) Sample_LES (class in credit.data) Sample_WRF (class in credit.data) sampler (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) samples_dir (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) sampling_timesteps (credit.diffusion.GaussianDiffusion attribute) save() (credit.replay_buffer.ReplayBuffer method) save_backup_weights (credit.trainers.base_trainer.BaseTrainer attribute) save_best_weights (credit.trainers.base_trainer.BaseTrainer attribute) save_every_epoch (credit.trainers.base_trainer.BaseTrainer attribute) save_loc (credit.trainers.base_trainer.BaseTrainer attribute) save_metric_vars (credit.trainers.base_trainer.BaseTrainer attribute) save_model() (credit.models.base_model.BaseModel method) save_netcdf_clean() (in module credit.output) save_netcdf_diag() (in module credit.output) save_netcdf_increment() (in module credit.output) (in module credit.trainers.ic_optimization) save_state_dict() (in module credit.models.checkpoint) save_unsharded_model() (credit.models.checkpoint.TorchFSDPCheckpointIO method) save_unsharded_optimizer() (credit.models.checkpoint.TorchFSDPCheckpointIO method) save_vars (credit.credit_ptype.CreditPostProcessor attribute) scale (credit.attend.Attend attribute) (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.unet_diffusion.RMSNorm attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) scale_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) scale_latents_multilevel() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) scale_loss() (credit.models.checkpoint.OptimizerWrapper method) scaler module scaler (credit.postblock.BridgeScalerTransformer attribute) (credit.postblock.scaler.BridgeScalerTransformer attribute) (credit.preblock.scaler.BridgeScalerTransformer attribute) scaler_3d (_transforms.NormalizeState_Quantile attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_3ds (_transforms.NormalizeState_Quantile attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_df (_transforms.NormalizeState_Quantile attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_file (_transforms.NormalizeState_Quantile attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_path (credit.postblock.BridgeScalerTransformer attribute) (credit.postblock.scaler.BridgeScalerTransformer attribute) (credit.preblock.scaler.BridgeScalerTransformer attribute) scaler_surf (_transforms.NormalizeState_Quantile attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scaler_surfs (_transforms.NormalizeState_Quantile attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) scalers (in module scaler) scheduler_type (credit.trainers.base_trainer.BaseTrainer attribute) scrip_from_curvilinear() (in module credit.grid) scrip_from_netcdf() (in module credit.grid) scrip_from_rectilinear() (in module credit.grid) SCSEAttention (class in credit.models.unet_attention_modules) SDLWrapper (class in credit.models.wxformer.sdl_inference_wrapper) seed (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) seed_everything() (in module credit.seed) SegmentationModel (class in credit.models.unet) self_condition (credit.diffusion.GaussianDiffusion attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) sequential_attn (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) set_decoder_modulation() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_decoder_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_decoder_style_vector() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_encoder_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_epoch() (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep method) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher method) (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher method) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher method) (credit.datasets.sequential_multistep.DistributedSequentialDataset method) (credit.datasets.wrfmultistep.WRFMultiStep method) set_globals() (in module credit.datasets) set_grad_checkpointing() (credit.models.swin.SwinTransformerV2Cr method) set_manual_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) set_noise_factors() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) setup() (credit.trainers.trainer_downscaling.TrainerDownscaling method) (in module credit.distributed) (in module model_summary) setup_data_loading() (in module credit.datasets) shadow (credit.trainers.base_trainer.EMATracker attribute) shape (credit.regrid.Grid property) shard_batch() (in module credit.domain_parallel) (in module credit.domain_parallel.sharding) shard_dim (credit.domain_parallel.DomainParallelConv2d attribute) (credit.domain_parallel.DomainParallelConv3d attribute) (credit.domain_parallel.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.DomainParallelConvTranspose3d attribute) (credit.domain_parallel.DomainParallelInterpolate attribute) (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.DomainParallelPeriodicConv2d attribute) (credit.domain_parallel.layers.DomainParallelConv2d attribute) (credit.domain_parallel.layers.DomainParallelConv3d attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose2d attribute) (credit.domain_parallel.layers.DomainParallelConvTranspose3d attribute) (credit.domain_parallel.layers.DomainParallelInterpolate attribute) (credit.domain_parallel.layers.DomainParallelPeriodicConv2d attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) shard_tensor() (in module credit.domain_parallel) (in module credit.domain_parallel.sharding) shared_mem_draw_wrapper() (in module credit.visualization_tools) sharp (credit.models.camulator.UpBlock attribute) (credit.models.camulator.UpBlockPS attribute) (credit.models.wxformer.crossformer.UpBlockPS attribute) should_not_checkpoint() (in module credit.distributed) shuffle (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrfmultistep.RepeatingIndexSampler attribute) sigmoid (credit.models.unet_attention_modules.CoordinateAttention attribute) (credit.models.unet_attention_modules.ECABlock attribute) (credit.models.unet_attention_modules.LightSpatialAttention attribute) (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter attribute) (credit.skebs.BackscatterCNN attribute) sigmoid_beta_schedule() (in module credit.diffusion) sindex2dates() (credit.datasets.datamap.DataMap method) single_sample_forward() (credit.losses.almost_fair_crps.AlmostFairKCRPSLoss method) (credit.losses.kcrps.KCRPSLoss method) SinusoidalPosEmb (class in credit.models.unet_diffusion) size (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) (credit.domain_parallel.DomainParallelInterpolate attribute) (credit.domain_parallel.layers.DomainParallelInterpolate attribute) (credit.replay_buffer.ReplayBuffer attribute) size_full (credit.datasets.les_singlestep.LESDataset attribute) size_list (credit.datasets.les_singlestep.LESDataset attribute) SKEBS (class in credit.skebs) skip_nan_prune (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) skip_periods (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep.RepeatingIndexSampler attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) skip_target (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) skip_validation (credit.trainers.base_trainer.BaseTrainer attribute) smoothness (credit.ensemble.spherical.SphericalNoise attribute) source (credit.regrid.Regridder attribute) sp_index (credit.skebs.SKEBS attribute) spatial_attention (credit.models.unet_attention_modules.EfficientMixedAttention attribute) spec2grid() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) (credit.skebs.SKEBS method) spec_coef_is_initialized (credit.skebs.SKEBS attribute) specific_humidity_var (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] spectral_lambda_reg (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) spectral_wavenum_init (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) SpectralLoss2D (class in credit.losses.spectral) SphericalNoise (class in credit.ensemble.spherical) SphericalRandomField (class in credit.ensemble.spherical) split_and_reshape() (credit.models.base_model.BaseModel method) (in module credit.output) spread_error() (in module credit.verification.ensemble) SqrtTransform (class in credit.preblock) (class in credit.preblock.sqrt) 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) standard_normal_noise() (in module credit.diffusion) StandardScaler (class in credit.datasets.om4_multistep_batcher) start_datetime (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) start_epoch (credit.trainers.base_trainer.BaseTrainer attribute) state_dict() (credit.models.checkpoint.OptimizerWrapper method) (credit.trainers.base_trainer.EMATracker method) state_trans (credit.skebs.SKEBS attribute) state_transformer (rollout_samudra.ForecastProcessor attribute) state_vars (credit.models.graph.GraphResTransfGRU attribute) static_dim_size (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) static_inds (credit.skebs.SKEBS attribute) static_metadata (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) (credit.datasets.era5.ARCOERA5Dataset attribute) (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) (credit.datasets.goes.GOESDataset attribute) (credit.datasets.hrrr.HRRRDataset attribute), [1] (credit.datasets.local.LocalDataset attribute) (credit.datasets.mrms.MRMSDataset attribute) (credit.datasets.multi_source.MultiSourceDataset attribute), [1] (credit.datasets.tisr.TISRDataset attribute) static_source_key (credit.postblock.geopotential.GeopotentialDiagnostic attribute) (credit.postblock.GeopotentialDiagnostic attribute) static_variables (_transforms.ToTensor attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) std (_transforms.NormalizeTendency attribute) std_ds (_transforms.NormalizeState attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) std_ds_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) std_tensors (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) std_tensors_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) stdev (credit.transforms_downscaling.Zscore attribute) step (credit.trainers.base_trainer.EMATracker attribute) step() (credit.models.checkpoint.OptimizerWrapper method) (credit.scheduler.CosineAnnealingWarmupRestarts method) step_in_cycle (credit.scheduler.CosineAnnealingWarmupRestarts attribute) step_postblocks (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) step_preblocks (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) StochasticDecompositionLayer (class in credit.models.wxformer.stochastic_decomposition_layer) stop_after_epoch (credit.trainers.base_trainer.BaseTrainer attribute) stop_event (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) stop_signal (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch attribute) stopping_patience (credit.trainers.base_trainer.BaseTrainer attribute) store (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) store_latents() (credit.models.wxformer.sdl_inference_wrapper.SDLWrapper method) store_path (credit.datasets.era5.WeatherBench2ERA5Dataset attribute) subset_extent() (credit.credit_ptype.CreditPostProcessor method) subset_patch() (in module credit.data) supported_encoders (in module credit.models.unet) (in module credit.models.unet_downscaling) supported_models (in module credit.models.unet) (in module credit.models.unet_downscaling) (in module credit.skebs) surf_down_LW_ind (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) surf_down_solar_ind (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) surf_LH_ind (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) surf_LR_ind (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) surf_SH_ind (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) surf_solar_ind (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) surf_up_LW_ind (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) surf_up_solar_ind (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) surface (in module credit.nwp) surface_channels (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.skebs.SKEBS attribute) surface_files (credit.datasets.realtime_predict.RealtimePredictDataset attribute) surface_geopotential_var (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] surface_pressure (credit.physics_core.physics_hybrid_sigma_level attribute) surface_pressure_var (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] surface_variables (_transforms.NormalizeState attribute) (_transforms.NormalizeState_Quantile attribute) (_transforms.NormalizeTendency attribute) (_transforms.ToTensor attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) swap() (credit.trainers.base_trainer.EMATracker method) swin_from_yaml() (in module credit.models.swin) SwinTransformerV2Cr (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.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) T_ind_start (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) T_inds (credit.skebs.SKEBS attribute) target (credit.datasets.downscaling_dataset.Sample attribute) (credit.regrid.Regridder attribute) target_ERA5_images (credit.data.Sample attribute) target_shift_size (credit.models.swin.SwinTransformerV2CrBlock attribute) temperature_var (credit.postblock.geopotential.GeopotentialDiagnostic attribute), [1] (credit.postblock.GeopotentialDiagnostic attribute), [1] template_path (credit.output_downscaling.OutputWriter attribute) templates (credit.output_downscaling.OutputWrangler attribute) temporal_correlation (credit.ensemble.temporal.TemporalNoise attribute) temporal_mode (credit.datasets.base_dataset.BaseDataset attribute) TemporalNoise (class in credit.ensemble.temporal) TensorMap (class in credit.ocean.samudra_constants) TensorPadding (class in credit.boundary_padding) theta (credit.models.unet_diffusion.SinusoidalPosEmb attribute) time_coord (credit.datasets.local.LocalDataset attribute) time_emb_proj (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) time_encode (credit.data.Sample_diag attribute) (credit.data.Sample_dscale attribute) (credit.data.Sample_WRF attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.swin_wrf.WRFTransformer attribute) time_mlp (credit.models.unet_diffusion.UnetDiffusion attribute) time_steps (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) time_to_emb (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) times_b (credit.replay_buffer.TOADataLoader attribute) (trainerERA5_v1.TOADataLoader attribute) TimeStepper (class in credit.trainers.ic_optimization) TISRDataset (class in credit.datasets.tisr) tnames (credit.datasets.downscaling_dataset.DownscalingDataset attribute) to() (credit.models.graph.GraphResTransfGRU method) to_device (credit.preblock.concat.ConcatToTensor attribute) (credit.preblock.ConcatToTensor attribute) to_emb (credit.models.wxformer.crossformer_diffusion.UpBlock attribute) to_out (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) to_qkv (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.unet_diffusion.Attention attribute) (credit.models.unet_diffusion.LinearAttention attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) to_tensor() (credit.datasets.downscaling_dataset.DownscalingDataset method) TOA (credit.replay_buffer.TOADataLoader attribute) (trainerERA5_v1.TOADataLoader attribute) TOA_down_solar_ind (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) TOA_OLR_ind (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) TOA_solar_ind (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.GlobalEnergyFixer attribute) TOA_up_OLR_ind (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) TOA_up_solar_ind (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) TOADataLoader (class in credit.replay_buffer) (class in trainerERA5_v1) tolerance (credit.datasets.goes.GOESDataset attribute) top (credit.transforms_downscaling.Pad attribute) TorchFSDPCheckpointIO (class in credit.models.checkpoint) TorchFSDPModel (class in credit.models.checkpoint) total_column_water() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) total_dim (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.swin_wrf.WRFTransformer attribute) total_dry_air_mass() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) total_input_channels (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) total_len (credit.datasets.les_singlestep.LESDataset attribute) total_length (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) total_n_vars (credit.models.graph.GraphResTransfGRU attribute) total_seq_len (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) total_steps (credit.scheduler.LinearWarmupCosineScheduler attribute) ToTensor (class in _transforms) ToTensor_BridgeScaler (class in credit.transforms.transforms_quantile) ToTensor_ERA5_and_Forcing (class in credit.transforms.transforms_global) ToTensorLES (class in credit.transforms.transforms_les) ToTensorWRF (class in credit.transforms.transforms_wrf) tracer_indices (credit.postblock.gen1.TracerFixer attribute) (credit.postblock.TracerFixer attribute) tracer_thres (credit.postblock.gen1.TracerFixer attribute) (credit.postblock.TracerFixer attribute) tracer_thres_max (credit.postblock.gen1.TracerFixer attribute) (credit.postblock.TracerFixer attribute) TracerFixer (class in credit.postblock) (class in credit.postblock.gen1) train module train() (rollout_metrics_noisy_model.Objective method) (train.Objective method) (train_gen1.Objective method) (train_les.Objective method) (train_multistep.Objective method) (train_wrf.Objective method) (train_wrf_multi.Objective method) train_alpha (credit.skebs.SKEBS attribute) train_gen1 module train_gen2 module train_les module train_multistep module train_one_epoch() (credit.replay_buffer.Trainer method) (credit.trainers.base_trainer.BaseTrainer method) (credit.trainers.ic_optimization.TrainerIC method) (credit.trainers.trainer_downscaling.TrainerDownscaling method) (credit.trainers.trainer_gen2.TrainerERA5Gen2 method) (credit.trainers.trainer_om4_samudra.TrainerSamudra method) (credit.trainers.trainerERA5_Diffusion.TrainerERA5Diffusion method) (credit.trainers.trainerERA5_ensemble.TrainerERA5Ensemble method) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 method) (credit.trainers.trainerLES.TrainerLES method) (credit.trainers.trainerWRF.TrainerWRF method) (credit.trainers.trainerWRF_multi.TrainerWRFMulti method) (trainer_multistep_deprecated.Trainer method) (trainerERA5_multistep_concat.Trainer method) (trainerERA5_multistep_sum.Trainer method) (trainerERA5_multistep_v1.Trainer method) (trainerERA5_v1.Trainer method) (trainerERA5_v2.Trainer method) train_one_epoch_mode (credit.trainers.base_trainer.BaseTrainer attribute) train_wrf module train_wrf_multi module Trainer (class in credit.replay_buffer) (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) (in module credit.trainers.trainer_gen2) (in module credit.trainers.trainerERA5gen1) trainer_multistep_deprecated module trainer_types (in module credit.trainers) TrainerDownscaling (class in credit.trainers.trainer_downscaling) trainerERA5_multistep_concat module trainerERA5_multistep_sum module trainerERA5_multistep_v1 module trainerERA5_v1 module trainerERA5_v2 module TrainerERA5Diffusion (class in credit.trainers.trainerERA5_Diffusion) TrainerERA5Ensemble (class in credit.trainers.trainerERA5_ensemble) TrainerERA5Gen1 (class in credit.trainers.trainerERA5gen1) TrainerERA5Gen2 (class in credit.trainers.trainer_gen2) TrainerIC (class in credit.trainers.ic_optimization) TrainerLES (class in credit.trainers.trainerLES) TrainerSamudra (class in credit.trainers.trainer_om4_samudra) TrainerWRF (class in credit.trainers.trainerWRF) TrainerWRFMulti (class in credit.trainers.trainerWRF_multi) training_data_check() (in module credit.parser) training_loss (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) training_metric (credit.trainers.base_trainer.BaseTrainer attribute) transform (credit.data.ERA5_and_Forcing_Dataset attribute) (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESDataset attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) transform() (_transforms.NormalizeState method) (_transforms.NormalizeState_Quantile method) (_transforms.NormalizeTendency method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_quantile.BridgescalerScaleState method) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar method) (credit.transforms.transforms_wrf.NormalizeWRF method) transform_array() (_transforms.NormalizeState method) (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) (credit.transforms.transforms_les.NormalizeLES method) (credit.transforms.transforms_quantile.BridgescalerScaleState method) (credit.transforms.transforms_wrf.NormalizeWRF method) transform_data() (credit.credit_ptype.CreditPostProcessor method) transform_dataset() (_transforms.NormalizeState method) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing method) transform_era5_times() (in module scaler) Transformer (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_diffusion) (class in credit.models.wxformer.crossformer_downscaling) transformer (credit.models.graph.GraphResidualBlock attribute) TransformerConv (class in credit.models.graph) transforms (credit.transforms_downscaling.DataTransforms attribute) trim() (credit.domain_parallel.halo_exchange.HaloExchange static method) (credit.domain_parallel.HaloExchange static method) tropics_only_dissipation (credit.skebs.SKEBS attribute) tsi_data (in module credit.solar) U U_ind_end (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) U_ind_start (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) U_inds (credit.skebs.SKEBS attribute) u_transformer (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) unet_config (in module credit.models.unet_diffusion) UnetDiffusion (class in credit.models.unet_diffusion) unnormalize (credit.diffusion.GaussianDiffusion attribute) unnormalize_numpy_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) unnormalize_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) unnormalize_tensor_prognostics() (credit.datasets.om4_multistep_batcher.StandardScaler method) (credit.ocean.samudra_data.Normalize method) unnormalize_to_zero_to_one() (in module credit.diffusion_utils) unpad() (credit.boundary_padding.TensorPadding method) (credit.skebs.BackscatterCNN method) unscale_grad() (credit.models.checkpoint.OptimizerWrapper method) unstagger_winds() (in module credit.gefs) unused (credit.datasets.datamap.VarDict attribute) UnWeightedMetrics (class in credit.metrics_downscaling) unwrap() (credit.models.checkpoint.ModelWrapper method) (credit.models.checkpoint.OptimizerWrapper method) (credit.models.checkpoint.TorchFSDPModel method) unwrap_model() (credit.models.checkpoint.FSDPOptimizerWrapper method) up (credit.models.dscale_wrf.UTransformer attribute) (credit.models.fuxi.UTransformer attribute) (credit.models.swin_wrf.UTransformer attribute) up_block1 (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) up_block2 (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) up_block3 (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) up_block4 (credit.models.camulator.Camulator attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) UpBlock (class in credit.models.camulator) (class in credit.models.crossformer) (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) (class in credit.models.wxformer.crossformer) (class in credit.models.wxformer.crossformer_diffusion) (class in credit.models.wxformer.crossformer_downscaling) UpBlockPS (class in credit.models.camulator) (class in credit.models.wxformer.crossformer) update() (credit.replay_buffer.ReplayBuffer method) (credit.trainers.base_trainer.EMATracker 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) 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) update_x() (in module credit.datasets.channel_layout) upper_air (in module credit.nwp) upper_air_pressure (credit.physics_core.physics_pressure_level attribute) ups (credit.models.unet_diffusion.UnetDiffusion attribute) Upsample() (in module credit.models.unet_diffusion) upsample_v_conv (credit.models.crossformer.CrossFormer attribute) (credit.models.crossformer.UpBlock attribute) upsample_with_ps (credit.models.wxformer.crossformer.CrossFormer attribute) use_edge_attr (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) use_interp (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) use_noise (credit.models.fuxi.UTransformer attribute) use_padding (credit.ensemble.spherical.SphericalNoise attribute) (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) use_post_block (credit.ensemble.bred_vector.BredVector attribute) (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.debugger_model.DebuggerModel attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.unet.SegmentationModel attribute) (credit.models.unet_diffusion.UnetDiffusion attribute) (credit.models.unet_downscaling.DownscalingSegmentationModel attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) use_power_loss (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) use_scheduler (credit.trainers.base_trainer.BaseTrainer attribute) use_spectral_loss (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) use_spectral_norm (credit.models.camulator.Camulator attribute) (credit.models.crossformer.CrossFormer attribute) (credit.models.dscale_wrf.DscaleTransformer attribute) (credit.models.fuxi.Fuxi attribute) (credit.models.graph.GraphResidualBlock attribute) (credit.models.graph.GraphResTransfGRU attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin_wrf.WRFTransformer attribute) (credit.models.wxformer.crossformer.CrossFormer attribute) (credit.models.wxformer.crossformer_diffusion.CrossFormerDiffusion attribute) (credit.models.wxformer.crossformer_downscaling.DownscalingCrossFormer attribute) use_statics (credit.skebs.SKEBS attribute) UTransformer (class in credit.models.dscale_wrf) (class in credit.models.fuxi) (class in credit.models.swin_wrf) V V_ind_end (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) V_ind_start (credit.postblock.gen1.GlobalEnergyFixer attribute) (credit.postblock.gen1.GlobalEnergyFixerUpDown attribute) (credit.postblock.GlobalEnergyFixer attribute) V_inds (credit.skebs.SKEBS attribute) valid_batches_per_epoch (credit.trainers.base_trainer.BaseTrainer attribute) VALID_DATA_TYPES (credit.preblock.base.BasePreblock attribute) VALID_FIELD_TYPES (in module credit.datasets.base_dataset) valid_forecast_len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.skebs.SKEBS attribute) (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) valid_history_len (credit.datasets.downscaling_dataset.DownscalingDataset attribute) (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) valid_indices (credit.datasets.om4_multistep_batcher.Predict_Ocean_Batcher attribute) VALID_PRODUCTS (in module credit.datasets.hrrr) validate() (credit.trainers.base_trainer.BaseTrainer method) (credit.trainers.ic_optimization.TrainerIC method) (credit.trainers.trainer_downscaling.TrainerDownscaling method) (credit.trainers.trainer_gen2.TrainerERA5Gen2 method) (credit.trainers.trainer_om4_samudra.TrainerSamudra method) (credit.trainers.trainerERA5_Diffusion.TrainerERA5Diffusion method) (credit.trainers.trainerERA5_ensemble.TrainerERA5Ensemble method) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 method) (credit.trainers.trainerLES.TrainerLES method) (credit.trainers.trainerWRF.TrainerWRF method) (credit.trainers.trainerWRF_multi.TrainerWRFMulti method), [1] (trainer_multistep_deprecated.Trainer method) (trainerERA5_multistep_concat.Trainer method) (trainerERA5_multistep_sum.Trainer method) (trainerERA5_multistep_v1.Trainer method) (trainerERA5_v1.Trainer method) (trainerERA5_v2.Trainer method) validate_args() (in module credit.parser) validate_data() (in module credit.ocean.samudra_data) validate_sharding_constraints() (in module credit.domain_parallel) (in module credit.domain_parallel.convert) validation (credit.losses.downscaling_loss.DownscalingLoss attribute) (credit.losses.DownscalingLoss attribute) (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) var_dict (credit.datasets.base_dataset.AbstractBaseDataset attribute) (credit.datasets.base_dataset.BaseDataset attribute) (credit.datasets.goes.GOESDataset attribute) var_levels (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) VAR_REGISTRY (in module credit.datasets.hrrr) var_weights (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) VarDict (class in credit.datasets.datamap) variable_transforms (in module credit.animation) variable_weights() (in module credit.losses.weighted_loss) variables (_transforms.NormalizeState attribute) (_transforms.NormalizeState_Quantile attribute) (_transforms.NormalizeTendency attribute) (_transforms.ToTensor attribute) (credit.datasets.datamap.DataMap attribute) (credit.output_downscaling.OutputWriter attribute) (credit.postblock.BridgeScalerTransformer attribute) (credit.postblock.scaler.BridgeScalerTransformer attribute) (credit.preblock.log.LogTransform attribute) (credit.preblock.LogTransform attribute) (credit.preblock.regrid.Regridder attribute) (credit.preblock.Regridder attribute) (credit.preblock.scaler.BridgeScalerTransformer attribute) (credit.preblock.sqrt.SqrtTransform attribute) (credit.preblock.SqrtTransform attribute) (credit.transforms.transforms_quantile.BridgescalerScaleState attribute) (credit.transforms.transforms_quantile.NormalizeState_Quantile_Bridgescalar attribute) (credit.transforms.transforms_quantile.ToTensor_BridgeScaler attribute) (credit.xr_sampler.XRSamplerByYear attribute) VariableTotalLoss2D (class in credit.losses) (class in credit.losses.weighted_loss) variance_scale (credit.ensemble.spherical.SphericalNoise attribute) varname_diagnostic (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_dyn_forcing (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_forcing (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) varname_static (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_surface (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) varname_upper_air (credit.data.ERA5_Dataset_Distributed attribute) (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.transforms.transforms_global.Normalize_ERA5_and_Forcing attribute) (credit.transforms.transforms_global.ToTensor_ERA5_and_Forcing attribute) (credit.transforms.transforms_les.NormalizeLES attribute) (credit.transforms.transforms_les.ToTensorLES attribute) (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) varname_upper_air_outside (credit.transforms.transforms_wrf.NormalizeWRF attribute) (credit.transforms.transforms_wrf.ToTensorWRF attribute) varnum_diag (credit.ensemble.bred_vector.BredVector attribute) (credit.trainers.trainer_gen2.TrainerERA5Gen2 attribute) (credit.trainers.trainerERA5gen1.TrainerERA5Gen1 attribute) vars (credit.losses.les_loss.LESLoss2D attribute) (credit.losses.VariableTotalLoss2D attribute) (credit.losses.weighted_loss.VariableTotalLoss2D attribute) (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.metrics_downscaling.UnWeightedMetrics attribute) version (in module credit._version) version_tuple (in module credit._version) vrtdivspec() (credit.pol_lapdiff_filt.Diffusion_and_Pole_Filter method) W w_lat (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.metrics_downscaling.UnWeightedMetrics attribute) w_var (credit.metrics.LatWeightedMetrics attribute) (credit.metrics.LatWeightedMetricsClimatology attribute) (credit.metrics.LatWeightedMetricsEnsemble attribute) (credit.metrics_downscaling.UnWeightedMetrics attribute) warmup_steps (credit.scheduler.CosineAnnealingWarmupRestarts attribute) (credit.scheduler.LinearWarmupCosineScheduler attribute) wavenum_init (credit.losses.power.PSDLoss attribute) (credit.losses.spectral.SpectralLoss2D attribute) WeatherBench2ERA5Dataset (class in credit.datasets.era5) weight (credit.domain_parallel.DomainParallelConv2d property) (credit.domain_parallel.DomainParallelPeriodicConv2d property) (credit.domain_parallel.layers.DomainParallelConv2d property) (credit.domain_parallel.layers.DomainParallelPeriodicConv2d property) (credit.models.graph.LayerNorm attribute) weighted_sum() (credit.physics_core.physics_hybrid_sigma_level method) (credit.physics_core.physics_pressure_level method) WeightedRMSE (class in credit.replay_buffer) weights (credit.models.unet_diffusion.RandomOrLearnedSinusoidalPosEmb attribute) wet (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) wet_mask (credit.datasets.om4_multistep_batcher.StandardScaler attribute) wet_surface (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) WetMaskBlock (class in credit.postblock) (class in credit.postblock.wet_mask_samudra) which_forecast (credit.data.Predict_Dataset attribute) window_area (credit.models.swin.SwinTransformerV2CrBlock attribute) window_partition() (in module credit.models.swin) window_reverse() (in module credit.models.swin) window_size (credit.models.camulator.Attention attribute) (credit.models.crossformer.Attention attribute) (credit.models.swin.SwinTransformerV2Cr attribute) (credit.models.swin.WindowMultiHeadAttention attribute) (credit.models.swin.WindowMultiHeadAttentionNoPos attribute) (credit.models.wxformer.crossformer.Attention attribute) (credit.models.wxformer.crossformer_downscaling.Attention attribute) WindowMultiHeadAttention (class in credit.models.swin) WindowMultiHeadAttentionNoPos (class in credit.models.swin) worker (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) worker() (in module credit.datasets.era5_multistep) (in module credit.datasets.sequential_multistep) (in module credit.datasets.wrfmultistep) worker_process() (credit.datasets.era5_multistep_batcher.MultiprocessingBatcherPrefetch method) world_size (credit.data.Predict_Dataset attribute) (credit.datasets.era5_multistep.ERA5_and_Forcing_MultiStep attribute) (credit.datasets.era5_multistep_batcher.ERA5_MultiStep_Batcher attribute) (credit.datasets.era5_multistep_batcher.Predict_Dataset_Batcher attribute) (credit.datasets.les_singlestep.LESPredict attribute) (credit.datasets.om4_multistep_batcher.Ocean_MultiStep_Batcher attribute) (credit.datasets.om4_multistep_batcher.Ocean_Tensor_Batcher attribute) (credit.datasets.realtime_predict.RealtimePredictDataset attribute) (credit.datasets.sequential_multistep.DistributedSequentialDataset attribute) (credit.datasets.wrf_singlestep.WRFPredict attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) (credit.domain_parallel.DomainParallelManager attribute) (credit.domain_parallel.manager.DomainParallelManager attribute) WRF_file_indices (credit.datasets.wrf_singlestep.WRFDataset attribute) (credit.datasets.wrfmultistep.WRFMultiStep attribute) WRF_input (credit.data.Sample_diag attribute) (credit.data.Sample_WRF attribute) WRF_target (credit.data.Sample_diag attribute) (credit.data.Sample_WRF attribute) WRFDataset (class in credit.datasets.wrf_singlestep) WRFMultiStep (class in credit.datasets.wrfmultistep) WRFPredict (class in credit.datasets.wrf_singlestep) WRFTransformer (class in credit.models.swin_wrf) write() (credit.output_downscaling.OutputWriter method) write_every (credit.skebs.SKEBS attribute) write_rollout_debug_files (credit.skebs.SKEBS attribute) write_to_netcdf() (credit.credit_ptype.CreditPostProcessor method) write_train_debug_files (credit.skebs.SKEBS attribute) X x_slice (credit.datasets.goes.GOESDataset attribute) xclassdict (credit.transforms_downscaling.DataTransforms attribute) XRSamplerByYear (class in credit.xr_sampler) XSigmoidLoss (class in credit.losses.xsigmoid) XTanhLoss (class in credit.losses.xtanh) Y y_slice (credit.datasets.goes.GOESDataset attribute) Z z_h_ln (credit.models.graph.GateCell attribute) z_x_ln (credit.models.graph.GateCell attribute) zero_grad() (credit.models.checkpoint.OptimizerWrapper method) zero_out_levels_top_of_model (credit.skebs.SKEBS attribute) zonal_spectrum() (in module credit.verification.standard) Zscore (class in credit.transforms_downscaling) zstride (credit.datasets.datamap.DataMap attribute) (credit.output_downscaling.OutputWriter attribute)
