fla-core==0.5.1
transformers>=4.45.0

[benchmark]
matplotlib
datasets>=3.3.0

[conv1d]
causal-conv1d>=1.4.0

[cpu]
fla-core[cpu]==0.5.1

[cuda]
fla-core[cuda]==0.5.1

[npu]
fla-core[npu]==0.5.1

[rocm]
fla-core[rocm]==0.5.1

[test]
pytest

[tilelang]
tilelang>=0.1.9

[xpu]
fla-core[xpu]==0.5.1
