absl-py==2.5.0
equinox==0.13.8
jax==0.11.2
jax-cuda13-pjrt==0.11.2
jax-cuda13-plugin==0.11.2
jaxlib==0.11.2
jaxtyping==0.3.11
ml_dtypes==0.6.0
numpy==2.5.3
nvidia-cublas==13.8.0.4
nvidia-cuda-cccl==13.3.4.3.1
nvidia-cuda-crt==13.4.92
nvidia-cuda-cupti==13.4.92
nvidia-cuda-nvcc==13.4.92
nvidia-cuda-nvrtc==13.4.92
nvidia-cuda-runtime==13.4.92
nvidia-cudnn-cu13==9.26.0.51
nvidia-cufft==12.4.0.43
nvidia-cusolver==12.3.4.7
nvidia-cusparse==12.8.6.72
nvidia-nccl-cu13==2.31.2
nvidia-nvjitlink==13.4.92
nvidia-nvshmem-cu13==3.7.2
nvidia-nvvm==13.4.92
opt_einsum==3.4.0
optax==0.2.8
scipy==1.18.1
typing_extensions==4.16.0
wadler_lindig==0.1.7
