Add TAU_ML_microphysics: 9-input ML warm-rain autoconversion emulator - #11
Open
addisug wants to merge 1 commit into
Open
Add TAU_ML_microphysics: 9-input ML warm-rain autoconversion emulator#11addisug wants to merge 1 commit into
addisug wants to merge 1 commit into
Conversation
Documents and provides sources for a neural-network emulator of PUMAS's TAU bin-resolved warm-rain (cloud-rain autoconversion/accretion) microphysics, ported from the wkchuang/PUMAS fork (branch 1nn_tau_emulator, commit 8b76bac) onto the CCPP-ized pumas_cam-release_v1.39 baseline in cesm3_0_alpha09e (CAM cam6_4_186). Differs from the 'emulated' scheme already present upstream: 9 NN inputs (adds cloud/rain size-distribution shape parameters pgam/lamc/lamr/n0r) instead of 7, dual forward/inverse quantile interpolation instead of a single simplified version, and in-cloud/in-precip NN inputs with lcldm/precip_frac rescaling applied to the output tendencies afterward. No FTorch/PyTorch dependency -- pure Fortran + netCDF. Critical gotcha documented up front: micro_mg_warm_rain and this whole scheme only exist in the CAM7 physics driver, not CAM6. Using a CAM6 compset (e.g. F2000climo) fails at startup with "micro_pumas_cam_readnl:: ERROR reading namelist" since the Fortran namelist reader chokes on the unrecognized variable. Use a CAM7 compset (e.g. F2000dev) instead. Tested with a 1-day F2000dev/f09_f09_mg17 smoke test on Derecho (built with --run-unsupported, since this compset/resolution pairing isn't yet validated) -- NN weights load and the run completes cleanly.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Test plan