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 100 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 100 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
increase simulation run time
57c3ac5
@xkykai
wall restoration to maintain strratification
d5a320e
@xkykai
baseclosure doublegyre with wallrestoration
0cd618a
@xkykai
change filenames
4abf7e0
@xkykai
using wider zone for NN closure
7cf7090
@xkykai
fix variable error
70dbe2f
@xkykai
update NN configuration
fd57bf8
@xkykai
NDE double gyre script for seasonal forcing and wall restoration
b582447
@xkykai
run NNclosure with Ri nof BBLkappazonelast55
6235bc4
@xkykai
updated NN model with no Ri
3609c21
@xkykai
run double gyre with NNclosure with no Ri input kappazonelast55
823e8c7
@xkykai
add fluxes calculations and zonal average
893b7bc
@xkykai
Merge branch 'main' into xk/embed-nn
ea77669
@xkykai
Merge branch 'main' into xk/embed-nn
3428f5d
@xkykai
using centered second order in z instead of WENO
4a5a3e1
@xkykai
remove background vertical scalar diffusivity
0f6bf66
@xkykai
diffusivity fields indexing for base closure
8385229
@xkykai
change temperature restoration to 30days
45eef0a
@xkykai
change vertical advection scheme to WENO5
e29bfd9
@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
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