torch
einops
toml
tqdm
numpy
safetensors

[dev]
coverage
parameterized
diffusers>=0.40.0

[kernels]
tilelang

[kernels:sys_platform == "linux"]
triton

[kernels:sys_platform == "win32"]
triton-windows
