matplotlib
seaborn
scikit-learn
pandas
nbdev
jupyter
dm-haiku
test_tube
jax[cpu]
tqdm
optax
pydantic<2,>=1.9.0
deprecation

[dev]
torch>=1.7.0
