torch>=2.0
numpy>=1.21

[all]
cort-transformer[dev,plotting,training]

[dev]
pytest>=7.0
pytest-cov>=4.0
black>=23.0
isort>=5.12
mypy>=1.0
ruff>=0.1

[plotting]
matplotlib>=3.5
seaborn>=0.12

[training]
tiktoken>=0.5
datasets>=2.0
tqdm>=4.0
