numpy==1.24.2
pandas==1.4.4
scikit-learn==1.2.1
tensorboard
torch==2.1.2
torch-geometric==2.4.0
torchmetrics==1.2.1
seaborn==0.12.2
pytorch-lightning==2.1.3
diffusers==0.26.3
torchcde==0.2.5
mamba-ssm==1.2.0.post1
causal-conv1d>=1.2.0
torchinfo==1.8.0
timm==0.9.16
schedulefree==1.2.1
torch-ema==0.3
hydra-core==1.3.0
linear-attention-transformer==0.19.1