flax
jax
jaxlib
numpy
overrides

[:python_version < "3.7.0"]
dataclasses
