numpy>=1.20.0
pandas>=1.3.0
jax>=0.10.1
flax>=0.12.7
jaxtyping>=0.3.10
omegaconf>=2.3
orbax-checkpoint>=0.12.0
wandb>=0.28.0
huggingface_hub>=0.24.0

[all]
estimint[download,gpu,scenarios,train,viz]

[download]
requests>=2.28.0
appdirs>=1.4.0

[gpu]
jax[cuda12]>=0.10.1

[scenarios]
mintstate>=0.3.0

[train]
duckdb>=0.8.0
grain>=0.2.16
hydra-core>=1.3.2
optax>=0.2.8
tqdm>=4.67.3

[viz]
matplotlib>=3.4.0
