dacite
jax
flax
ml-collections
tensorflow
tensorflow-datasets
