fla-core==0.5.0
transformers

[benchmark]
matplotlib
datasets>=3.3.0

[conv1d]
causal-conv1d>=1.4.0

[test]
pytest

[tilelang]
tilelang
