AUTHORS
LICENSE
README.md
pyproject.toml
setup.cfg
tests/test_chunkwise_vs_recurrent.py
tests/test_generation.py
tests/test_weight_conversion.py
xlstm/__init__.py
xlstm/utils.py
xlstm/xlstm_block_stack.py
xlstm/xlstm_lm_model.py
xlstm.egg-info/PKG-INFO
xlstm.egg-info/SOURCES.txt
xlstm.egg-info/dependency_links.txt
xlstm.egg-info/requires.txt
xlstm.egg-info/top_level.txt
xlstm/blocks/__init__.py
xlstm/blocks/xlstm_block.py
xlstm/blocks/mlstm/__init__.py
xlstm/blocks/mlstm/backends.py
xlstm/blocks/mlstm/block.py
xlstm/blocks/mlstm/cell.py
xlstm/blocks/mlstm/layer.py
xlstm/blocks/slstm/__init__.py
xlstm/blocks/slstm/block.py
xlstm/blocks/slstm/cell.py
xlstm/blocks/slstm/layer.py
xlstm/blocks/slstm/src/__init__.py
xlstm/blocks/slstm/src/cuda_init.py
xlstm/blocks/slstm/src/cuda/lstm_pointwise.cu
xlstm/blocks/slstm/src/cuda/slstm.cc
xlstm/blocks/slstm/src/cuda/slstm.h
xlstm/blocks/slstm/src/cuda/slstm_backward.cu
xlstm/blocks/slstm/src/cuda/slstm_backward_cut.cu
xlstm/blocks/slstm/src/cuda/slstm_forward.cu
xlstm/blocks/slstm/src/cuda/slstm_pointwise.cu
xlstm/blocks/slstm/src/cuda/slstm_pointwise.cuh
xlstm/blocks/slstm/src/util/blas.cu
xlstm/blocks/slstm/src/util/blas.h
xlstm/blocks/slstm/src/util/cuda_error.cu
xlstm/blocks/slstm/src/util/cuda_error.h
xlstm/blocks/slstm/src/util/device_assert.h
xlstm/blocks/slstm/src/util/inline_ops.cuh
xlstm/blocks/slstm/src/util/inline_ops_2bf16.cuh
xlstm/blocks/slstm/src/util/inline_ops_2fp16.cuh
xlstm/blocks/slstm/src/util/inline_ops_bf16.cuh
xlstm/blocks/slstm/src/util/inline_ops_fp16.cuh
xlstm/blocks/slstm/src/util/inline_print.cuh
xlstm/blocks/slstm/src/util/support.h
xlstm/blocks/slstm/src/util/util.h
xlstm/blocks/slstm/src/vanilla/__init__.py
xlstm/blocks/slstm/src/vanilla/lstm.py
xlstm/blocks/slstm/src/vanilla/slstm.py
xlstm/components/__init__.py
xlstm/components/conv.py
xlstm/components/feedforward.py
xlstm/components/init.py
xlstm/components/linear_headwise.py
xlstm/components/ln.py
xlstm/components/util.py
xlstm/xlstm_large/__init__.py
xlstm/xlstm_large/components.py
xlstm/xlstm_large/from_pretrained.py
xlstm/xlstm_large/generate.py
xlstm/xlstm_large/model.py
xlstm/xlstm_large/utils.py