CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
atomic_gdn2/__init__.py
atomic_gdn2/config.py
atomic_gdn2/domain.py
atomic_gdn2/layer.py
atomic_gdn2/layout.py
atomic_gdn2/pipeline.py
atomic_gdn2/precision.py
atomic_gdn2/py.typed
atomic_gdn2/bwd/__init__.py
atomic_gdn2/bwd/b4.py
atomic_gdn2/bwd/mega.py
atomic_gdn2/fwd/__init__.py
atomic_gdn2/fwd/cd_slim.py
atomic_gdn2/fwd/scores.py
atomic_gdn2/fwd/solve.py
atomic_gdn2/reference/__init__.py
atomic_gdn2/reference/ref.py
atomic_gdn2/training/__init__.py
atomic_gdn2/training/canary.py
atomic_gdn2/training/data.py
atomic_gdn2/training/model.py
atomic_gdn2/training/train.py
atomic_ops/__init__.py
atomic_ops/configs.py
atomic_ops/fallback.py
atomic_ops/gdn2_bwd.py
atomic_ops/gdn2_fwd.py
atomic_ops/gdn2_pipeline.py
atomic_ops/py.typed
atomic_ops/reference.py
atomic_ops/utils.py
atomic_ops.egg-info/PKG-INFO
atomic_ops.egg-info/SOURCES.txt
atomic_ops.egg-info/dependency_links.txt
atomic_ops.egg-info/requires.txt
atomic_ops.egg-info/top_level.txt
tests/test_clip_config_plumbing.py
tests/test_configs.py
tests/test_gdn2_full_math_correctness.py
tests/test_imports.py
tests/test_pallas.py
tests/test_reference.py