torch>=2.0.0
torchrl>=0.1.1
tensordict>=0.1.1
lightning>=2.0.5
hydra-core
hydra-colorlog
omegaconf
pyrootutils
rich
einops
wandb
matplotlib
scipy

[dev]
black
ruff
pre-commit>=3.3.3

[graph]
torch_geometric

[testing]
pytest
pytest-cov
