jax==0.4.20
jaxlib==0.4.20
numpy==1.26.1
opt-einsum==3.3.0
