setuptools
torch
einops
transformers

[flash_attn]
flash_attn

[test]
pytest
numpy
biopython
matplotlib
seaborn
