API Reference

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

• credit
  • credit.animation
  • credit.applications
    • credit.applications.calc_global_solar
    • credit.applications.data_server
    • credit.applications.ensemble_eval
    • credit.applications.gefs_init
    • credit.applications.gfs_init
    • credit.applications.graph_edges
    • credit.applications.graph_edges_knn
    • credit.applications.preprocess
    • credit.applications.rollout_downscaling
    • credit.applications.rollout_gen2
    • credit.applications.rollout_les
    • credit.applications.rollout_metrics
    • credit.applications.rollout_metrics_noisy_ic
    • credit.applications.rollout_metrics_noisy_model
    • credit.applications.rollout_realtime
    • credit.applications.rollout_realtime_gen2
    • credit.applications.rollout_samudra
    • credit.applications.rollout_to_netcdf
    • credit.applications.rollout_to_netcdf_gen2
    • credit.applications.rollout_wrf
    • credit.applications.scaler
    • credit.applications.train_gen1
    • credit.applications.train_gen2
    • credit.applications.train_les
    • credit.applications.train_wrf
    • credit.applications.train_wrf_multi
  • credit.attend
  • credit.boundary_padding
  • credit.cli
    • credit.cli.__main__
  • credit.conda_env
  • credit.credit_ptype
  • credit.data
  • credit.datasets
    • credit.datasets.count_channels
    • credit.datasets.gen_1
      • credit.datasets.gen_1.datamap
      • credit.datasets.gen_1.downscaling_dataset
      • credit.datasets.gen_1.era5_multistep
      • credit.datasets.gen_1.era5_multistep_batcher
      • credit.datasets.gen_1.era5_singlestep
      • credit.datasets.gen_1.les_singlestep
      • credit.datasets.gen_1.load_dataset_and_dataloader
      • credit.datasets.gen_1.om4_multistep_batcher
      • credit.datasets.gen_1.realtime_predict
      • credit.datasets.gen_1.sequential_multistep
      • credit.datasets.gen_1.wrf_singlestep
      • credit.datasets.gen_1.wrfmultistep
    • credit.datasets.gen_2
      • credit.datasets.gen_2.base_dataset
      • credit.datasets.gen_2.channel_utils
      • credit.datasets.gen_2.era5
      • credit.datasets.gen_2.gefs
      • credit.datasets.gen_2.gefs_download
      • credit.datasets.gen_2.gfs
      • credit.datasets.gen_2.gfs_download
      • credit.datasets.gen_2.goes
      • credit.datasets.gen_2.grid_utils
      • credit.datasets.gen_2.hrrr
      • credit.datasets.gen_2.hrrr_download
      • credit.datasets.gen_2.local
      • credit.datasets.gen_2.mrms
      • credit.datasets.gen_2.mrms_download
      • credit.datasets.gen_2.multi_source
      • credit.datasets.gen_2.tisr
  • 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.base
    • credit.losses.crps
    • credit.losses.gen_1
      • credit.losses.gen_1.almost_fair_crps
      • credit.losses.gen_1.covariance
      • credit.losses.gen_1.downscaling_loss
      • credit.losses.gen_1.kcrps
      • credit.losses.gen_1.les_loss
      • credit.losses.gen_1.logcosh
      • credit.losses.gen_1.power
      • credit.losses.gen_1.spectral
      • credit.losses.gen_1.weighted_loss
      • credit.losses.gen_1.xsigmoid
      • credit.losses.gen_1.xtanh
    • credit.losses.msle
  • credit.metadata
  • credit.metrics
    • credit.metrics.anomaly
    • credit.metrics.base
    • credit.metrics.common
    • credit.metrics.gen_1
      • credit.metrics.gen_1.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.models.wxformer.wxformer_next
  • credit.nwp
  • credit.ocean
    • credit.ocean.samudra_constants
    • credit.ocean.samudra_data
  • credit.output
  • credit.output_downscaling
  • credit.output_gen2
  • credit.parallel
    • credit.parallel.collectives
    • credit.parallel.domain
    • credit.parallel.fsdp2
    • credit.parallel.mesh
    • credit.parallel.tensor_parallel
  • credit.parser
  • credit.pbs
  • credit.physics_constants
  • credit.physics_core
  • credit.pol_lapdiff_filt
  • credit.postblock
    • credit.postblock.advect
    • credit.postblock.base
    • credit.postblock.conservation
    • credit.postblock.exp
    • credit.postblock.gen1
    • credit.postblock.geopotential
    • credit.postblock.hybrid_interp
    • credit.postblock.mslp
    • credit.postblock.pressure_interp
    • credit.postblock.reconstruct
    • credit.postblock.scaler
    • credit.postblock.square
    • credit.postblock.wet_mask_samudra
    • credit.postblock.wind_filter
  • credit.preblock
    • credit.preblock.advect
    • credit.preblock.base
    • credit.preblock.concat
    • credit.preblock.device
    • credit.preblock.fill_values
    • credit.preblock.hybrid_interp
    • credit.preblock.log
    • credit.preblock.norm
    • credit.preblock.regrid
    • credit.preblock.rename
    • credit.preblock.scaler
    • credit.preblock.sqrt
  • credit.registry
  • credit.regrid
  • credit.replay_buffer
  • credit.samplers
  • credit.scheduler
  • credit.seed
  • credit.skebs
  • credit.slurm
  • 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
• ensemble_eval
• rollout_samudra
• rollout_metrics
• train_wrf_multi
• graph_edges_knn
• rollout_realtime
• rollout_to_netcdf
• calc_global_solar
• rollout_downscaling
• benchmark_parallelism
• rollout_realtime_gen2
• rollout_to_netcdf_gen2
• rollout_metrics_noisy_ic
• rollout_metrics_noisy_model

[1]

Created with sphinx-autoapi