Getting Started

tensorclouds is a Python package for Euclidean Equivariant Neural Networks in Flax/JAX. It provides tools and modules for building and training E3NNs, and a zoo of pre-built models.

• Getting Started
• API Reference
  • tensorclouds
    • Submodules
      • tensorclouds.data
        • Submodules
      • tensorclouds.loss
        • Submodules
      • tensorclouds.nn
        • Submodules
        • Classes
        • Package Contents
      • tensorclouds.random
        • Submodules
      • tensorclouds.tensorcloud
        • Classes
        • Module Contents
      • tensorclouds.transport
        • Submodules
      • tensorclouds.utils
        • Functions
        • Module Contents
    • Classes
    • Package Contents
      • TensorCloud
        • TensorCloud.irreps_array
        • TensorCloud.mask_irreps_array
        • TensorCloud.coord
        • TensorCloud.mask_coord
        • TensorCloud.label
        • TensorCloud.err
        • TensorCloud.irreps_array
        • TensorCloud.mask_irreps_array
        • TensorCloud.coord
        • TensorCloud.mask_coord
        • TensorCloud.label
        • TensorCloud.annotations
        • TensorCloud.shape
        • TensorCloud.empty()
        • TensorCloud.ones_like()
        • TensorCloud.zeros_like()
        • TensorCloud.irreps
        • TensorCloud.mask
        • TensorCloud.replace()
        • TensorCloud.concatenate()
        • TensorCloud.zeros()
        • TensorCloud.stack()
        • TensorCloud.split()
        • TensorCloud.centralize()
        • TensorCloud.plot()
        • TensorCloud.dot()
        • TensorCloud.norm()

• Index