LICENSE
MANIFEST.in
README.md
requirements.txt
setup.cfg
setup.py
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/matmul.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/sddmm.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/sparse_softmax.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/spmm.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/autograd/matmul.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/cpu/matmul.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/cpu/sddmm.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/cpu/sparse_softmax.cpp
/private/home/dianaml/src/facebookresearch/xformers/xformers/components/attention/csrc/cpu/spmm.cpp
xformers/__init__.py
xformers/ops.py
xformers/test.py
xformers/utils.py
xformers.egg-info/PKG-INFO
xformers.egg-info/SOURCES.txt
xformers.egg-info/dependency_links.txt
xformers.egg-info/not-zip-safe
xformers.egg-info/requires.txt
xformers.egg-info/top_level.txt
xformers/benchmarks/__init__.py
xformers/benchmarks/benchmark_core.py
xformers/benchmarks/benchmark_encoder.py
xformers/benchmarks/benchmark_nystrom_utils.py
xformers/benchmarks/benchmark_pytorch_transformer.py
xformers/benchmarks/benchmark_revnet.py
xformers/benchmarks/benchmark_sddmm.py
xformers/benchmarks/benchmark_triton_blocksparse.py
xformers/benchmarks/benchmark_triton_dropout.py
xformers/benchmarks/benchmark_triton_fused_linear.py
xformers/benchmarks/benchmark_triton_layernorm.py
xformers/benchmarks/benchmark_triton_softmax.py
xformers/benchmarks/benchmark_triton_stride_sum.py
xformers/benchmarks/benchmark_vit_timm.py
xformers/benchmarks/utils.py
xformers/benchmarks/LRA/__init__.py
xformers/benchmarks/LRA/batch_fetch_results.py
xformers/benchmarks/LRA/batch_submit.py
xformers/benchmarks/LRA/run_grid_search.py
xformers/benchmarks/LRA/run_tasks.py
xformers/benchmarks/LRA/run_with_submitit.py
xformers/benchmarks/LRA/code/__init__.py
xformers/benchmarks/LRA/code/dataset.py
xformers/benchmarks/LRA/code/model_wrapper.py
xformers/components/__init__.py
xformers/components/activations.py
xformers/components/in_proj_container.py
xformers/components/multi_head_dispatch.py
xformers/components/residual.py
xformers/components/reversible.py
xformers/components/attention/__init__.py
xformers/components/attention/_sputnik_sparse.py
xformers/components/attention/attention_mask.py
xformers/components/attention/attention_patterns.py
xformers/components/attention/base.py
xformers/components/attention/blocksparse.py
xformers/components/attention/core.py
xformers/components/attention/favor.py
xformers/components/attention/fourier_mix.py
xformers/components/attention/global_tokens.py
xformers/components/attention/lambda_layer.py
xformers/components/attention/linformer.py
xformers/components/attention/local.py
xformers/components/attention/nystrom.py
xformers/components/attention/ortho.py
xformers/components/attention/random.py
xformers/components/attention/scaled_dot_product.py
xformers/components/attention/utils.py
xformers/components/attention/feature_maps/__init__.py
xformers/components/attention/feature_maps/base.py
xformers/components/attention/feature_maps/softmax.py
xformers/components/feedforward/__init__.py
xformers/components/feedforward/base.py
xformers/components/feedforward/fused_mlp.py
xformers/components/feedforward/mlp.py
xformers/components/positional_embedding/__init__.py
xformers/components/positional_embedding/base.py
xformers/components/positional_embedding/rotary.py
xformers/components/positional_embedding/sine.py
xformers/components/positional_embedding/vocab.py
xformers/factory/__init__.py
xformers/factory/block_factory.py
xformers/factory/hydra_helper.py
xformers/factory/model_factory.py
xformers/helpers/__init__.py
xformers/helpers/timm_sparse_attention.py
xformers/sparse/__init__.py
xformers/sparse/_csr_ops.py
xformers/sparse/csr_tensor.py
xformers/sparse/utils.py
xformers/triton/__init__.py
xformers/triton/dropout.py
xformers/triton/fused_linear_layer.py
xformers/triton/k_activations.py
xformers/triton/k_dropout.py
xformers/triton/k_fused_matmul_bw.py
xformers/triton/k_fused_matmul_fw.py
xformers/triton/k_layer_norm.py
xformers/triton/k_softmax.py
xformers/triton/k_sum.py
xformers/triton/layer_norm.py
xformers/triton/softmax.py
xformers/triton/sum_strided.py
xformers/triton/utils.py