torch>=2.0.0
pytorch-lightning>=2.0.0
omegaconf>=2.3.0
numpy>=1.23.0
h5py>=3.7.0
tqdm>=4.64.0
einops>=0.6.0
scipy>=1.10.0

[all]
flash-attn>=2.0.0
wandb>=0.16.0

[dev]
pytest>=7.0.0
build>=1.0
twine>=4.0

[flash]
flash-attn>=2.0.0

[logging]
wandb>=0.16.0
