Console Output
+ source /etc/profile.d/larson-group.sh
++ export GIT_EDITOR=vi
++ GIT_EDITOR=vi
++ export SVN_EDITOR=vi
++ SVN_EDITOR=vi
++ export OMP_STACKSIZE=1048579
++ OMP_STACKSIZE=1048579
++ export LMOD_ROOT=/opt/lmod/
++ LMOD_ROOT=/opt/lmod/
++ source /opt/lmod//lmod/lmod/init/bash
+++ '[' -z '' ']'
+++ case "$-" in
+++ __lmod_vx=x
+++ '[' -n x ']'
+++ set +x
Shell debugging temporarily silenced: export LMOD_SH_DBG_ON=1 for this output (/usr/local/spack/opt/spack/linux-pop22-cascadelake/gcc-12.2.0/lmod-8.7.37-bi3kyxcdrfgw3y7vv2k7c5rjxg75qzju/lmod/lmod/init/bash)
Shell debugging restarted
+++ unset __lmod_vx
+++ find /usr/local/spack/share/spack/lmod/linux-pop22-x86_64/Core -print -quit
++ export MODULEPATH=/usr/local/spack/share/spack/lmod/linux-pop22-x86_64/Core
++ MODULEPATH=/usr/local/spack/share/spack/lmod/linux-pop22-x86_64/Core
+ module load gcc netcdf-fortran
+ '[' -z '' ']'
+ case "$-" in
+ __lmod_sh_dbg=x
+ '[' -n x ']'
+ set +x
Shell debugging temporarily silenced: export LMOD_SH_DBG_ON=1 for Lmod's output
Lmod Warning:
-------------------------------------------------------------------------------
The following dependent module(s) are not currently loaded:
gcc-runtime/13.1.0-gzedt3f (required by: mpich/4.2.1, hdf5/1.14.3-42x3zrv,
netcdf-c/4.9.2, netcdf-fortran/4.5.3)
-------------------------------------------------------------------------------
Shell debugging restarted
+ unset __lmod_sh_dbg
+ return 0
+ python3 tests/run_jax_timestep_gradient_test.py
==> No compatible Python found for cpu; uv will download Python 3.12
==> Installing uv 0.11.32 in /home/jenkins/workspace/clubb_branch_jax/.clubb-jax-tools/bin
downloading uv 0.11.32 x86_64-unknown-linux-gnu
installing to /home/jenkins/workspace/clubb_branch_jax/.clubb-jax-tools/bin
uv
uvx
everything's installed!
==> uv is ready: uv 0.11.32 (x86_64-unknown-linux-gnu)
Downloading cpython-3.12.13-linux-x86_64-gnu (download) (32.6MiB)
Downloaded cpython-3.12.13-linux-x86_64-gnu (download)
Using CPython 3.12.13
Creating virtual environment at: .venv-jax
Activate with: source .venv-jax/bin/activate
==> Installing JAX 0.11.0 (cpu) and requirements from /home/jenkins/workspace/clubb_branch_jax/clubb_jax/requirements.txt
Using Python 3.12.13 environment at: .venv-jax
Resolved 15 packages in 324ms
Downloading pygments (1.2MiB)
Downloading jaxlib (83.2MiB)
Downloading jax (3.1MiB)
Downloading netcdf4 (9.7MiB)
Downloading numpy (15.9MiB)
Downloading cftime (1.6MiB)
Downloading scipy (33.7MiB)
Downloaded cftime
Downloaded pygments
Downloaded netcdf4
Downloaded jax
Downloaded numpy
Downloaded jaxlib
Downloaded scipy
Prepared 15 packages in 2.89s
Installed 15 packages in 41ms
+ certifi==2026.7.22
+ cftime==1.6.6
+ iniconfig==2.3.0
+ jax==0.11.0
+ jaxlib==0.11.0
+ ml-dtypes==0.6.0
+ netcdf4==1.7.4
+ numpy==2.5.3
+ opt-einsum==3.4.0
+ packaging==26.3
+ pluggy==1.6.0
+ pygments==2.21.0
+ pytest==9.1.1
+ scipy==1.18.1
+ tabulate==0.10.0
CLUBB JAX runtime:
Environment: /home/jenkins/workspace/clubb_branch_jax/.venv-jax (Python 3.12)
Accelerator: cpu
JAX: 0.11.0 (jaxlib 0.11.0)
Device: cpu:0
An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
nzm = 88 -- nzt = 87
Initialized bomex case: nzm=88, nzt=87, ngrdcol=1
dt_main=60.0s, time=0.0s to 60.0s, 1 steps
bomex: full driver gradient/JVP/finite difference/eager checks passed
CLUBB cleanup complete.
nzm = 199 -- nzt = 198
Initialized atex case: nzm=199, nzt=198, ngrdcol=1
dt_main=60.0s, time=0.0s to 60.0s, 1 steps
atex: full driver gradient/JVP/finite difference/eager checks passed
CLUBB cleanup complete.
