torch>=2.1
transformers>=4.37
triton>=2.1

[dev]
pytest==6.2.5
pytest-forked
pytest-asyncio==0.16.0
black==22.3.0
isort==5.13.2
