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

[dev]
torch>=1.7.0
causalgraphicalmodels
pre-commit
