google-tunix[prod]>=0.1.6
optax>=0.2.5
multimethod>=2.0
