tensorflow>=2.16.1
numpy
pyDOE
matplotlib
jax>=0.4.28
flax>=0.8.4
optax>=0.2.2
