torch>=2.0.0
numpy>=1.24.0
scipy>=1.10.0
matplotlib>=3.6.0
seaborn>=0.12.0
tqdm>=4.64.0
pydantic>=2.0.0
typer>=0.9.0
rich>=13.0.0
wandb>=0.15.0
networkx>=3.0
scikit-learn>=1.3.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
isort>=5.12.0
flake8>=6.0.0
mypy>=1.0.0
pre-commit>=3.0.0

[docs]
mkdocs>=1.5.0
mkdocs-material>=9.0.0
mkdocstrings[python]>=0.22.0
mkdocs-jupyter>=0.24.0
mkdocs-mermaid2-plugin>=1.0.0

[jax]
jax>=0.4.0
jaxlib>=0.4.0
flax>=0.7.0
optax>=0.1.7
