Pins jax==0.4.23 (xla matched), flax==0.7.5, te==1.2.1#493

terrykong wants to merge 2 commits into

archive-24.01-develNVIDIA/JAX-Toolbox:archive-24.01-develfrom

pin-base-libsNVIDIA/JAX-Toolbox:pin-base-libsCopy head branch name to clipboard