LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
benchmarks/__init__.py
benchmarks/benchmark_cauchy.py
benchmarks/benchmark_cauchy_tune.py
benchmarks/benchmark_inference.py
benchmarks/benchmark_training.py
benchmarks/run_all.py
csrc/common.h
csrc/reverse_scan.cuh
csrc/s4/cauchy.cpp
csrc/s4/cauchy.py
csrc/s4/cauchy_cuda.cu
csrc/s4/map.h
csrc/s4/tune_cauchy.py
csrc/s4/tuner.py
csrc/s4/tuning_setup.py
csrc/s4/vandermonde.py
csrc/selective_scan/bindings.cpp
csrc/selective_scan/selective_scan.cuh
csrc/selective_scan/selective_scan.h
csrc/selective_scan/selective_scan_backward.cuh
csrc/selective_scan/selective_scan_cpu.cpp
csrc/selective_scan/uninitialized_copy.cuh
csrc/selective_scan/backward_kernels/selective_scan_fp32_complex_bilinear.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_complex_dirac.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_complex_mamba.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_complex_zoh.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_real_bilinear.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_real_dirac.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_real_mamba.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_real_rglru.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_real_s7.cu
csrc/selective_scan/backward_kernels/selective_scan_fp32_real_zoh.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_complex_bilinear.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_complex_dirac.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_complex_mamba.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_complex_zoh.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_real_bilinear.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_real_dirac.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_real_mamba.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_real_rglru.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_real_s7.cu
csrc/selective_scan/forward_kernels/selective_scan_fp32_real_zoh.cu
csrc/simplified_scan/bindings.cpp
csrc/simplified_scan/simplified_scan.cuh
csrc/simplified_scan/simplified_scan.h
csrc/simplified_scan/simplified_scan_backward.cuh
csrc/simplified_scan/simplified_scan_cpu.cpp
csrc/simplified_scan/backward_kernels/simplified_scan_bilinear.cu
csrc/simplified_scan/backward_kernels/simplified_scan_dirac.cu
csrc/simplified_scan/backward_kernels/simplified_scan_zoh.cu
csrc/simplified_scan/forward_kernels/simplified_scan_fp32_bilinear.cu
csrc/simplified_scan/forward_kernels/simplified_scan_fp32_dirac.cu
csrc/simplified_scan/forward_kernels/simplified_scan_fp32_zoh.cu
docs/source/conf.py
docs/source/api/__init__.py
docs/source/guides/__init__.py
docs/source/tutorials/__init__.py
lrnnx/__init__.py
lrnnx.egg-info/PKG-INFO
lrnnx.egg-info/SOURCES.txt
lrnnx.egg-info/dependency_links.txt
lrnnx.egg-info/requires.txt
lrnnx.egg-info/top_level.txt
lrnnx/architectures/__init__.py
lrnnx/architectures/classifier.py
lrnnx/architectures/embedding.py
lrnnx/architectures/language_model.py
lrnnx/architectures/lru_unet.py
lrnnx/core/__init__.py
lrnnx/core/base.py
lrnnx/core/convolution.py
lrnnx/core/discretization.py
lrnnx/layers/block.py
lrnnx/layers/mha.py
lrnnx/layers/mlp.py
lrnnx/models/__init__.py
lrnnx/models/lti/__init__.py
lrnnx/models/lti/base.py
lrnnx/models/lti/centaurus.py
lrnnx/models/lti/lru.py
lrnnx/models/lti/s4.py
lrnnx/models/lti/s4d.py
lrnnx/models/lti/s5.py
lrnnx/models/ltv/__init__.py
lrnnx/models/ltv/base.py
lrnnx/models/ltv/mamba.py
lrnnx/models/ltv/rglru.py
lrnnx/models/ltv/s5.py
lrnnx/models/ltv/s7.py
lrnnx/ops/__init__.py
lrnnx/ops/rglru_scan.py
lrnnx/ops/s4_kernel_interface.py
lrnnx/ops/s4_utils.py
lrnnx/ops/s7_scan.py
lrnnx/ops/selective_scan.py
lrnnx/ops/simplified_scan.py
lrnnx/ops/torch.py
lrnnx/ops/triton/__init__.py
lrnnx/ops/triton/layer_norm.py
lrnnx/ops/triton/selective_state_update.py
lrnnx/ops/triton/simplified_state_update.py
lrnnx/ops/triton/softplus.py
lrnnx/utils/__init__.py
lrnnx/utils/generation.py
lrnnx/utils/init.py
tests/__init__.py
tests/architectures/test_language_model.py
tests/models/test_lti/test_centaurus.py
tests/models/test_lti/test_lru.py
tests/models/test_lti/test_s4.py
tests/models/test_lti/test_s4d.py
tests/models/test_lti/test_s5_lti.py
tests/models/test_ltv/test_event_mamba.py
tests/models/test_ltv/test_event_s5.py
tests/models/test_ltv/test_mamba.py
tests/models/test_ltv/test_rglru.py
tests/models/test_ltv/test_s5_ltv.py
tests/models/test_ltv/test_s7.py
tests/ops/mamba/test_selective_scan.py
tests/ops/mamba/test_selective_scan_async.py
tests/ops/mamba/test_selective_state_update.py
tests/ops/rglru/test_rglru_scan.py
tests/ops/s4/test_cauchy.py
tests/ops/s4/test_vandermonde.py
tests/ops/s5/test_simplified_scan.py
tests/ops/s5/test_simplified_scan_async.py
tests/ops/s5/test_simplified_state_update.py
tests/ops/s7/test_s7_scan.py
tests/utils/__init__.py
tests/utils/test_generation.py