README.md
pyproject.toml
hf_attention_normalizers/__init__.py
hf_attention_normalizers/backends.py
hf_attention_normalizers/paged.py
hf_attention_normalizers/varlen.py
hf_attention_normalizers.egg-info/PKG-INFO
hf_attention_normalizers.egg-info/SOURCES.txt
hf_attention_normalizers.egg-info/dependency_links.txt
hf_attention_normalizers.egg-info/requires.txt
hf_attention_normalizers.egg-info/top_level.txt
hf_attention_normalizers/kernels/__init__.py
hf_attention_normalizers/kernels/block_csr_softmax1_triton.py
hf_attention_normalizers/kernels/paged_softmax1_hopper.py
hf_attention_normalizers/kernels/paged_sparse_entmax_hopper.py
hf_attention_normalizers/kernels/softmax1_hopper.py
hf_attention_normalizers/kernels/sparse_entmax_hopper.py
hf_attention_normalizers/kernels/sparse_entmax_triton.py
hf_attention_normalizers/kernels/varlen_softmax1_hopper.py
hf_attention_normalizers/kernels/varlen_sparse_entmax_hopper.py
hf_attention_normalizers/vutils/__init__.py
hf_attention_normalizers/vutils/entmax.py
hf_attention_normalizers/vutils/softmax1.py
hf_attention_normalizers/vutils/sparsemax.py
tests/test_flash_attention.py
tests/test_flex_attention.py
tests/test_hopper_attention.py
tests/test_paged_attention.py
tests/test_sparse_entmax_triton.py
tests/test_varlen_attention.py