LICENSE
README.md
pyproject.toml
src/colabfold_kernels/__init__.py
src/colabfold_kernels/contract.py
src/colabfold_kernels/device.py
src/colabfold_kernels/diffusion.py
src/colabfold_kernels/fused_ops.py
src/colabfold_kernels/grads.py
src/colabfold_kernels/mgpu_attn.py
src/colabfold_kernels/opm.py
src/colabfold_kernels/packed_attn.py
src/colabfold_kernels/transition.py
src/colabfold_kernels/tri_flash.py
src/colabfold_kernels/tri_mul.py
src/colabfold_kernels/volta.py
src/colabfold_kernels.egg-info/PKG-INFO
src/colabfold_kernels.egg-info/SOURCES.txt
src/colabfold_kernels.egg-info/dependency_links.txt
src/colabfold_kernels.egg-info/requires.txt
src/colabfold_kernels.egg-info/top_level.txt
tests/test_af2_fused_kernels.py
tests/test_diffusion_loader.py
tests/test_fused_ops_grad.py
tests/test_legacy_gdp.py
tests/test_ln_fusions.py
tests/test_ln_gdp_grad.py
tests/test_packed_attention_bounds.py
tests/test_pallas_vjp.py
tests/test_volta_vjp.py