numpy>=2.2
torch==2.6.0
triton>=3.2
setuptools
einops
vidrial==0.1.1

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