Skip to main content
Back to top
Ctrl
+
K
cuEquivariance
Search
Ctrl
+
K
Search
Ctrl
+
K
cuEquivariance
Table of Contents
Tutorials
Groups and Representations
Data Layouts
Equivariant Tensor Product
Segmented Tensor Product
Beta features with unstable API
cuequivariance
Rep
Irrep
SO3
O3
SU2
clebsch_gordan
Irreps
IrrepsLayout
IrrepsAndLayout
SegmentedTensorProduct
Operation
SegmentedPolynomial
EquivariantPolynomial
Descriptors
spherical_harmonics
fully_connected_tensor_product
channelwise_tensor_product
elementwise_tensor_product
linear
symmetric_contraction
x_rotation
y_rotation
yxy_rotation
assume
cuequivariance-jax
RepArray
from_segments
as_irreps_array
concatenate
randn
equivariant_polynomial
segmented_polynomial
LayerNorm
spherical_harmonics
cuequivariance-torch
EquivariantTensorProduct
TensorProduct
ChannelWiseTensorProduct
FullyConnectedTensorProduct
Linear
SymmetricContraction
TransposeIrrepsLayout
SphericalHarmonics
Rotation
Inversion
encode_rotation_angle
vector_to_euler_angles
BatchNorm
FullyConnectedTensorProductConv
Release Notes
equivariant_tensor_product
equivariant_tensor_product
#
