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
    • Segmented Polynomials
    • Segmented Tensor Product
  • cuequivariance
    • Operation
    • SegmentedOperand
    • SegmentedTensorProduct
    • SegmentedPolynomial
    • Rep
    • Irrep
    • SO3
    • O3
    • SU2
    • clebsch_gordan
    • Irreps
    • IrrepsLayout
    • IrrepsAndLayout
    • 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
    • indexed_linear
  • cuequivariance-torch
    • SegmentedPolynomial
    • EquivariantTensorProduct
    • TensorProduct
    • ChannelWiseTensorProduct
    • FullyConnectedTensorProduct
    • Linear
    • SymmetricContraction
    • TransposeIrrepsLayout
    • SphericalHarmonics
    • Rotation
    • Inversion
    • encode_rotation_angle
    • vector_to_euler_angles
    • triangle_attention
    • triangle_multiplicative_update
    • BatchNorm
    • FullyConnectedTensorProductConv
  • Release Notes

Python Module Index

c
 
c
- cuequivariance
    cuequivariance.group_theory.descriptors
cuequivariance_jax
cuequivariance_torch
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.