API Reference

This page contains auto-generated API reference documentation [1].

• train
• 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.trainerERA5_Diffusion
    • credit.trainers.trainerERA5_ensemble
    • credit.trainers.trainerERA5gen1
    • credit.trainers.trainerLES
    • credit.trainers.trainerWRF
    • credit.trainers.trainerWRF_multi
    • credit.trainers.trainer_downscaling
    • credit.trainers.trainer_gen2
    • credit.trainers.trainer_om4_samudra
    • 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
• scaler
• gfs_init
• train_wrf
• gefs_init
• train_les
• train_gen1
• train_gen2
• data_server
• graph_edges
• rollout_les
• rollout_wrf
• _transforms
• ensemble_eval
• model_summary
• trainerERA5_v2
• trainerERA5_v1
• rollout_samudra
• rollout_metrics
• train_wrf_multi
• graph_edges_knn
• train_multistep
• rollout_realtime
• rollout_to_netcdf
• calc_global_solar
• rollout_downscaling
• rollout_realtime_gen2
• rollout_to_netcdf_gen2
• rollout_metrics_noisy_ic
• trainerERA5_multistep_v1
• trainerERA5_multistep_sum
• rollout_metrics_noisy_model
• trainerERA5_multistep_concat
• trainer_multistep_deprecated

[1]

Created with sphinx-autoapi