mindscience.e3nn
=========================

nn
----------------

.. mscnautosummary::
    :toctree: e3nn
    :nosignatures:

    mindscience.e3nn.nn.Activation
    mindscience.e3nn.nn.BatchNorm
    mindscience.e3nn.nn.FullyConnectedNet
    mindscience.e3nn.nn.Gate
    mindscience.e3nn.nn.NormActivation
    mindscience.e3nn.nn.OneHot
    mindscience.e3nn.nn.Scatter
    mindscience.e3nn.nn.soft_one_hot_linspace
    mindscience.e3nn.nn.soft_unit_step
    mindscience.e3nn.nn.SoftOneHotLinspace

o3
----------------

.. mscnautosummary::
    :toctree: e3nn
    :nosignatures:

    mindscience.e3nn.o3.angles_to_matrix
    mindscience.e3nn.o3.angles_to_xyz
    mindscience.e3nn.o3.change_basis_real_to_complex
    mindscience.e3nn.o3.compose_angles
    mindscience.e3nn.o3.ElementwiseTensorProduct
    mindscience.e3nn.o3.FullTensorProduct
    mindscience.e3nn.o3.FullyConnectedTensorProduct
    mindscience.e3nn.o3.identity_angles
    mindscience.e3nn.o3.Irrep
    mindscience.e3nn.o3.Irreps
    mindscience.e3nn.o3.Linear
    mindscience.e3nn.o3.LinearBias
    mindscience.e3nn.o3.matrix_to_angles
    mindscience.e3nn.o3.matrix_x
    mindscience.e3nn.o3.matrix_y
    mindscience.e3nn.o3.matrix_z
    mindscience.e3nn.o3.Norm
    mindscience.e3nn.o3.rand_angles
    mindscience.e3nn.o3.so3_generators
    mindscience.e3nn.o3.spherical_harmonics
    mindscience.e3nn.o3.SphericalHarmonics
    mindscience.e3nn.o3.su2_generators
    mindscience.e3nn.o3.TensorProduct
    mindscience.e3nn.o3.TensorSquare
    mindscience.e3nn.o3.wigner_3j
    mindscience.e3nn.o3.wigner_D
    mindscience.e3nn.o3.xyz_to_angles

so2_conv
----------------

.. mscnautosummary::
    :toctree: e3nn
    :nosignatures:

    mindscience.e3nn.so2_conv.SO3Rotation
    mindscience.e3nn.so2_conv.SO2Convolution
    mindscience.e3nn.so2_conv.init_edge_rot_mat

utils
----------------

.. mscnautosummary::
    :toctree: e3nn
    :nosignatures:

    mindscience.e3nn.utils.Ncon
    mindscience.e3nn.utils.radius
    mindscience.e3nn.utils.radius_full
    mindscience.e3nn.utils.radius_graph
    mindscience.e3nn.utils.radius_graph_full