torchrl>=0.2.0
tensordict>=0.2.0
lightning>=2.1.0
hydra-core
hydra-colorlog
omegaconf
pyrootutils
rich
einops
wandb
matplotlib
scipy
robust-downloader

[dev]
black
ruff
pre-commit>=3.3.3

[graph]
torch_geometric

[testing]
pytest
pytest-cov
