zhangrenchao commited on
Commit
81979e8
·
verified ·
1 Parent(s): 956cd31

Add English model card

Browse files
Files changed (1) hide show
  1. README.md +92 -0
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - OneScience
7
+ - Earth Science
8
+ - Coupled Climate
9
+ - Ocean Emulator
10
+ - Atmosphere Emulator
11
+ - Long-Term Climate
12
+ frameworks: PyTorch
13
+ ---
14
+
15
+ <p align="center"><strong><span style="font-size: 30px;">SamudrACE</span></strong></p>
16
+
17
+ # Model Introduction
18
+
19
+ SamudrACE combines three-dimensional atmosphere and ocean emulators in a fast coupled climate model. SST, sea ice, and surface fluxes are exchanged in physical state space for stable long-term simulation.
20
+
21
+ Paper: SamudrACE: Fast and Accurate Coupled Climate Modeling With 3D Ocean and Atmosphere Emulators
22
+ https://arxiv.org/abs/2509.12490
23
+
24
+ # Model Description
25
+
26
+ The model was proposed by teams from Ai2, New York University, Princeton University, NOAA/GFDL, Columbia University, and collaborators. It was trained with a 200-year GFDL CM4 pre-industrial control simulation. ACE2-style atmosphere and SamudraI-style ocean components exchange physical states for coupled climate simulation and drift diagnosis.
27
+
28
+ # Use Cases
29
+
30
+ | Use Case | Description |
31
+ | :---: | :--- |
32
+ | Coupled simulation | Drive one ocean step with 20 atmosphere steps. |
33
+ | Climate drift diagnosis | Assess atmosphere, heat, and salinity proxies. |
34
+ | Global climate emulation | Preserve a logical one-degree multilevel state. |
35
+ | ModelScope/OneCode execution | Validate structured data, training, inference, climate metrics, and visualization. |
36
+ | Multi-GPU training | Start multi-process training through `torchrun`. |
37
+
38
+ # Usage Instructions
39
+
40
+ ```bash
41
+ hf download OneScience-Group/SamudrACE --local-dir ./SamudrACE
42
+ cd SamudrACE
43
+ ```
44
+
45
+ ### Environment Dependencies
46
+
47
+ **Hardware Requirements**
48
+
49
+ - A GPU or DCU is recommended.
50
+ - A CPU can be used for connectivity validation with the default small-sample configuration.
51
+ - DCU users should install DTK 25.04.2 or a compatible OneScience-recommended version first.
52
+
53
+ **DCU Environment**
54
+
55
+ ```bash
56
+ # Activate DTK and Conda first
57
+ conda create -n onescience311 python=3.11 -y
58
+ conda activate onescience311
59
+ pip install onescience[earth-dcu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
60
+ ```
61
+
62
+ **GPU Environment**
63
+
64
+ ```bash
65
+ # Activate Conda first
66
+ conda create -n onescience311 python=3.11 -y libstdcxx-ng=12 libgcc-ng=12 gcc_linux-64=12 gxx_linux-64=12
67
+ conda activate onescience311
68
+ pip install onescience[earth-gpu] -i http://mirrors.onescience.ai:3141/pypi/simple/ --trusted-host mirrors.onescience.ai
69
+ ```
70
+
71
+ ```bash
72
+ python scripts/fake_data.py
73
+ python scripts/train.py
74
+ torchrun --nproc_per_node=2 --nnodes=1 --master_addr="localhost" --master_port=29500 scripts/train.py
75
+ python scripts/inference.py
76
+ python scripts/result.py
77
+ ```
78
+
79
+ Synthetic tiles retain all 126 fields enumerated by the paper and the 20-to-1 coupling ratio. Training performs coupled backpropagation through 20 atmosphere steps and one ocean step, and both single-process and two-process DDP runs have been verified. Inference restores the checkpoint and preserves 46 atmosphere channels, 80 ocean channels, original tile coordinates, and the incomplete-global marker. Training results are saved to `result/checkpoints/samudrace.pt`; inference and drift diagnostics are saved under `result/output/` and `result/evaluation/`.
80
+
81
+ # Official OneScience Information
82
+
83
+ | Platform | OneScience Main Repository | Skills Repository |
84
+ |---|---|---|
85
+ | Gitee | https://gitee.com/onescience-ai/onescience | https://gitee.com/onescience-ai/oneskills |
86
+ | GitHub | https://github.com/onescience-ai/OneScience | https://github.com/onescience-ai/oneskills |
87
+
88
+ # Citation and License
89
+
90
+ This repository is an independent engineering reproduction of the public SamudrACE specifications.
91
+
92
+ The original paper, official code, weights, and GFDL CM4 data remain subject to their respective licenses and terms.