numpy
matplotlib
jax
jaxlib
dm-haiku
optax
ott-jax
networkx
blackjax
sympy

[dev]
autoflake
black
flake8
isort>5.0.0
ipdb
pytest
pytest-plt
typeguard
