numpy>=2.2
torch>=2.6.0
triton>=3.2
setuptools>=69.5.1
einops
vidrial==0.2.1

[dev]
twine
pytest>=8.3.4
transformers>=4.56.0
pytest-xdist
cibuildwheel
auditwheel
wheel
build
tiktoken
datasets
httpx
tqdm
pyyaml
bokeh
matplotlib
pymdown-extensions
mkdocs
mkdocstrings-python
requests
wandb
zarr
psutil
click
tabulate
flash-linear-attention
flash_attn==2.7.3
pytest-randomly
pytest-rerunfailures
