jax>=0.5.0
jaxtyping>=0.3.0
pandas>=2.0.3
tqdm>=4.67.1

[dev]
pandas-stubs<3
ruff

[docs]
quartodoc
blackjax>=1.2.5
optax>=0.2.4
plotly>=7.1.0
jupyter>=1.1.1
imageio>=2.37.0
jupyter-cache>=1.0.1
griffe<2
quarto-cli>=1.10.19

[experiments]
arviz>=0.23.4
blackjax>=1.2.5
optax>=0.2.4

[test]
pytest>=8.3.5
