LICENSE
README.md
pyproject.toml
src/kpnn2/__init__.py
src/kpnn2/_adjacency_spec.py
src/kpnn2/_align.py
src/kpnn2/_attributions.py
src/kpnn2/_constraint.py
src/kpnn2/_errors.py
src/kpnn2/_gather.py
src/kpnn2/_generator.py
src/kpnn2/_identity.py
src/kpnn2/_layout.py
src/kpnn2/_mask_tensor.py
src/kpnn2/_masked_linear.py
src/kpnn2/_packed_linear.py
src/kpnn2/_packed_multihead_attention.py
src/kpnn2/_parse.py
src/kpnn2/_parse_adjacency.py
src/kpnn2/_serialize.py
src/kpnn2/_spec.py
src/kpnn2.egg-info/PKG-INFO
src/kpnn2.egg-info/SOURCES.txt
src/kpnn2.egg-info/dependency_links.txt
src/kpnn2.egg-info/requires.txt
src/kpnn2.egg-info/top_level.txt
src/kpnn2/_aggregation/__init__.py
src/kpnn2/_aggregation/_bind.py
src/kpnn2/_aggregation/_dispatch.py
src/kpnn2/_aggregation/_registry.py
src/kpnn2/_aggregation/_methods/__init__.py
src/kpnn2/_aggregation/_methods/_rauter_mangano_2026.py