# It's possible that these could be made more lenient
dm-haiku>=0.0.5
jax>=0.2.28
jaxlib>=0.1.74
