LICENSE
MANIFEST.in
NOTICE
README.md
pyproject.toml
flash_xlstm_exp.egg-info/PKG-INFO
flash_xlstm_exp.egg-info/SOURCES.txt
flash_xlstm_exp.egg-info/dependency_links.txt
flash_xlstm_exp.egg-info/requires.txt
flash_xlstm_exp.egg-info/top_level.txt
fsp/__init__.py
fsp/benchmark/__init__.py
fsp/benchmark/bench_autotune_vs_baseline.py
fsp/kernels/__init__.py
fsp/kernels/autotune.py
fsp/vendor/__init__.py
fsp/vendor/torch_fw/__init__.py
fsp/vendor/torch_fw/chunkwise_gates.py
fsp/vendor/torch_fw/fw.py
fsp/vendor/torch_fw/fw_parallel.py
fsp/vendor/torch_fw/fw_recurrent.py
fsp/vendor/triton_kernels/__init__.py
fsp/vendor/triton_kernels/chunkwise_param_heuristics.py
fsp/vendor/triton_kernels/fw_kernel_parallel.py
fsp/vendor/triton_kernels/fw_kernel_recurrent.py
fsp/vendor/triton_kernels/kernel_param_heuristics.py
fsp/vendor/triton_kernels/xl_chunk_init.py
fsp/vendor/utils/__init__.py
fsp/vendor/utils/kernels.py
fsp/vendor/utils/torch_utils.py
tests/test_autotune_equivalence.py