Skip to main content
Ctrl+K
cuEquivariance - Home cuEquivariance - Home

cuEquivariance

cuEquivariance - Home cuEquivariance - Home

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
    • EquivariantTensorProduct
    • Operation
    • 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_tensor_product
    • tensor_product
    • 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
  • Tutorials

Tutorials#

  • Groups and Representations
  • Data Layouts
  • Equivariant Tensor Product
  • Segmented Tensor Product
  • Beta features with unstable API

previous

NVIDIA cuEquivariance Documentation

next

Groups and Representations
NVIDIA NVIDIA
Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2024-2025, NVIDIA Corporation & Affiliates.