tqdm
ipywidgets
matplotlib
numpy==1.24.4
hf-transfer
huggingface_hub

[dev]
pytest
pytest-cov
flake8
black
isort
mypy

[docs]
pydocs

[gpu]
cupy-cuda12x
torch==2.6.0
einops
einx
