hatchling
jax
jaxopt>=0.6
typing-extensions>=4.5.0
wandb>=0.14.2
pandas==1.5.3
plotly>=5.14.1
flax>=0.6.10
tensorflow_datasets
