Add checkpoint and update README.md (#1)

- Add checkpoint and draft README.md (fb025b7c096f73c44641ac8b59dbaa6225901bc9)
- Add initial condition and forcing data for 1xCO2 and 3xCO2 (6757374b7e354af142424c5b6d2be1da90a3008d)
- Fix typo per Oli's review (f95fb80bbaa093f1bed5d68fbc21f1eb31e93da4)

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.nc filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,29 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+---
+
+<img src="ACE-logo.png" alt="Logo for the ACE Project" style="width: auto; height: 50px;">
+
+# ACE2-SOM
+
+Ai2 Climate Emulator (ACE) is a family of models designed to simulate atmospheric variability from the time scale of days to centuries.
+
+**Disclaimer: ACE models are research tools and should not be used for operational climate predictions.**
+
+ACE2-SOM is the first model we have trained to emulate the sensitivity of climate to substantial increases in carbon dioxide.  It consists of ACE2 coupled to a highly simplified physics-based ocean model, called a slab ocean model (SOM), and is trained on equilibrium-climate output from GFDL's SHiELD model coupled to an identical slab ocean with carbon dioxide concentrations of the present-day, double the present-day, and quadruple the present-day (1xCO2, 2xCO2, and 4xCO2).
+
+Quick links:
+- 📃 [Paper (pre-print coming soon)]()
+- 💻 [Code](https://github.com/ai2cm/ace)
+- 💬 [Docs](https://ai2-climate-emulator.readthedocs.io/en/stable/)
+- 📂 [All Models](https://huggingface.co/collections/allenai/ace-67327d822f0f0d8e0e5e6ca4)
+
+Briefly, the strengths of ACE2-SOM are:
+- accurate emulation of equilibrium climates with CO2 concentrations ranging from 1xCO2 to 4xCO2, including changes in the spatial patterns of temperature and precipitation, and changes in precipitation extremes between climates
+- realistic emulation of surface and lower-to-middle atmosphere fields with gradually changing CO2 concentration between 1xCO2 and 4xCO2.
+
+Some known weaknesses are:
+- Sea-ice coverage, ocean heat transport, and ocean mixed layer depth are prescribed based on a present-day climatologies and therefore do not respond to changes in CO2.  Presribed sea-ice in particular means projections with ACE2-SOM are missing an important feedback mechanism known to amplify warming in the polar regions.
+- Abrupt regime shifts in stratospheric temperature and moisture can occur in inference runs with changing CO2 concentration.
+- Adjustment to an abrupt change in CO2 concentration occurs too quickly, violating energy conservation.
+- Sensitivities of the surface and top of atmosphere radiative fluxes to changes in CO2 with the atmospheric state held fixed are not always realistic.

ace2_som_ckpt.tar

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b191174e45ac16d86f3a5ccfa99b8952866ec6482b19e08edbd0f05114f5f9ab
+size 1823641466

forcing_data/1xCO2/forcing_2032.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5617718ced22f19f7a37a82f33c51cedd7b4a7b56f65766333cc8f3e33c8cd6
+size 945235410

forcing_data/1xCO2/forcing_2033.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c4890f3abb099b85ffdab952a7bf3d70507894d0f63c6f23e5d0c5b0c4fb232
+size 942674557

forcing_data/1xCO2/forcing_2034.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c17aec8be4941cbbe1479731e18efd9a4ed47906b274ff25596109603e9733bb
+size 942687017

forcing_data/1xCO2/forcing_2035.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e6dfd8bc61d45dcea8e82c69b374546d0b6df7fb536453ce024c3a56c47e4fe1
+size 942672153

forcing_data/3xCO2/forcing_2032.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:81aafd1b94c738d7593840a78d477ebcc16c73636784c713c237136bc557d3fe
+size 945257461

forcing_data/3xCO2/forcing_2033.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7fbc16f828522b654e1a15e5cd81d6f8f594254b288b4a97c27cfa01ca8ccdfa
+size 942659253

forcing_data/3xCO2/forcing_2034.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8871101e46a22b55c99163700cb99831df96b70a9f8a3a61ca3887a35cc294f7
+size 942658976

forcing_data/3xCO2/forcing_2035.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:202ef54198f0e380d06894ccf5c8147dfe9f24f62dab7133268c457a65ef54ec
+size 942675350

inference_config.yaml

@@ -0,0 +1,23 @@
+experiment_dir: /output_directory
+n_forward_steps: 400 
+forward_steps_in_memory: 50
+checkpoint_path: /ace2_som_ckpt.tar
+logging:
+  log_to_screen: true
+  log_to_wandb: false
+  log_to_file: true
+  project: ace
+initial_condition:
+  path: /initial_conditions/1xCO2/ic.nc
+  start_indices:
+    times:
+      - "2032-01-01T00:00:00"
+forcing_loader:
+  dataset:
+    data_path: /forcing_data/1xCO2
+    n_repeats: 1  # Use this to extend the four years of forcing data to the desired length
+  num_data_workers: 4
+data_writer:
+  save_prediction_files: true
+  save_monthly_files: false
+  names: ["TMP2m", "VGRD10m", "PRATEsfc"]

initial_conditions/1xCO2/ic.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f481537cc80e4dd4c41f4e629563b2f62b4c3b1d9ede3db67f429ed3669fab13
+size 9959599

initial_conditions/3xCO2/ic.nc

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:449b50d6b97d360d70bb52f90ee604e933a84d1168cab7658c2fba20774ef751
+size 9959599