mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-04-24 19:50:55 +00:00
8 lines
211 B
Python
8 lines
211 B
Python
# 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']
|