einops
torch>=1.8

[test]
black
kaleido
mypy
pre-commit
plotly
pytest
pytest-cov
ruff
types-requests

[train]
lightning~=2.0.0
tensorboard~=2.14.0
tiktoken~=0.4.0
torchdata>=0.6.0
tqdm
