Add square-symmetry augmentations and sampling fixes#8

JiamingPan wants to merge 1 commit into

mainnkern/cosmo_diffusion:mainfrom

codex/augmentationsnkern/cosmo_diffusion:codex/augmentationsCopy head branch name to clipboard