| { |
| "model_name": "GenCast", |
| "model_type": "gencast", |
| "architectures": [ |
| "GenCast" |
| ], |
| "framework": "JAX with Haiku", |
| "domain": "atmosphere", |
| "task": "probabilistic-global-weather-forecasting", |
| "implementation": { |
| "entry_point": "model/gencast.py", |
| "scope": "official-equivalent model API with a configurable mini random-initialization profile" |
| }, |
| "architecture": { |
| "family": "graph-based denoising diffusion model", |
| "mesh_refinement_level": 4, |
| "mesh_nodes": 2562, |
| "latent_size": 512, |
| "mlp_hidden_layers": 1, |
| "sparse_transformer_layers": 16, |
| "attention_heads": 4, |
| "feed_forward_size": 2048, |
| "attention_k_hop": 16, |
| "attention_type": "triblockdiag_mha" |
| }, |
| "data": { |
| "dataset": "ERA5", |
| "input_duration_hours": 24, |
| "forecast_step_hours": 12, |
| "target_channels": 84, |
| "pressure_levels_hpa": [ |
| 50, |
| 100, |
| 150, |
| 200, |
| 250, |
| 300, |
| 400, |
| 500, |
| 600, |
| 700, |
| 850, |
| 925, |
| 1000 |
| ], |
| "pressure_level_variables": [ |
| "geopotential", |
| "specific_humidity", |
| "temperature", |
| "u_component_of_wind", |
| "v_component_of_wind", |
| "vertical_velocity" |
| ], |
| "surface_target_variables": [ |
| "2m_temperature", |
| "10m_u_component_of_wind", |
| "10m_v_component_of_wind", |
| "mean_sea_level_pressure", |
| "sea_surface_temperature", |
| "total_precipitation_12hr" |
| ] |
| }, |
| "sampling": { |
| "noise_levels": 20, |
| "minimum_noise_level": 0.03, |
| "maximum_noise_level": 80.0, |
| "rho": 7.0 |
| }, |
| "configuration_sources": [ |
| "conf/config.yaml", |
| "model/gencast.py", |
| "model/graphcast/gencast.py", |
| "model/data_loader.py" |
| ] |
| } |
|
|
