file 1 file 2 # same commits # commits file 1 # commits file 2 xformers/components/attention/scaled_dot_product.py xformers/components/attention/core.py 5 7 9 xformers/triton/dropout.py xformers/triton/k_dropout.py 4 8 5 xformers/triton/dropout.py xformers/benchmarks/benchmark_triton_dropout.py 4 8 4 xformers/factory/block_factory.py xformers/factory/model_factory.py 3 8 5 xformers/factory/block_factory.py xformers/components/__init__.py 3 8 7 stubs/torch/linalg/__init__.pyi stubs/numpy/__init__.pyi 3 3 3 stubs/torch/nn/__init__.pyi stubs/numpy/__init__.pyi 3 3 3 stubs/torch/nn/__init__.pyi stubs/torch/linalg/__init__.pyi 3 3 3 stubs/torch/nn/init.pyi stubs/numpy/__init__.pyi 3 3 3 stubs/torch/nn/init.pyi stubs/torch/linalg/__init__.pyi 3 3 3 stubs/torch/nn/init.pyi stubs/torch/nn/__init__.pyi 3 3 3 stubs/torch/sparse/__init__.pyi stubs/numpy/__init__.pyi 3 3 3 stubs/torch/sparse/__init__.pyi stubs/torch/linalg/__init__.pyi 3 3 3 stubs/torch/sparse/__init__.pyi stubs/torch/nn/__init__.pyi 3 3 3 stubs/torch/sparse/__init__.pyi stubs/torch/nn/init.pyi 3 3 3 xformers/components/attention/nystrom.py xformers/components/attention/scaled_dot_product.py 3 5 7 xformers/components/attention/attention_mask.py xformers/components/__init__.py 2 2 7 xformers/components/attention/base.py xformers/components/__init__.py 2 5 7 xformers/components/attention/base.py xformers/components/attention/attention_mask.py 2 5 2 xformers/components/attention/favor.py xformers/components/attention/base.py 2 5 5 xformers/components/multi_head_dispatch.py xformers/components/attention/base.py 2 4 5 xformers/benchmarks/benchmark_triton_dropout.py xformers/triton/k_dropout.py 2 4 5 xformers/triton/k_activations.py xformers/triton/dropout.py 2 2 8 xformers/triton/k_fused_matmul_bw.py xformers/benchmarks/benchmark_triton_dropout.py 2 3 4 xformers/triton/k_fused_matmul_bw.py xformers/triton/dropout.py 2 3 8 xformers/components/attention/global_tokens.py xformers/components/attention/base.py 2 3 5 xformers/components/attention/global_tokens.py xformers/components/attention/blocksparse.py 2 3 9 xformers/components/attention/lambda_layer.py xformers/components/attention/blocksparse.py 2 2 9 xformers/components/attention/lambda_layer.py xformers/components/attention/global_tokens.py 2 2 3 xformers/components/attention/linformer.py xformers/components/attention/blocksparse.py 2 2 9 xformers/components/attention/linformer.py xformers/components/attention/global_tokens.py 2 2 3 xformers/components/attention/linformer.py xformers/components/attention/lambda_layer.py 2 2 2 xformers/components/attention/random.py xformers/components/attention/base.py 2 3 5 xformers/components/attention/random.py xformers/components/attention/global_tokens.py 2 3 3 xformers/components/attention/random.py xformers/components/attention/local.py 2 3 3 xformers/components/positional_embedding/rotary.py xformers/components/multi_head_dispatch.py 2 2 4 xformers/triton/fused_linear_layer.py xformers/triton/dropout.py 2 3 8 xformers/triton/fused_linear_layer.py xformers/triton/k_fused_matmul_bw.py 2 3 3 xformers/components/attention/core.py xformers/components/attention/blocksparse.py 2 9 9 xformers/components/attention/core.py xformers/components/attention/global_tokens.py 2 9 3 xformers/components/attention/core.py xformers/components/attention/base.py 2 9 5 xformers/components/attention/core.py xformers/__init__.py 2 9 8 xformers/components/attention/core.py xformers/components/attention/random.py 2 9 3 xformers/components/attention/core.py xformers/components/attention/favor.py 2 9 5 xformers/components/attention/scaled_dot_product.py xformers/components/attention/blocksparse.py 2 7 9 xformers/components/attention/scaled_dot_product.py xformers/components/attention/global_tokens.py 2 7 3 xformers/components/attention/scaled_dot_product.py xformers/components/attention/base.py 2 7 5 stubs/torch/__init__.pyi stubs/numpy/__init__.pyi 2 3 3 stubs/torch/__init__.pyi stubs/torch/linalg/__init__.pyi 2 3 3 stubs/torch/__init__.pyi stubs/torch/nn/__init__.pyi 2 3 3 stubs/torch/__init__.pyi stubs/torch/nn/init.pyi 2 3 3 stubs/torch/__init__.pyi stubs/torch/sparse/__init__.pyi 2 3 3 stubs/torch/autograd/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/autograd/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/autograd/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/autograd/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/autograd/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/autograd/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/autograd/profiler.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/autograd/profiler.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/autograd/profiler.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/autograd/profiler.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/autograd/profiler.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/autograd/profiler.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/autograd/profiler.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/cuda/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/cuda/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/cuda/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/cuda/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/cuda/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/cuda/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/cuda/__init__.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/cuda/__init__.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/fft/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/fft/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/fft/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/fft/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/fft/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/fft/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/fft/__init__.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/fft/__init__.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/fft/__init__.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/hub.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/hub.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/hub.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/hub.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/hub.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/hub.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/hub.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/hub.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/hub.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/hub.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/nn/functional.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/nn/functional.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/nn/functional.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/nn/functional.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/nn/functional.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/nn/functional.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/nn/functional.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/nn/functional.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/nn/functional.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/nn/functional.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/nn/functional.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/nn/functional/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/nn/functional/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/nn/functional/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/nn/functional/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/nn/functional/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/nn/functional/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/nn/functional/__init__.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/nn/functional/__init__.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/nn/functional/__init__.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/nn/functional/__init__.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/nn/functional/__init__.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/nn/functional/__init__.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/nn/utils/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/nn/utils/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/nn/utils/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/nn/utils/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/nn/utils/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/nn/utils/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/nn/utils/__init__.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/nn/utils/__init__.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/nn/utils/__init__.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/nn/utils/__init__.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/nn/utils/__init__.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/nn/utils/__init__.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/nn/utils/__init__.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/onnx.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/onnx.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/onnx.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/onnx.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/onnx.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/onnx.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/onnx.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/onnx.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/onnx.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/onnx.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/onnx.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/onnx.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/onnx.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/onnx.pyi stubs/torch/nn/utils/__init__.pyi 2 2 2 stubs/torch/ops.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/ops.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/ops.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/ops.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/ops.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/ops.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/ops.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/nn/utils/__init__.pyi 2 2 2 stubs/torch/ops.pyi stubs/torch/onnx.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/optim/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/optim/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/optim/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/optim/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/optim/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/optim/__init__.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/nn/utils/__init__.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/onnx.pyi 2 2 2 stubs/torch/optim/__init__.pyi stubs/torch/ops.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/profiler/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/profiler/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/profiler/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/profiler/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/profiler/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/profiler/__init__.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/nn/utils/__init__.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/onnx.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/ops.pyi 2 2 2 stubs/torch/profiler/__init__.pyi stubs/torch/optim/__init__.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/random/__init__.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/random/__init__.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/random/__init__.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/random/__init__.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/random/__init__.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/random/__init__.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/nn/utils/__init__.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/onnx.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/ops.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/optim/__init__.pyi 2 2 2 stubs/torch/random/__init__.pyi stubs/torch/profiler/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/utils/data.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/utils/data.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/utils/data.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/utils/data.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/utils/data.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/utils/data.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/nn/utils/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/onnx.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/ops.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/optim/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/profiler/__init__.pyi 2 2 2 stubs/torch/utils/data.pyi stubs/torch/random/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/numpy/__init__.pyi 2 2 3 stubs/torch/utils/model_zoo.pyi stubs/torch/linalg/__init__.pyi 2 2 3 stubs/torch/utils/model_zoo.pyi stubs/torch/nn/__init__.pyi 2 2 3 stubs/torch/utils/model_zoo.pyi stubs/torch/nn/init.pyi 2 2 3 stubs/torch/utils/model_zoo.pyi stubs/torch/sparse/__init__.pyi 2 2 3 stubs/torch/utils/model_zoo.pyi stubs/torch/__init__.pyi 2 2 3 stubs/torch/utils/model_zoo.pyi stubs/torch/autograd/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/autograd/profiler.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/cuda/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/fft/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/hub.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/nn/functional.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/nn/functional/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/nn/utils/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/onnx.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/ops.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/optim/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/profiler/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/random/__init__.pyi 2 2 2 stubs/torch/utils/model_zoo.pyi stubs/torch/utils/data.pyi 2 2 2 xformers/components/residual.py xformers/factory/block_factory.py 2 4 8 xformers/triton/__init__.py xformers/benchmarks/benchmark_triton_dropout.py 2 3 4 xformers/triton/__init__.py xformers/triton/dropout.py 2 3 8 xformers/components/attention/nystrom.py xformers/components/attention/base.py 2 5 5 xformers/components/attention/nystrom.py xformers/components/attention/core.py 2 5 9 xformers/components/attention/utils.py xformers/components/attention/nystrom.py 2 4 5 xformers/triton/k_layer_norm.py xformers/triton/layer_norm.py 2 2 5 experimental/ragged_inference/__init__.py experimental/__init__.py 1 1 1 experimental/ragged_inference/garbage_pad_ragged_acts.py experimental/__init__.py 1 1 1 experimental/ragged_inference/garbage_pad_ragged_acts.py experimental/ragged_inference/__init__.py 1 1 1 experimental/ragged_inference/seq_kv_cache.py experimental/__init__.py 1 1 1 experimental/ragged_inference/seq_kv_cache.py experimental/ragged_inference/__init__.py 1 1 1 experimental/ragged_inference/seq_kv_cache.py experimental/ragged_inference/garbage_pad_ragged_acts.py 1 1 1 experimental/ragged_inference/triton_v2_matmul.py experimental/__init__.py 1 1 1 experimental/ragged_inference/triton_v2_matmul.py experimental/ragged_inference/__init__.py 1 1 1 experimental/ragged_inference/triton_v2_matmul.py experimental/ragged_inference/garbage_pad_ragged_acts.py 1 1 1 experimental/ragged_inference/triton_v2_matmul.py experimental/ragged_inference/seq_kv_cache.py 1 1 1 experimental/ragged_inference/triton_v2_qk_dotprod.py experimental/__init__.py 1 1 1 experimental/ragged_inference/triton_v2_qk_dotprod.py experimental/ragged_inference/__init__.py 1 1 1 experimental/ragged_inference/triton_v2_qk_dotprod.py experimental/ragged_inference/garbage_pad_ragged_acts.py 1 1 1 experimental/ragged_inference/triton_v2_qk_dotprod.py experimental/ragged_inference/seq_kv_cache.py 1 1 1 experimental/ragged_inference/triton_v2_qk_dotprod.py experimental/ragged_inference/triton_v2_matmul.py 1 1 1 experimental/ragged_inference/triton_v2_ragged_qk_dotprod.py experimental/__init__.py 1 1 1 experimental/ragged_inference/triton_v2_ragged_qk_dotprod.py experimental/ragged_inference/__init__.py 1 1 1 experimental/ragged_inference/triton_v2_ragged_qk_dotprod.py experimental/ragged_inference/garbage_pad_ragged_acts.py 1 1 1 experimental/ragged_inference/triton_v2_ragged_qk_dotprod.py experimental/ragged_inference/seq_kv_cache.py 1 1 1 experimental/ragged_inference/triton_v2_ragged_qk_dotprod.py experimental/ragged_inference/triton_v2_matmul.py 1 1 1 experimental/ragged_inference/triton_v2_ragged_qk_dotprod.py experimental/ragged_inference/triton_v2_qk_dotprod.py 1 1 1 experimental/setup.py experimental/__init__.py 1 1 1 experimental/setup.py experimental/ragged_inference/__init__.py 1 1 1 experimental/setup.py experimental/ragged_inference/garbage_pad_ragged_acts.py 1 1 1 experimental/setup.py experimental/ragged_inference/seq_kv_cache.py 1 1 1 experimental/setup.py experimental/ragged_inference/triton_v2_matmul.py 1 1 1 experimental/setup.py experimental/ragged_inference/triton_v2_qk_dotprod.py 1 1 1 experimental/setup.py experimental/ragged_inference/triton_v2_ragged_qk_dotprod.py 1 1 1 xformers/components/attention/compositional.py xformers/components/__init__.py 1 1 7 xformers/components/attention/compositional.py xformers/components/attention/attention_mask.py 1 1 2 xformers/components/attention/compositional.py xformers/components/attention/base.py 1 1 5 xformers/components/attention/favor.py xformers/components/__init__.py 1 5 7 xformers/components/attention/favor.py xformers/components/attention/attention_mask.py 1 5 2 xformers/components/attention/favor.py xformers/components/attention/compositional.py 1 5 1 xformers/components/multi_head_dispatch.py xformers/components/__init__.py 1 4 7 xformers/components/multi_head_dispatch.py xformers/components/attention/attention_mask.py 1 4 2 xformers/components/multi_head_dispatch.py xformers/components/attention/compositional.py 1 4 1 xformers/components/multi_head_dispatch.py xformers/components/attention/favor.py 1 4 5 xformers/components/in_proj_container.py xformers/components/multi_head_dispatch.py 1 2 4 xformers/factory/model_factory.py xformers/components/__init__.py 1 5 7 xformers/__init__.py xformers/components/__init__.py 1 8 7 xformers/__init__.py xformers/components/attention/attention_mask.py 1 8 2 xformers/__init__.py xformers/components/attention/base.py 1 8 5 xformers/factory/block_factory.py xformers/components/multi_head_dispatch.py 1 8 4 xformers/benchmarks/benchmark_triton_dropout.py xformers/components/__init__.py 1 4 7 xformers/benchmarks/utils.py xformers/triton/k_dropout.py 1 1 5 xformers/benchmarks/utils.py xformers/benchmarks/benchmark_triton_dropout.py 1 1 4 xformers/triton/dropout.py xformers/benchmarks/utils.py 1 8 1 xformers/triton/dropout.py xformers/components/__init__.py 1 8 7 xformers/triton/k_activations.py xformers/triton/k_dropout.py 1 2 5 xformers/triton/k_activations.py xformers/benchmarks/benchmark_triton_dropout.py 1 2 4 xformers/triton/k_activations.py xformers/benchmarks/utils.py 1 2 1 xformers/benchmarks/benchmark_triton_stride_sum.py xformers/components/__init__.py 1 1 7 xformers/benchmarks/benchmark_triton_stride_sum.py xformers/benchmarks/benchmark_triton_dropout.py 1 1 4 xformers/benchmarks/benchmark_triton_stride_sum.py xformers/triton/dropout.py 1 1 8 xformers/triton/k_fused_matmul_bw.py xformers/components/__init__.py 1 3 7 xformers/triton/k_fused_matmul_bw.py xformers/benchmarks/benchmark_triton_stride_sum.py 1 3 1 xformers/triton/k_fused_matmul_bw.py xformers/triton/k_dropout.py 1 3 5 xformers/triton/k_sum.py xformers/components/__init__.py 1 1 7 xformers/triton/k_sum.py xformers/benchmarks/benchmark_triton_dropout.py 1 1 4 xformers/triton/k_sum.py xformers/triton/dropout.py 1 1 8 xformers/triton/k_sum.py xformers/benchmarks/benchmark_triton_stride_sum.py 1 1 1 xformers/triton/k_sum.py xformers/triton/k_fused_matmul_bw.py 1 1 3 xformers/triton/sum_strided.py xformers/components/__init__.py 1 1 7 xformers/triton/sum_strided.py xformers/benchmarks/benchmark_triton_dropout.py 1 1 4 xformers/triton/sum_strided.py xformers/triton/dropout.py 1 1 8 xformers/triton/sum_strided.py xformers/benchmarks/benchmark_triton_stride_sum.py 1 1 1 xformers/triton/sum_strided.py xformers/triton/k_fused_matmul_bw.py 1 1 3 xformers/triton/sum_strided.py xformers/triton/k_sum.py 1 1 1 xformers/triton/softmax.py xformers/triton/k_dropout.py 1 7 5 xformers/triton/softmax.py xformers/triton/dropout.py 1 7 8 xformers/triton/softmax.py xformers/components/multi_head_dispatch.py 1 7 4 xformers/triton/softmax.py xformers/factory/block_factory.py 1 7 8 xformers/components/attention/blocksparse.py xformers/components/attention/base.py 1 9 5 xformers/components/attention/blocksparse.py xformers/components/multi_head_dispatch.py 1 9 4 xformers/components/attention/blocksparse.py xformers/__init__.py 1 9 8 xformers/components/attention/global_tokens.py xformers/components/multi_head_dispatch.py 1 3 4 xformers/components/attention/global_tokens.py xformers/components/__init__.py 1 3 7 xformers/components/attention/global_tokens.py xformers/components/attention/attention_mask.py 1 3 2 xformers/components/attention/global_tokens.py xformers/__init__.py 1 3 8 xformers/components/attention/lambda_layer.py xformers/components/attention/base.py 1 2 5 xformers/components/attention/lambda_layer.py xformers/components/multi_head_dispatch.py 1 2 4 xformers/components/attention/linformer.py xformers/components/attention/base.py 1 2 5 xformers/components/attention/linformer.py xformers/components/multi_head_dispatch.py 1 2 4 xformers/components/attention/local.py xformers/components/attention/base.py 1 3 5 xformers/components/attention/local.py xformers/components/multi_head_dispatch.py 1 3 4 xformers/components/attention/local.py xformers/components/attention/blocksparse.py 1 3 9 xformers/components/attention/local.py xformers/components/attention/global_tokens.py 1 3 3 xformers/components/attention/local.py xformers/components/attention/lambda_layer.py 1 3 2 xformers/components/attention/local.py xformers/components/attention/linformer.py 1 3 2 xformers/components/attention/local.py xformers/components/__init__.py 1 3 7 xformers/components/attention/local.py xformers/factory/model_factory.py 1 3 5 xformers/components/attention/local.py xformers/factory/block_factory.py 1 3 8 xformers/components/attention/local.py xformers/components/reversible.py 1 3 3 xformers/components/attention/random.py xformers/components/multi_head_dispatch.py 1 3 4 xformers/components/attention/random.py xformers/components/attention/blocksparse.py 1 3 9 xformers/components/attention/random.py xformers/components/attention/lambda_layer.py 1 3 2 xformers/components/attention/random.py xformers/components/attention/linformer.py 1 3 2 xformers/components/attention/random.py xformers/components/__init__.py 1 3 7 xformers/components/attention/random.py xformers/components/attention/attention_mask.py 1 3 2 xformers/components/attention/random.py xformers/__init__.py 1 3 8 xformers/components/attention/random.py xformers/components/reversible.py 1 3 3 xformers/components/positional_embedding/rotary.py xformers/components/attention/base.py 1 2 5 xformers/components/positional_embedding/rotary.py xformers/components/attention/blocksparse.py 1 2 9 xformers/components/positional_embedding/rotary.py xformers/components/attention/global_tokens.py 1 2 3 xformers/components/positional_embedding/rotary.py xformers/components/attention/lambda_layer.py 1 2 2 xformers/components/positional_embedding/rotary.py xformers/components/attention/linformer.py 1 2 2 xformers/components/positional_embedding/rotary.py xformers/components/attention/local.py 1 2 3 xformers/components/positional_embedding/rotary.py xformers/components/attention/random.py 1 2 3 xformers/components/positional_embedding/rotary.py xformers/factory/block_factory.py 1 2 8 xformers/components/positional_embedding/rotary.py xformers/triton/softmax.py 1 2 7 xformers/benchmarks/LRA/code/__init__.py xformers/benchmarks/LRA/__init__.py 1 1 1 xformers/benchmarks/LRA/run_with_submitit.py xformers/benchmarks/LRA/__init__.py 1 1 1 xformers/benchmarks/LRA/run_with_submitit.py xformers/benchmarks/LRA/code/__init__.py 1 1 1 xformers/benchmarks/__init__.py xformers/benchmarks/LRA/__init__.py 1 1 1 xformers/benchmarks/__init__.py xformers/benchmarks/LRA/code/__init__.py 1 1 1 xformers/benchmarks/__init__.py xformers/benchmarks/LRA/run_with_submitit.py 1 1 1 xformers/test.py xformers/benchmarks/LRA/__init__.py 1 1 1 xformers/test.py xformers/benchmarks/LRA/code/__init__.py 1 1 1 xformers/test.py xformers/benchmarks/LRA/run_with_submitit.py 1 1 1 xformers/test.py xformers/benchmarks/__init__.py 1 1 1 xformers/factory/hydra_helper.py xformers/components/__init__.py 1 2 7 xformers/factory/hydra_helper.py xformers/factory/model_factory.py 1 2 5 xformers/factory/hydra_helper.py xformers/factory/block_factory.py 1 2 8 xformers/factory/hydra_helper.py xformers/components/attention/local.py 1 2 3 xformers/ops.py xformers/components/attention/_sputnik_sparse.py 1 1 2 xformers/sparse/__init__.py xformers/components/attention/_sputnik_sparse.py 1 1 2 xformers/sparse/__init__.py xformers/ops.py 1 1 1 xformers/sparse/_csr_ops.py xformers/components/attention/_sputnik_sparse.py 1 1 2 xformers/sparse/_csr_ops.py xformers/ops.py 1 1 1 xformers/sparse/_csr_ops.py xformers/sparse/__init__.py 1 1 1 xformers/sparse/csr_tensor.py xformers/components/attention/_sputnik_sparse.py 1 1 2 xformers/sparse/csr_tensor.py xformers/ops.py 1 1 1 xformers/sparse/csr_tensor.py xformers/sparse/__init__.py 1 1 1 xformers/sparse/csr_tensor.py xformers/sparse/_csr_ops.py 1 1 1 xformers/sparse/utils.py xformers/components/attention/_sputnik_sparse.py 1 1 2 xformers/sparse/utils.py xformers/ops.py 1 1 1 xformers/sparse/utils.py xformers/sparse/__init__.py 1 1 1 xformers/sparse/utils.py xformers/sparse/_csr_ops.py 1 1 1 xformers/sparse/utils.py xformers/sparse/csr_tensor.py 1 1 1 xformers/benchmarks/LRA/code/model_wrapper.py xformers/components/__init__.py 1 1 7 xformers/benchmarks/LRA/code/model_wrapper.py xformers/factory/model_factory.py 1 1 5 xformers/benchmarks/LRA/code/model_wrapper.py xformers/factory/block_factory.py 1 1 8 xformers/benchmarks/LRA/code/model_wrapper.py xformers/components/attention/local.py 1 1 3 xformers/benchmarks/LRA/code/model_wrapper.py xformers/factory/hydra_helper.py 1 1 2 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/components/__init__.py 1 2 7 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/factory/model_factory.py 1 2 5 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/factory/block_factory.py 1 2 8 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/components/attention/local.py 1 2 3 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/factory/hydra_helper.py 1 2 2 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/benchmarks/LRA/code/model_wrapper.py 1 2 1 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/triton/k_dropout.py 1 2 5 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/benchmarks/benchmark_triton_dropout.py 1 2 4 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/triton/dropout.py 1 2 8 xformers/benchmarks/benchmark_pytorch_transformer.py xformers/triton/k_fused_matmul_bw.py 1 2 3 xformers/triton/fused_linear_layer.py xformers/triton/k_activations.py 1 3 2 xformers/triton/fused_linear_layer.py xformers/triton/k_dropout.py 1 3 5 xformers/triton/fused_linear_layer.py xformers/benchmarks/benchmark_triton_dropout.py 1 3 4 xformers/triton/fused_linear_layer.py xformers/benchmarks/benchmark_pytorch_transformer.py 1 3 2 xformers/components/attention/core.py xformers/components/attention/lambda_layer.py 1 9 2 xformers/components/attention/core.py xformers/components/attention/linformer.py 1 9 2 xformers/components/attention/core.py xformers/components/__init__.py 1 9 7 xformers/components/attention/core.py xformers/components/attention/attention_mask.py 1 9 2 xformers/components/attention/core.py xformers/components/attention/_sputnik_sparse.py 1 9 2 xformers/components/attention/core.py xformers/components/reversible.py 1 9 3 xformers/components/attention/core.py xformers/components/attention/local.py 1 9 3 xformers/components/attention/fourier_mix.py xformers/components/attention/blocksparse.py 1 1 9 xformers/components/attention/fourier_mix.py xformers/components/attention/global_tokens.py 1 1 3 xformers/components/attention/fourier_mix.py xformers/components/attention/lambda_layer.py 1 1 2 xformers/components/attention/fourier_mix.py xformers/components/attention/linformer.py 1 1 2 xformers/components/attention/fourier_mix.py xformers/components/attention/core.py 1 1 9 xformers/components/attention/scaled_dot_product.py xformers/components/attention/lambda_layer.py 1 7 2 xformers/components/attention/scaled_dot_product.py xformers/components/attention/linformer.py 1 7 2 xformers/components/attention/scaled_dot_product.py xformers/components/attention/fourier_mix.py 1 7 1 xformers/components/attention/scaled_dot_product.py xformers/components/__init__.py 1 7 7 xformers/components/attention/scaled_dot_product.py xformers/components/attention/attention_mask.py 1 7 2 xformers/components/attention/scaled_dot_product.py xformers/__init__.py 1 7 8 xformers/components/attention/scaled_dot_product.py xformers/components/attention/random.py 1 7 3 xformers/components/attention/scaled_dot_product.py xformers/components/attention/favor.py 1 7 5 xformers/components/attention/scaled_dot_product.py xformers/components/attention/_sputnik_sparse.py 1 7 2 xformers/components/attention/__init__.py xformers/components/__init__.py 1 1 7 xformers/components/attention/__init__.py xformers/components/attention/attention_mask.py 1 1 2 xformers/components/attention/__init__.py xformers/components/attention/base.py 1 1 5 xformers/components/attention/__init__.py xformers/__init__.py 1 1 8 xformers/components/attention/__init__.py xformers/components/attention/global_tokens.py 1 1 3 xformers/components/attention/__init__.py xformers/components/attention/random.py 1 1 3 xformers/components/attention/__init__.py xformers/components/attention/core.py 1 1 9 xformers/components/attention/__init__.py xformers/components/attention/scaled_dot_product.py 1 1 7 stubs/numpy/__init__.pyi xformers/components/reversible.py 1 3 3 stubs/numpy/__init__.pyi xformers/components/attention/local.py 1 3 3 stubs/numpy/__init__.pyi xformers/components/attention/random.py 1 3 3 stubs/numpy/__init__.pyi xformers/components/attention/core.py 1 3 9 stubs/torch/linalg/__init__.pyi xformers/components/reversible.py 1 3 3 stubs/torch/linalg/__init__.pyi xformers/components/attention/local.py 1 3 3 stubs/torch/linalg/__init__.pyi xformers/components/attention/random.py 1 3 3 stubs/torch/linalg/__init__.pyi xformers/components/attention/core.py 1 3 9 stubs/torch/nn/__init__.pyi xformers/components/reversible.py 1 3 3 stubs/torch/nn/__init__.pyi xformers/components/attention/local.py 1 3 3 stubs/torch/nn/__init__.pyi xformers/components/attention/random.py 1 3 3 stubs/torch/nn/__init__.pyi xformers/components/attention/core.py 1 3 9 stubs/torch/nn/init.pyi xformers/components/reversible.py 1 3 3 stubs/torch/nn/init.pyi xformers/components/attention/local.py 1 3 3 stubs/torch/nn/init.pyi xformers/components/attention/random.py 1 3 3 stubs/torch/nn/init.pyi xformers/components/attention/core.py 1 3 9 stubs/torch/sparse/__init__.pyi xformers/components/reversible.py 1 3 3 stubs/torch/sparse/__init__.pyi xformers/components/attention/local.py 1 3 3 stubs/torch/sparse/__init__.pyi xformers/components/attention/random.py 1 3 3 stubs/torch/sparse/__init__.pyi xformers/components/attention/core.py 1 3 9 stubs/torch/__init__.pyi xformers/components/reversible.py 1 3 3 stubs/torch/__init__.pyi xformers/components/attention/local.py 1 3 3 stubs/torch/__init__.pyi xformers/components/attention/random.py 1 3 3 stubs/torch/__init__.pyi xformers/components/attention/core.py 1 3 9 stubs/torch/autograd/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/autograd/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/autograd/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/autograd/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/autograd/profiler.pyi xformers/components/reversible.py 1 2 3 stubs/torch/autograd/profiler.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/autograd/profiler.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/autograd/profiler.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/cuda/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/cuda/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/cuda/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/cuda/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/fft/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/fft/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/fft/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/fft/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/hub.pyi xformers/components/reversible.py 1 2 3 stubs/torch/hub.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/hub.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/hub.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/nn/functional.pyi xformers/components/reversible.py 1 2 3 stubs/torch/nn/functional.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/nn/functional.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/nn/functional.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/nn/functional/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/nn/functional/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/nn/functional/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/nn/functional/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/nn/utils/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/nn/utils/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/nn/utils/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/nn/utils/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/onnx.pyi xformers/components/reversible.py 1 2 3 stubs/torch/onnx.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/onnx.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/onnx.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/ops.pyi xformers/components/reversible.py 1 2 3 stubs/torch/ops.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/ops.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/ops.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/optim/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/optim/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/optim/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/optim/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/profiler/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/profiler/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/profiler/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/profiler/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/random/__init__.pyi xformers/components/reversible.py 1 2 3 stubs/torch/random/__init__.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/random/__init__.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/random/__init__.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/utils/data.pyi xformers/components/reversible.py 1 2 3 stubs/torch/utils/data.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/utils/data.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/utils/data.pyi xformers/components/attention/core.py 1 2 9 stubs/torch/utils/model_zoo.pyi xformers/components/reversible.py 1 2 3 stubs/torch/utils/model_zoo.pyi xformers/components/attention/local.py 1 2 3 stubs/torch/utils/model_zoo.pyi xformers/components/attention/random.py 1 2 3 stubs/torch/utils/model_zoo.pyi xformers/components/attention/core.py 1 2 9 xformers/components/residual.py xformers/__init__.py 1 4 8 xformers/components/residual.py xformers/components/attention/blocksparse.py 1 4 9 xformers/components/residual.py xformers/components/attention/core.py 1 4 9 xformers/components/residual.py xformers/components/__init__.py 1 4 7 xformers/triton/__init__.py xformers/__init__.py 1 3 8 xformers/triton/__init__.py xformers/components/attention/blocksparse.py 1 3 9 xformers/triton/__init__.py xformers/components/attention/core.py 1 3 9 xformers/triton/__init__.py xformers/components/residual.py 1 3 4 xformers/triton/__init__.py xformers/triton/k_dropout.py 1 3 5 xformers/triton/__init__.py xformers/triton/k_fused_matmul_bw.py 1 3 3 xformers/triton/__init__.py xformers/benchmarks/benchmark_pytorch_transformer.py 1 3 2 xformers/triton/__init__.py xformers/triton/fused_linear_layer.py 1 3 3 xformers/components/attention/nystrom.py xformers/components/attention/favor.py 1 5 5 xformers/components/attention/nystrom.py xformers/components/attention/_sputnik_sparse.py 1 5 2 xformers/components/attention/nystrom.py xformers/components/attention/blocksparse.py 1 5 9 xformers/components/feedforward/fused_mlp.py xformers/triton/k_dropout.py 1 1 5 xformers/components/feedforward/fused_mlp.py xformers/benchmarks/benchmark_triton_dropout.py 1 1 4 xformers/components/feedforward/fused_mlp.py xformers/triton/dropout.py 1 1 8 xformers/components/feedforward/fused_mlp.py xformers/triton/k_fused_matmul_bw.py 1 1 3 xformers/components/feedforward/fused_mlp.py xformers/benchmarks/benchmark_pytorch_transformer.py 1 1 2 xformers/components/feedforward/fused_mlp.py xformers/triton/fused_linear_layer.py 1 1 3 xformers/components/feedforward/fused_mlp.py xformers/triton/__init__.py 1 1 3 xformers/components/attention/utils.py xformers/components/attention/_sputnik_sparse.py 1 4 2 xformers/components/attention/utils.py xformers/components/attention/core.py 1 4 9 xformers/components/attention/utils.py xformers/components/attention/scaled_dot_product.py 1 4 7 xformers/components/attention/utils.py xformers/components/attention/base.py 1 4 5 xformers/triton/layer_norm.py xformers/components/reversible.py 1 5 3 xformers/triton/layer_norm.py xformers/components/attention/local.py 1 5 3 xformers/triton/layer_norm.py xformers/components/attention/random.py 1 5 3 xformers/triton/layer_norm.py xformers/components/attention/core.py 1 5 9 xformers/triton/layer_norm.py stubs/numpy/__init__.pyi 1 5 3 xformers/triton/layer_norm.py stubs/torch/linalg/__init__.pyi 1 5 3 xformers/triton/layer_norm.py stubs/torch/nn/__init__.pyi 1 5 3 xformers/triton/layer_norm.py stubs/torch/nn/init.pyi 1 5 3 xformers/triton/layer_norm.py stubs/torch/sparse/__init__.pyi 1 5 3 xformers/triton/layer_norm.py stubs/torch/__init__.pyi 1 5 3 xformers/triton/layer_norm.py stubs/torch/autograd/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/autograd/profiler.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/cuda/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/fft/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/hub.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/nn/functional.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/nn/functional/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/nn/utils/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/onnx.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/ops.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/optim/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/profiler/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/random/__init__.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/utils/data.pyi 1 5 2 xformers/triton/layer_norm.py stubs/torch/utils/model_zoo.pyi 1 5 2 xformers/triton/layer_norm.py xformers/factory/block_factory.py 1 5 8 xformers/triton/layer_norm.py xformers/components/residual.py 1 5 4 setup.py packaging/pkg_helpers.bash 1 1 1 xformers/benchmarks/benchmark_triton_fused_linear.py xformers/triton/k_fused_matmul_bw.py 1 1 3 xformers/benchmarks/benchmark_triton_fused_linear.py xformers/triton/fused_linear_layer.py 1 1 3 xformers/triton/k_fused_matmul_fw.py xformers/triton/k_fused_matmul_bw.py 1 1 3 xformers/triton/k_fused_matmul_fw.py xformers/triton/fused_linear_layer.py 1 1 3 xformers/triton/k_fused_matmul_fw.py xformers/benchmarks/benchmark_triton_fused_linear.py 1 1 1 xformers/components/positional_embedding/__init__.py stubs/torch/__init__.pyi 1 2 3 xformers/components/positional_embedding/__init__.py xformers/components/multi_head_dispatch.py 1 2 4 xformers/components/positional_embedding/__init__.py xformers/factory/block_factory.py 1 2 8 xformers/components/positional_embedding/__init__.py xformers/triton/softmax.py 1 2 7 xformers/components/positional_embedding/__init__.py xformers/components/positional_embedding/rotary.py 1 2 2 xformers/components/positional_embedding/sine.py stubs/torch/__init__.pyi 1 1 3 xformers/components/positional_embedding/sine.py xformers/components/positional_embedding/__init__.py 1 1 2 xformers/benchmarks/benchmark_revnet.py xformers/components/reversible.py 1 1 3 xformers/triton/k_layer_norm.py xformers/factory/block_factory.py 1 2 8 xformers/triton/k_layer_norm.py xformers/components/residual.py 1 2 4 xformers/triton/k_softmax.py xformers/triton/k_dropout.py 1 1 5 xformers/triton/k_softmax.py xformers/triton/dropout.py 1 1 8 xformers/triton/k_softmax.py xformers/triton/softmax.py 1 1 7 xformers/benchmarks/LRA/run_tasks.py xformers/components/multi_head_dispatch.py 1 1 4 xformers/benchmarks/LRA/run_tasks.py xformers/factory/block_factory.py 1 1 8 xformers/benchmarks/LRA/run_tasks.py xformers/triton/softmax.py 1 1 7 xformers/benchmarks/LRA/run_tasks.py xformers/components/positional_embedding/rotary.py 1 1 2 xformers/benchmarks/LRA/run_tasks.py xformers/components/positional_embedding/__init__.py 1 1 2 xformers/benchmarks/benchmark_encoder.py xformers/components/attention/blocksparse.py 1 1 9