Skip to content

Adding EKI race examples#531

Merged
odunbar merged 43 commits intomainfrom
rg/eki-race-examples
Jan 31, 2026
Merged

Adding EKI race examples#531
odunbar merged 43 commits intomainfrom
rg/eki-race-examples

Conversation

@rgjini
Copy link
Copy Markdown
Collaborator

@rgjini rgjini commented Nov 24, 2025 •
edited by odunbar
Loading

Purpose

Closes #530
Closes #541

Content

list of proposed tasks

  • Add Lorenz63 example
  • Add Lorenz96 constant forcing example
  • Add Lorenz96 spatially dependent forcing example
  • Add Lorenz96 Spatially dependent forcing example with neural network approximation
  • Saves many precomputed quantities (e.g. the estimated covariances) from initial run for time savings on future runs
  • Saves output results for ingestion by plotting scripts.
  • Plotting scripts have been written to display the distribution of iterations / model runs over the race

Misc

  • Bugfixed the (actually no-longer used here) avg_rmse computation

@rgjini rgjini changed the title Adding empty sciprt for the Lorenz63 example Adding EKI race examples Nov 24, 2025
rgjini and others added 28 commits November 24, 2025 17:16
@rgjini
Adding functions to run Lorenz 63 and 96 models, with associated stat…
ca2b759 
…istics calculation functions
@odunbar
Proposed framework for forcing
2af7dcc
@rgjini
adding some updated code for the examples (still in progress)
173db22
@rgjini
slowly adding different experiment requirements and finishing the pro…
972a922 
…blem setup: everything runs correctly so far
@rgjini
adding observational covariance matrix calculation
2ff8adf
@rgjini
spatial forcing race example finished. RMSE(mean(G(u))) currently not…
9fab6f7 
… being calculate properly
@rgjini
testing scalar case
738b422
@rgjini
both L96 with constant and spatial forcing working
2716ecb
@rgjini
Adding empty sciprt for the Lorenz63 example
8c90acb
@rgjini
Adding functions to run Lorenz 63 and 96 models, with associated stat…
963536b 
…istics calculation functions
@odunbar @rgjini
Proposed framework for forcing
afdffe4
@rgjini
adding some updated code for the examples (still in progress)
77ca1bc
@rgjini
slowly adding different experiment requirements and finishing the pro…
02aaaf1 
…blem setup: everything runs correctly so far
@rgjini
adding observational covariance matrix calculation
94ce097
@rgjini
spatial forcing race example finished. RMSE(mean(G(u))) currently not…
0f14c87 
… being calculate properly
@rgjini
testing scalar case
99dd4db
@rgjini
both L96 with constant and spatial forcing working
30700f9
@rgjini
Merge branch 'rg/eki-race-examples' of github.com:CliMA/EnsembleKalma…
45c1f6f 
…nProcesses.jl into rg/eki-race-examples
@rgjini
Fixing RMSE calculation of the mean of the output ensemble (mean(G)) …
018ac47 
…in script
@rgjini
working on neural network example, so far the code is running extreme…
839a872 
…ly slow
@rgjini
Lorenz63 example finished
9a0ba2a
@odunbar
formatting
abc213a
@odunbar
only evaluate NN once per sim, not once per RK stage, also add FT32 c…
e2ffb7e 
…hecks
@odunbar
save/load precomputable quantities (e.g. observational covariances)
c0d7e13
@odunbar
format
298ec50
@odunbar
save results in output/
5335ce5
@odunbar
add file saving for l96 and l63
76723b3
@odunbar
format
9d5ac9f
github-actions[bot]
github-actions Bot reviewed Jan 15, 2026
View reviewed changes
Copy link
Copy Markdown
Contributor

@github-actions github-actions Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Contributor License Agreement required

The following contributors must sign the CLA before this PR can be merged:

Please visit https://ecodesign.clima.caltech.edu/cla/ to review and upload the signed CLA.

Once completed, re-run the checks on this PR.

github-actions[bot]
github-actions Bot reviewed Jan 16, 2026
View reviewed changes
Copy link
Copy Markdown
Contributor

@github-actions github-actions Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Contributor License Agreement required

The following contributors must sign the CLA before this PR can be merged:

Please visit https://ecodesign.clima.caltech.edu/cla/ to review and sign the CLA.

How to sign: Authenticate with GitHub then click the "I agree" button.

Once completed, re-run the checks on this PR.

github-actions[bot]
github-actions Bot reviewed Jan 16, 2026
View reviewed changes
Copy link
Copy Markdown
Contributor

@github-actions github-actions Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Contributor License Agreement required

The following contributors must sign the CLA before this PR can be merged:

Please visit https://ecodesign.clima.caltech.edu/cla/ to review and sign the CLA.

How to sign: Authenticate with GitHub then click the "I agree" button.

Once completed, re-run the checks on this PR.

github-actions[bot]
github-actions Bot reviewed Jan 16, 2026
View reviewed changes
Copy link
Copy Markdown
Contributor

@github-actions github-actions Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Contributor License Agreement required

The following contributors must sign the CLA before this PR can be merged:

Please visit https://ecodesign.clima.caltech.edu/cla/ to review and sign the CLA.

How to sign: Authenticate with GitHub then click the "I agree" button.

Once completed, re-run the checks on this PR.

odunbar
odunbar approved these changes Jan 30, 2026
View reviewed changes
Copy link
Copy Markdown
Member

@odunbar odunbar left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

I was able to run several experiments, and the following are plots from a couple of the runs

l63 - one experiment
ImageImage

l96 - vec - multiple experiments
ImageImage

@odunbar odunbar merged commit aed11ee into main Jan 31, 2026
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@github-actions github-actions[bot] github-actions[bot] left review comments

@odunbar odunbar odunbar approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

avg_rmse, is not computing the average RMSE Implement EKI race examples

2 participants

@rgjini @odunbar