README.md
pyproject.toml
src/group_discovery/distributions.py
src/group_discovery/geometry_2d.py
src/group_discovery/geometry_3d.py
src/group_discovery/logger.py
src/group_discovery/torch_logm.py
src/group_discovery/utils.py
src/group_discovery.egg-info/PKG-INFO
src/group_discovery.egg-info/SOURCES.txt
src/group_discovery.egg-info/dependency_links.txt
src/group_discovery.egg-info/requires.txt
src/group_discovery.egg-info/top_level.txt
src/group_discovery/canonicalization/model.py
src/group_discovery/data/objects.py
src/group_discovery/data/sampling_dataset.py
src/group_discovery/data/MNIST/dataset.py
src/group_discovery/data/MNIST/group_transform.py
src/group_discovery/data/MNIST/preprocess.py
src/group_discovery/diffusion/MNIST/model.py
src/group_discovery/diffusion/group_actions/model.py
src/group_discovery/flow_matching/angle_to_angle.py
src/group_discovery/flow_matching/matrix_to_matrix.py
src/group_discovery/flow_matching/MNIST/model.py
src/group_discovery/flow_matching/arrow/model.py
src/group_discovery/flow_matching/arrow/net.py
src/group_discovery/flow_matching/arrow/oracle.py