triton==2.0.0
numba
numpy
torch
tqdm
more-itertools
tiktoken==0.3.1
ffmpeg-python==0.2.0

[dev]
pytest
scipy
black
flake8
isort
