README.rst
setup.py
fast_transformers/__init__.py
fast_transformers/masking.py
fast_transformers/transformers.py
fast_transformers/weight_mapper.py
fast_transformers/aggregate/__init__.py
fast_transformers/aggregate/aggregate_cpu.cpp
fast_transformers/aggregate/aggregate_cuda.cu
fast_transformers/aggregate/clustered_broadcast_cuda.cu
fast_transformers/attention/__init__.py
fast_transformers/attention/attention_layer.py
fast_transformers/attention/causal_linear_attention.py
fast_transformers/attention/clustered_attention.py
fast_transformers/attention/conditional_full_attention.py
fast_transformers/attention/exact_topk_attention.py
fast_transformers/attention/full_attention.py
fast_transformers/attention/improved_clustered_attention.py
fast_transformers/attention/improved_clustered_causal_attention.py
fast_transformers/attention/linear_attention.py
fast_transformers/attention/reformer_attention.py
fast_transformers/attention_registry/__init__.py
fast_transformers/attention_registry/registry.py
fast_transformers/attention_registry/spec.py
fast_transformers/builders/__init__.py
fast_transformers/builders/attention_builders.py
fast_transformers/builders/base.py
fast_transformers/builders/transformer_builders.py
fast_transformers/causal_product/__init__.py
fast_transformers/causal_product/causal_product_cpu.cpp
fast_transformers/causal_product/causal_product_cuda.cu
fast_transformers/clustering/__init__.py
fast_transformers/clustering/hamming/__init__.py
fast_transformers/clustering/hamming/cluster_cpu.cpp
fast_transformers/clustering/hamming/cluster_cuda.cu
fast_transformers/hashing/__init__.py
fast_transformers/hashing/hash_cpu.cpp
fast_transformers/hashing/hash_cuda.cu
fast_transformers/recurrent/__init__.py
fast_transformers/recurrent/_utils.py
fast_transformers/recurrent/transformers.py
fast_transformers/recurrent/attention/__init__.py
fast_transformers/recurrent/attention/cross_attention/__init__.py
fast_transformers/recurrent/attention/cross_attention/attention_layer.py
fast_transformers/recurrent/attention/cross_attention/full_attention.py
fast_transformers/recurrent/attention/cross_attention/linear_attention.py
fast_transformers/recurrent/attention/self_attention/__init__.py
fast_transformers/recurrent/attention/self_attention/attention_layer.py
fast_transformers/recurrent/attention/self_attention/full_attention.py
fast_transformers/recurrent/attention/self_attention/linear_attention.py
fast_transformers/sparse_product/__init__.py
fast_transformers/sparse_product/clustered_sparse_product_cpu.cpp
fast_transformers/sparse_product/clustered_sparse_product_cuda.cu
fast_transformers/sparse_product/sparse_product_cpu.cpp
fast_transformers/sparse_product/sparse_product_cuda.cu
pytorch_fast_transformers.egg-info/PKG-INFO
pytorch_fast_transformers.egg-info/SOURCES.txt
pytorch_fast_transformers.egg-info/dependency_links.txt
pytorch_fast_transformers.egg-info/requires.txt
pytorch_fast_transformers.egg-info/top_level.txt
tests/aggregate/__init__.py
tests/aggregate/test_aggregate_cpu.py
tests/aggregate/test_aggregate_gpu.py
tests/aggregate/test_clustered_broadcast_gpu.py
tests/causal_product/__init__.py
tests/causal_product/test_causal_product.py
tests/causal_product/test_causal_product_cpu.py
tests/causal_product/test_causal_product_gpu.py
tests/clustering/__init__.py
tests/clustering/hamming/__init__.py
tests/clustering/hamming/test_cluster_cpu.py
tests/clustering/hamming/test_cluster_gpu.py
tests/clustering/hamming/test_python_api_gpu.py
tests/clustering/hamming/time_python_api_gpu.py
tests/hashing/__init__.py
tests/hashing/test_hash_cpu.py
tests/hashing/test_hash_gpu.py
tests/recurrent/__init__.py
tests/recurrent/test_transformer_decoder.py
tests/recurrent/test_transformer_encoder.py
tests/recurrent/attention/__init__.py
tests/recurrent/attention/cross_attention/__init__.py
tests/recurrent/attention/cross_attention/test_attention_layer.py
tests/recurrent/attention/cross_attention/test_full_attention.py
tests/recurrent/attention/cross_attention/test_linear_attention.py
tests/recurrent/attention/self_attention/__init__.py
tests/recurrent/attention/self_attention/test_attention_layer.py
tests/recurrent/attention/self_attention/test_full_attention.py
tests/recurrent/attention/self_attention/test_linear_attention.py
tests/sparse_product/__init__.py
tests/sparse_product/test_clustered_sparse_product_backward_cpu.py
tests/sparse_product/test_clustered_sparse_product_backward_gpu.py
tests/sparse_product/test_clustered_sparse_product_cpu.py
tests/sparse_product/test_clustered_sparse_product_gpu.py
tests/sparse_product/test_clustered_sparse_weighted_average_cpu.py
tests/sparse_product/test_clustered_sparse_weighted_average_gpu.py
tests/sparse_product/test_sparse_product_backward_cpu.py
tests/sparse_product/test_sparse_product_backward_gpu.py
tests/sparse_product/test_sparse_product_cpu.py
tests/sparse_product/test_sparse_product_gpu.py
tests/sparse_product/test_sparse_weighted_average_cpu.py
tests/sparse_product/test_sparse_weighted_average_gpu.py