| { | |
| "model_name": "AI2 Climate Emulator (ACE)", | |
| "model_type": "ace", | |
| "architectures": [ | |
| "ACEModel" | |
| ], | |
| "framework": "PyTorch", | |
| "domain": "atmosphere-and-climate", | |
| "task": "global-climate-emulation", | |
| "implementation": { | |
| "entry_point": "model/ace.py", | |
| "scope": "40-channel to 44-channel SFNO emulator with autoregressive rollout" | |
| }, | |
| "architecture": { | |
| "family": "spherical Fourier neural operator", | |
| "grid": "Legendre-Gauss", | |
| "grid_shape": [ | |
| 180, | |
| 360 | |
| ], | |
| "input_channels": 40, | |
| "output_channels": 44, | |
| "prognostic_channels": 34, | |
| "forcing_channels": 6, | |
| "embedding_size": 256, | |
| "layers": 8, | |
| "filter_type": "linear", | |
| "operator_type": "dhconv", | |
| "spectral_layers": 3, | |
| "mlp_ratio": 2.0 | |
| }, | |
| "data": { | |
| "dataset": "FV3GFS", | |
| "time_step_hours": 6, | |
| "training_members": 10, | |
| "validation_members": 1, | |
| "protocol": "fv3gfs_40_to_44" | |
| }, | |
| "configuration_sources": [ | |
| "conf/config.yaml", | |
| "model/ace.py", | |
| "model/sfno.py", | |
| "model/variables.py" | |
| ] | |
| } | |