.gitignore
.pre-commit-config.yaml
LICENSE.md
MANIFEST.in
README.md
pyproject.toml
pytest.ini
quick_install.sh
requirements.txt
setup.cfg
setup.py
.github/workflows/python-publish.yml
.github/workflows/python_packages.yml
examples/run_cogvideox.py
examples/run_flux.py
examples/run_mochi.py
src/para_attn/__init__.py
src/para_attn/_version.py
src/para_attn/config.py
src/para_attn/para_attn_interface.py
src/para_attn/primitives.py
src/para_attn.egg-info/PKG-INFO
src/para_attn.egg-info/SOURCES.txt
src/para_attn.egg-info/dependency_links.txt
src/para_attn.egg-info/requires.txt
src/para_attn.egg-info/top_level.txt
src/para_attn/context_parallel/__init__.py
src/para_attn/context_parallel/diffusers_adapters/__init__.py
src/para_attn/context_parallel/diffusers_adapters/cogvideox.py
src/para_attn/context_parallel/diffusers_adapters/flux.py
src/para_attn/context_parallel/diffusers_adapters/mochi.py
src/para_attn/ops/__init__.py
tests/__init__.py
tests/test_interface.py
tests/context_parallel/__init__.py
tests/context_parallel/test_diffusers_adapters.py
tests/ops/__init__.py
tests/ops/test_ops.py