# Execute the register_operator functions
import jaxpm._src.base_ops
import jaxpm._src.painting_ops
import jaxpm.ops
import jaxpm.painting
from jaxpm._src.spmd_config import SPMDConfig

__all__ = ['SPMDConfig']