README.md
pyproject.toml
src/cortex/__init__.py
src/cortex/config.py
src/cortex/factory.py
src/cortex/py.typed
src/cortex/types.py
src/cortex/utils.py
src/cortex/blocks/__init__.py
src/cortex/blocks/adapter.py
src/cortex/blocks/base.py
src/cortex/blocks/passthrough.py
src/cortex/blocks/postup.py
src/cortex/blocks/preup.py
src/cortex/blocks/registry.py
src/cortex/blocks/column/__init__.py
src/cortex/blocks/column/auto.py
src/cortex/blocks/column/column.py
src/cortex/blocks/column/routers.py
src/cortex/blocks/column/tokens.py
src/cortex/cells/__init__.py
src/cortex/cells/base.py
src/cortex/cells/conv.py
src/cortex/cells/lstm.py
src/cortex/cells/mlstm.py
src/cortex/cells/registry.py
src/cortex/cells/slstm.py
src/cortex/cells/xl.py
src/cortex/cells/core/__init__.py
src/cortex/cells/core/axon_cell.py
src/cortex/cells/core/axon_layer.py
src/cortex/kernels/__init__.py
src/cortex/kernels/cuda/__init__.py
src/cortex/kernels/cuda/rtu/__init__.py
src/cortex/kernels/cuda/rtu/rtu_seq_allin.py
src/cortex/kernels/cuda/rtu/rtu_seq_full.py
src/cortex/kernels/cuda/rtu/rtu_stream_diag_cuda.py
src/cortex/kernels/cuda/rtu/rtu_stream_full_cuda.py
src/cortex/kernels/cuda/srht/__init__.py
src/cortex/kernels/cuda/srht/srht_cuda.py
src/cortex/kernels/pytorch/__init__.py
src/cortex/kernels/pytorch/conv1d.py
src/cortex/kernels/pytorch/lstm.py
src/cortex/kernels/pytorch/mlstm.py
src/cortex/kernels/pytorch/slstm.py
src/cortex/kernels/pytorch/srht.py
src/cortex/kernels/pytorch/txl.py
src/cortex/kernels/pytorch/rtu/__init__.py
src/cortex/kernels/pytorch/rtu/rtu_stream_diag.py
src/cortex/kernels/pytorch/rtu/rtu_stream_fullrank.py
src/cortex/kernels/triton/__init__.py
src/cortex/kernels/triton/lstm.py
src/cortex/kernels/triton/conv1d/__init__.py
src/cortex/kernels/triton/conv1d/channel_mixing.py
src/cortex/kernels/triton/mlstm/__init__.py
src/cortex/kernels/triton/mlstm/torch/__init__.py
src/cortex/kernels/triton/mlstm/torch/bw.py
src/cortex/kernels/triton/mlstm/torch/bw_parallel_dK.py
src/cortex/kernels/triton/mlstm/torch/bw_parallel_dQ.py
src/cortex/kernels/triton/mlstm/torch/bw_parallel_dV.py
src/cortex/kernels/triton/mlstm/torch/bw_recurrent.py
src/cortex/kernels/triton/mlstm/torch/chunkwise_gates.py
src/cortex/kernels/triton/mlstm/torch/fw.py
src/cortex/kernels/triton/mlstm/torch/fw_parallel.py
src/cortex/kernels/triton/mlstm/torch/fw_recurrent.py
src/cortex/kernels/triton/mlstm/torch/fwbw.py
src/cortex/kernels/triton/mlstm/torch/utils.py
src/cortex/kernels/triton/mlstm/triton/__init__.py
src/cortex/kernels/triton/mlstm/triton/bw_kernel_parallel_dK.py
src/cortex/kernels/triton/mlstm/triton/bw_kernel_parallel_dQ.py
src/cortex/kernels/triton/mlstm/triton/bw_kernel_parallel_dV.py
src/cortex/kernels/triton/mlstm/triton/bw_kernel_recurrent.py
src/cortex/kernels/triton/mlstm/triton/chunkwise_kernel_param_heuristics.py
src/cortex/kernels/triton/mlstm/triton/fw_kernel_parallel.py
src/cortex/kernels/triton/mlstm/triton/fw_kernel_recurrent.py
src/cortex/kernels/triton/mlstm/triton/kernel_param_heuristics.py
src/cortex/kernels/triton/mlstm/utils/__init__.py
src/cortex/kernels/triton/mlstm/utils/kernels.py
src/cortex/kernels/triton/rtu/__init__.py
src/cortex/kernels/triton/rtu/lowrank.py
src/cortex/kernels/triton/rtu/stream_diag.py
src/cortex/kernels/triton/rtu/utils.py
src/cortex/kernels/triton/slstm/__init__.py
src/cortex/kernels/triton/slstm/torch/__init__.py
src/cortex/kernels/triton/slstm/torch/fwbw.py
src/cortex/kernels/triton/slstm/triton_fused/slstm_bw.py
src/cortex/kernels/triton/slstm/triton_fused/slstm_fw.py
src/cortex/kernels/triton/slstm/triton_fused/triton_utils.py
src/cortex/stacks/__init__.py
src/cortex/stacks/auto.py
src/cortex/stacks/base.py
src/cortex/stacks/xlstm.py
src/cortexcore.egg-info/PKG-INFO
src/cortexcore.egg-info/SOURCES.txt
src/cortexcore.egg-info/dependency_links.txt
src/cortexcore.egg-info/requires.txt
src/cortexcore.egg-info/top_level.txt
tests/test_adapter.py
tests/test_axon_layer.py
tests/test_causal_conv1d_kernel.py
tests/test_column_block.py
tests/test_cortex_stack.py
tests/test_layernorm_skip.py
tests/test_lstm_cell.py
tests/test_mlstm_cell.py
tests/test_rtu_stream_diag.py
tests/test_rtu_stream_fullrank.py
tests/test_slstm_cell.py
tests/test_xl_cell.py