Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Implement neural network parameterization of salty turbulent mixing in the upper ocean #3819

Draft
wants to merge 119 commits into
base: main
Choose a base branch
from
Draft

Implement neural network parameterization of salty turbulent mixing in the upper ocean #3819

wants to merge 119 commits into from

Conversation

xkykai
Copy link
Collaborator

@xkykai xkykai commented Oct 3, 2024

This PR aims to incorporate trained neural networks that represent the entrainment process due to turbulent convection in the upper ocean into Oceananigans.jl. This parameterization is designed to work using the TEOS-10 nonlinear equation of state, hence it works only in typical oceanographic conditions on Earth.

xkykai added 30 commits June 24, 2024 05:08
@xkykai
working version on CPU
13cd632
@xkykai
a working model for GPU!
bb70e5c
@xkykai
it actually works on the gpu now!
e845dd5
@xkykai
build scaling from tuple values
b6552ab
@xkykai
Update XinKaiLocalVerticalDiffusivity with 2 Pr values
c9a9c65
@xkykai
fix function construct_scaling to construct_zeromeanunitvariance_scaling
9a07b82
@xkykai
update using KernelParameters
74c8b61
@xkykai
fix nn closure bug
6f97c3f
@xkykai
test script for nn closure
65f15e4
@xkykai
fix N2 average, tracer diffusivity expression for local diffusivity c…
cb601ce 
…losure with 2 Pr
@xkykai
2Pr version of local physical closure
dbe5759
@xkykai
nonlocal physical closure of vertical diffusivity
e9ac435
@xkykai
2Pr nonlocal physical closure
5e548da
@xkykai
working GPU version of NN closure with scaling and correction
772c235
@xkykai
validation script for oceananigans NN implementation
9c6873f
@xkykai
close file and record video of validation
f6d508a
@xkykai
Update Project.toml with new package dependencies
1faa5a6
@xkykai
Update NN closure model to use a larger neural network
930cc7e
@xkykai
Remove unused import of StaticArrays
fd95fc6
@xkykai
Add ComponentArrays dependency
72aebc1
@xkykai
add total_size and KernelParameters dependency
c2d336e
@xkykai
Coarsen LES data for NN closure model
d4a1156
@xkykai
rename file and compare LES with NN closure
ae7b563
@xkykai
run LES for hald sinusoid cooling
4d118c2
@xkykai
run 3D simulation with limited extent
fc081cc
@xkykai
add sponge at bottom
a78109e
@xkykai
fix metres to meters
ca8d78c
@xkykai
fix temperature flux
78e7a67
@xkykai
Calculate average velocities and tracers in 3D model LES simulation
eaf2183
@xkykai
reduce size of model
7e144ff
xkykai added 30 commits October 30, 2024 16:53
@xkykai
update neural network to new weights
0fba127
@xkykai
fix file name change
73afa68
@xkykai
run double gyre with centered second order and wall restoration
1fab19f
@xkykai
fix dynamic function invocation that doesn't affect the baseclosure s…
49d901f 
…cript somehow
@xkykai
add y resolution variable
534d21d
@xkykai
run NN closure recording xz slices
2551c38
@xkykai
recording and plotting xz yz slices and fluxes
ff272c4
@xkykai
change default z advection scheme to weno 5
93bf0bc
@xkykai
run with linear ramp seasonal forcing
d36677f
@xkykai
fix T_seasonal
ce68cc7
@xkykai
fix temperature restoration
e1603ca
@xkykai
implement different NNclosure with Ri zone
a3ab871
@xkykai
run double gyre on updated closure
51acb9d
@xkykai
new NN closure witth Rifirstzone
33f2ea8
@xkykai
fix NN closure
36cecb6
@xkykai
fix NN_closure
946d710
@xkykai
updated base closure with new calibration
0c55c44
@xkykai
run base closure and CATKE with mode waters
dada2c5
@xkykai
fix advection scheme and file name
1d0ef1f
@xkykai
new NN closure including Ri
4898be4
@xkykai
actual new closure including Ri
3e84aaf
@xkykai
run double gyre with new NN closure
88f7fa4
@xkykai
run CATKE for 10800days
030c890
@xkykai
Merge branch 'main' into xk/embed-nn
f4f7383
@xkykai
flush mode water after equilibration
35dbad3
@xkykai
fix filename
88e3cf1
@xkykai
Revert "fix filename"
5b34027 
This reverts commit 88e3cf1.
@xkykai
Revert "flush mode water after equilibration"
8e458e0 
This reverts commit 35dbad3.
@xkykai
Revert "Merge branch 'main' into xk/embed-nn"
34f7437 
This reverts commit f4f7383, reversing
changes made to 030c890.
@xkykai
fix script changes for modewater after equilibration
ebf6639
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@glwagner glwagner glwagner left review comments

@navidcy navidcy navidcy left review comments

At least 1 approving review is required to merge this pull request.

Assignees

@xkykai xkykai

Labels
experimental feature 🧪 Because danger is the spice of life help wanted 🦮 plz halp (guide dog provided) parameterizations Fudge factors and hand-waving turbulence closures 🎐
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@xkykai @navidcy @glwagner