Started by an SCM change863655285cd4ce9d65cec1b44cf2e6e578f4d08fe73655285cd4ce9d65cec1b44cf2e6e578f4d08fe7origin/master3655285cd4ce9d65cec1b44cf2e6e578f4d08fe73655285cd4ce9d65cec1b44cf2e6e578f4d08fe7origin/master88d63a0f595d93dccafcc9e7917c00fe97fd034a35d63a0f595d93dccafcc9e7917c00fe97fd034a35origin/grid_fixesd63a0f595d93dccafcc9e7917c00fe97fd034a35d63a0f595d93dccafcc9e7917c00fe97fd034a35origin/grid_fixes79c5c01fc5da140dc13df06c2919c4f4f5a0318b5fc5c01fc5da140dc13df06c2919c4f4f5a0318b5forigin/changes_to_not_breakc5c01fc5da140dc13df06c2919c4f4f5a0318b5fc5c01fc5da140dc13df06c2919c4f4f5a0318b5forigin/changes_to_not_break701344f8a40d6116af1b567307eb389b4be1009fc41344f8a40d6116af1b567307eb389b4be1009fc4refs/remotes/origin/changes_to_not_break1344f8a40d6116af1b567307eb389b4be1009fc41344f8a40d6116af1b567307eb389b4be1009fc4refs/remotes/origin/changes_to_not_break851d226e398bae00fe10255274a70f3451b5cc7d9e1d226e398bae00fe10255274a70f3451b5cc7d9eorigin/wangara_updates1d226e398bae00fe10255274a70f3451b5cc7d9e1d226e398bae00fe10255274a70f3451b5cc7d9eorigin/wangara_updates5087a1e7ac8ce5005013a526945f3d000f423e0a25b7a1e7ac8ce5005013a526945f3d000f423e0a25brefs/remotes/origin/master7a1e7ac8ce5005013a526945f3d000f423e0a25b7a1e7ac8ce5005013a526945f3d000f423e0a25brefs/remotes/origin/master7a1e7ac8ce5005013a526945f3d000f423e0a25b7a1e7ac8ce5005013a526945f3d000f423e0a25brefs/remotes/origin/masterhttps://github.com/larson-group/sam_clubb.gitfalse#50815986101824718sam_clubb_nvhpc_build #508508false5087941SUCCESS1788425106596http://carson.math.uwm.edu/jenkins/view/all/job/sam_clubb_nvhpc_build/508/stats_netcdf.F90faeef517653ebf7aeab4fc52be8f5019101dd0621788372901000http://carson.math.uwm.edu/jenkins/user/noreplynoreplynoreply@github.comClubb JAX (#1351) * Testing new porting shortcut by creating a jax version of advance_xm_wpxp.py - it wen't well. Also updating the porting shortcut with more tips. * Small update to porting shortcut * Iterative improvements to porting shortcut * Using porting script to port advance_xp2_xpyp in one go - still some issues that seems to require rerunning to fix issues, which is sad but maybe the best we can do with such a simple method * Little bug fixes * Updates to make the ports create better jax in the sense that it is more jit-ready * Adding a more general plan to also use CLUBB-JAX and using it on advance_xm_wpxp * Cleanup * More properly jax-izing the current jax ports * More formatting of ported jax code * More formatting of ported jax code * Updates in pursuit of full jit-compatibility * Adding jit compile test * Fully jitting advance_xp2_xpyp * Copying parameter_indices from CLUBB-JAX but subtracting 1 to avoid -1 in code * Little update to port shortcut * Jitting and formatting CLUBB-JAX * Mostly formatting * Updating jax code * Improving stats outputting * Compile mono flux range loops with JAX * Improvement to jax mixing length port * Organizing stats code * Making jax work on GPU * Improvements to prescribe forcings code * Mainly prescribe forcings formatting work * Small improvements, getting rid of unneccesary weird jit statement * Radiation porting completed. Bugsrad and silhs radiation were intentionally left out for now. * Removing stale code and adding a proposal.md file detailing a potential future feature to make the python_api work with current jax code * Proper port of model flags and parameters and constants and such - those there were all kinda put into one file before * new jenkinsfile added * Tiny cleanup and adding back some stats outputs for microphysics that we need for some reason * limited to only one process. Find out built time * made jenkins jax tests faster by running them not so extensively * Mainly working on README, also some other tiny things: changing default test stats list to standard instead of all_stats, and renaming run_jax.sh to run_jax_wrapper.sh --------- Co-authored-by: Daniel Ebel <debel@uwm.edu> 2026-09-02 13:15:01 -0500faeef517653ebf7aeab4fc52be8f5019101dd062Clubb JAX (#1351)editstats_netcdf.F90githttp://carson.math.uwm.edu/jenkins/user/noreplynoreplyfalse507http://carson.math.uwm.edu/jenkins/view/all/job/sam_clubb_nvhpc_build/507/