jax==0.5.0
keras==3.9.0
tensorflow==2.18.0