LICENSE
README.md
pyproject.toml
src/scree/__init__.py
src/scree/_core.py
src/scree.egg-info/PKG-INFO
src/scree.egg-info/SOURCES.txt
src/scree.egg-info/dependency_links.txt
src/scree.egg-info/requires.txt
src/scree.egg-info/top_level.txt
src/scree/bridges/__init__.py
src/scree/bridges/_dlpack.py
src/scree/bridges/_hf_padded.py
src/scree/bridges/_torch_nested.py
src/scree/kernels/__init__.py
src/scree/kernels/reference/__init__.py
src/scree/kernels/reference/varlen_attention.py
src/scree/kernels/reference/varlen_layernorm.py
src/scree/kernels/reference/varlen_rmsnorm.py
src/scree/kernels/reference/varlen_softmax.py
src/scree/kernels/triton/__init__.py
src/scree/kernels/triton/_autograd.py
src/scree/kernels/triton/_backward.py
src/scree/kernels/triton/varlen_attention.py
src/scree/kernels/triton/varlen_layernorm.py
src/scree/kernels/triton/varlen_rmsnorm.py
tests/test_array_api.py
tests/test_autograd.py
tests/test_bridges.py
tests/test_core.py
tests/test_jax.py
tests/test_mlx.py
tests/test_properties.py
tests/test_varlen_kernels.py