fla-core==0.6.0
transformers<6,>=5.6.2

[benchmark]
matplotlib
pandas
datasets>=3.3.0

[conv1d]
causal-conv1d>=1.7.0

[cpu]
fla-core[cpu]==0.6.0

[cuda]
fla-core[cuda]==0.6.0

[npu]
fla-core[npu]==0.6.0

[rocm]
fla-core[rocm]==0.6.0

[test]
pytest
pytest-xdist

[tilelang]
tilelang>=0.1.9

[xpu]
fla-core[xpu]==0.6.0
