LICENSE
README.md
pyproject.toml
./xmetai/__init__.py
./xmetai/registries.py
./xmetai/cli/__init__.py
./xmetai/cli/train.py
./xmetai/cli/train_lightning.py
./xmetai/config/__init__.py
./xmetai/config/config.py
./xmetai/config/instantiate.py
./xmetai/config/lazy.py
./xmetai/core/__init__.py
./xmetai/core/checkpoint.py
./xmetai/core/defaults.py
./xmetai/core/evaluator.py
./xmetai/core/hooks.py
./xmetai/core/launch.py
./xmetai/core/lion.py
./xmetai/core/muon.py
./xmetai/core/optimizer.py
./xmetai/core/scheduler.py
./xmetai/core/trainer.py
./xmetai/data/__init__.py
./xmetai/data/loader.py
./xmetai/data/sampler.py
./xmetai/data/weather/__init__.py
./xmetai/data/weather/data_util.py
./xmetai/data/weather/era5.py
./xmetai/data/weather/graphcast_dataset.py
./xmetai/data/weather/graphcast_grid.py
./xmetai/data/weather/meteonet_dataset.py
./xmetai/data/weather/multi_zarr_dataset.py
./xmetai/data/weather/radar_dataset.py
./xmetai/layers/__init__.py
./xmetai/layers/activation.py
./xmetai/layers/afnonet_blocks.py
./xmetai/layers/async_multimodal.py
./xmetai/layers/attention.py
./xmetai/layers/darknet_unet.py
./xmetai/layers/drop.py
./xmetai/layers/earth_conv.py
./xmetai/layers/earth_spectral_conv.py
./xmetai/layers/embed.py
./xmetai/layers/encoding.py
./xmetai/layers/gated_deltanet2.py
./xmetai/layers/graphcast_blocks.py
./xmetai/layers/helpers.py
./xmetai/layers/lora.py
./xmetai/layers/mamba_vision.py
./xmetai/layers/mlp.py
./xmetai/layers/natten.py
./xmetai/layers/natten3d_unet.py
./xmetai/layers/norm.py
./xmetai/layers/puyun_blocks.py
./xmetai/layers/rotary_attention.py
./xmetai/layers/swin.py
./xmetai/layers/tianquan_blocks.py
./xmetai/layers/weight_init.py
./xmetai/layers/wrappers.py
./xmetai/lightning/__init__.py
./xmetai/lightning/lightning_model_wrapper.py
./xmetai/loss/__init__.py
./xmetai/loss/charbonnier_loss.py
./xmetai/loss/crps_loss.py
./xmetai/loss/dynamics_loss.py
./xmetai/loss/kl_loss.py
./xmetai/loss/pinn_loss.py
./xmetai/loss/precip_loss.py
./xmetai/metrics/__init__.py
./xmetai/metrics/ips.py
./xmetai/metrics/mse.py
./xmetai/metrics/ps.py
./xmetai/metrics/radar_echo.py
./xmetai/metrics/tcc.py
./xmetai/metrics/threshold.py
./xmetai/models/__init__.py
./xmetai/models/afnonet.py
./xmetai/models/base_predictor.py
./xmetai/models/diffusion_forecast.py
./xmetai/models/ema.py
./xmetai/models/graphcast.py
./xmetai/models/group_vae.py
./xmetai/models/meta.py
./xmetai/models/onnx_infer.py
./xmetai/models/pred_head.py
./xmetai/models/puyun.py
./xmetai/models/tianquan.py
./xmetai/models/utransformer.py
./xmetai/models/vit.py
./xmetai/models/vit_ens.py
./xmetai/recipes/__init__.py
./xmetai/recipes/dataloader.py
./xmetai/recipes/optim.py
./xmetai/recipes/schedule.py
./xmetai/recipes/train.py
./xmetai/utils/__init__.py
./xmetai/utils/collect_env.py
./xmetai/utils/colormaps.py
./xmetai/utils/comm.py
./xmetai/utils/env.py
./xmetai/utils/events.py
./xmetai/utils/file_io.py
./xmetai/utils/logger.py
./xmetai/utils/memory.py
./xmetai/utils/radar.py
./xmetai/utils/reanalyse.py
./xmetai/utils/registry.py
./xmetai/utils/timer.py
./xmetai/utils/assets/china_100000_full_wgs84.geojson
xmetai/__init__.py
xmetai/registries.py
xmetai/cli/__init__.py
xmetai/cli/train.py
xmetai/cli/train_lightning.py
xmetai/config/__init__.py
xmetai/config/config.py
xmetai/config/instantiate.py
xmetai/config/lazy.py
xmetai/core/__init__.py
xmetai/core/checkpoint.py
xmetai/core/defaults.py
xmetai/core/evaluator.py
xmetai/core/hooks.py
xmetai/core/launch.py
xmetai/core/lion.py
xmetai/core/muon.py
xmetai/core/optimizer.py
xmetai/core/scheduler.py
xmetai/core/trainer.py
xmetai/data/__init__.py
xmetai/data/loader.py
xmetai/data/sampler.py
xmetai/data/weather/__init__.py
xmetai/data/weather/data_util.py
xmetai/data/weather/era5.py
xmetai/data/weather/graphcast_dataset.py
xmetai/data/weather/graphcast_grid.py
xmetai/data/weather/meteonet_dataset.py
xmetai/data/weather/multi_zarr_dataset.py
xmetai/data/weather/radar_dataset.py
xmetai/layers/__init__.py
xmetai/layers/activation.py
xmetai/layers/afnonet_blocks.py
xmetai/layers/async_multimodal.py
xmetai/layers/attention.py
xmetai/layers/darknet_unet.py
xmetai/layers/drop.py
xmetai/layers/earth_conv.py
xmetai/layers/earth_spectral_conv.py
xmetai/layers/embed.py
xmetai/layers/encoding.py
xmetai/layers/gated_deltanet2.py
xmetai/layers/graphcast_blocks.py
xmetai/layers/helpers.py
xmetai/layers/lora.py
xmetai/layers/mamba_vision.py
xmetai/layers/mlp.py
xmetai/layers/natten.py
xmetai/layers/natten3d_unet.py
xmetai/layers/norm.py
xmetai/layers/puyun_blocks.py
xmetai/layers/rotary_attention.py
xmetai/layers/swin.py
xmetai/layers/tianquan_blocks.py
xmetai/layers/weight_init.py
xmetai/layers/wrappers.py
xmetai/lightning/__init__.py
xmetai/lightning/lightning_model_wrapper.py
xmetai/loss/__init__.py
xmetai/loss/charbonnier_loss.py
xmetai/loss/crps_loss.py
xmetai/loss/dynamics_loss.py
xmetai/loss/kl_loss.py
xmetai/loss/pinn_loss.py
xmetai/loss/precip_loss.py
xmetai/metrics/__init__.py
xmetai/metrics/ips.py
xmetai/metrics/mse.py
xmetai/metrics/ps.py
xmetai/metrics/radar_echo.py
xmetai/metrics/tcc.py
xmetai/metrics/threshold.py
xmetai/models/__init__.py
xmetai/models/afnonet.py
xmetai/models/base_predictor.py
xmetai/models/diffusion_forecast.py
xmetai/models/ema.py
xmetai/models/graphcast.py
xmetai/models/group_vae.py
xmetai/models/meta.py
xmetai/models/onnx_infer.py
xmetai/models/pred_head.py
xmetai/models/puyun.py
xmetai/models/tianquan.py
xmetai/models/utransformer.py
xmetai/models/vit.py
xmetai/models/vit_ens.py
xmetai/recipes/__init__.py
xmetai/recipes/dataloader.py
xmetai/recipes/optim.py
xmetai/recipes/schedule.py
xmetai/recipes/train.py
xmetai/utils/__init__.py
xmetai/utils/collect_env.py
xmetai/utils/colormaps.py
xmetai/utils/comm.py
xmetai/utils/env.py
xmetai/utils/events.py
xmetai/utils/file_io.py
xmetai/utils/logger.py
xmetai/utils/memory.py
xmetai/utils/radar.py
xmetai/utils/reanalyse.py
xmetai/utils/registry.py
xmetai/utils/timer.py
xmetai/utils/assets/china_100000_full_wgs84.geojson
xmetai_core.egg-info/PKG-INFO
xmetai_core.egg-info/SOURCES.txt
xmetai_core.egg-info/dependency_links.txt
xmetai_core.egg-info/entry_points.txt
xmetai_core.egg-info/requires.txt
xmetai_core.egg-info/top_level.txt