jax
jaxlib>=0.1.37
chex
flax
numpy
pyyaml
matplotlib
dotmap

[:python_version < "3.8"]
pickle5
