jax==0.5.0
keras==3.10.0
tensorflow==2.18.0