# --------- pytorch --------- #
torch==2.3.1
torchvision>=0.15.0
torch_geometric>=2.0.0
lightning>=2.0.0
torchmetrics>=0.11.4

# --------- hydra --------- #
hydra-core==1.3.2
hydra-colorlog==1.2.0
hydra-optuna-sweeper==1.2.0

# --------- loggers --------- #
wandb

# --------- others --------- #
rootutils
pre-commit
rich
matplotlib
torchdyn>=1.0.1
jaxtyping
ema-pytorch
pyro-ppl>=1.9.0
torchdiffeq
hydra-submitit-launcher
