"""Define useful layers for PyTorch."""

from .rotation import UnitQuaternion, SVDO
from .positional_encoding import FixedPositionalEncoding, RFF
from .product_keys import ProductKeyMemory
from .real_nvp import RealNVP
