MANIFEST.in
README.md
pyproject.toml
src/cortex/__init__.py
src/cortex/config.py
src/cortex/consistent_dropout.py
src/cortex/factory.py
src/cortex/py.typed
src/cortex/registry.py
src/cortex/routed_adapter.py
src/cortex/tokens.py
src/cortex/torch_init.py
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/gru_gating.py
src/cortex/blocks/passthrough.py
src/cortex/blocks/postup.py
src/cortex/blocks/postup_gated.py
src/cortex/blocks/preup.py
src/cortex/blocks/preup_gated.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/cells/__init__.py
src/cortex/cells/agalite.py
src/cortex/cells/base.py
src/cortex/cells/conv.py
src/cortex/cells/hf_llama.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/cuda/__init__.py
src/cortex/kernels/cuda/agalite/__init__.py
src/cortex/kernels/cuda/agalite/discounted_sum_binding.cpp
src/cortex/kernels/cuda/agalite/discounted_sum_cuda.py
src/cortex/kernels/cuda/agalite/discounted_sum_kernels.cu
src/cortex/kernels/cuda/rtu/__init__.py
src/cortex/kernels/cuda/rtu/rtu_seq_allin.py
src/cortex/kernels/cuda/rtu/rtu_seq_allin_binding.cpp
src/cortex/kernels/cuda/rtu/rtu_seq_allin_kernels.cu
src/cortex/kernels/cuda/rtu/rtu_seq_full.py
src/cortex/kernels/cuda/rtu/rtu_seq_full_binding.cpp
src/cortex/kernels/cuda/rtu/rtu_seq_full_kernels.cu
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/srht_binding.cpp
src/cortex/kernels/cuda/srht/srht_cuda.py
src/cortex/kernels/cuda/srht/srht_kernels.cu
src/cortex/kernels/pytorch/__init__.py
src/cortex/kernels/pytorch/agalite.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/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/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/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/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/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/rl/__init__.py
src/cortex/rl/diff_horde.py
src/cortex/rl/feature_extractors/__init__.py
src/cortex/rl/feature_extractors/box_cnn.py
src/cortex/rl/feature_extractors/config.py
src/cortex/rl/feature_extractors/factory.py
src/cortex/rl/feature_extractors/token_embed.py
src/cortex/rl/feature_extractors/token_mlp.py
src/cortex/rl/feature_extractors/token_perceiver.py
src/cortex/stacks/__init__.py
src/cortex/stacks/auto.py
src/cortex/stacks/base.py
src/cortex/stacks/hf.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_agalite.py
tests/test_axon_layer.py
tests/test_causal_conv1d_kernel.py
tests/test_column_block.py
tests/test_consistent_dropout.py
tests/test_cortex_stack.py
tests/test_diff_horde.py
tests/test_hf_llama_parity.py
tests/test_layernorm_skip.py
tests/test_lstm_cell.py
tests/test_mlstm_cell.py
tests/test_routed_adapter.py
tests/test_rtu_stream_diag.py
tests/test_rtu_stream_fullrank.py
tests/test_slstm_cell.py
tests/test_xl_cell.py