torch>=2.0.0
torchrl>=0.1.1
tensordict>=0.1.1
lightning>=2.0.0
hydra-core
hydra-colorlog
omegaconf
pyrootutils
rich
numpy
einops
wandb
pre-commit>=3.3.3
matplotlib
scipy
pydantic<2.0.0

[graph]
torch_geometric

[linting]
black
ruff

[testing]
pytest
