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 Tensor Product
    • Segmented Polynomial
    • PyTorch Examples
      • MACE Operations
      • Segmented Polynomial in PyTorch
    • JAX Examples
      • Segmented Polynomial in JAX
  • 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
    • triangle_multiplicative_update
    • triangle_attention
    • indexed_linear
  • cuequivariance-torch
    • SegmentedPolynomial
    • ChannelWiseTensorProduct
    • FullyConnectedTensorProduct
    • Linear
    • SymmetricContraction
    • TransposeIrrepsLayout
    • SphericalHarmonics
    • Rotation
    • Inversion
    • encode_rotation_angle
    • vector_to_euler_angles
    • triangle_attention
    • triangle_multiplicative_update
    • attention_pair_bias
    • BatchNorm
    • FullyConnectedTensorProductConv
  • Release Notes
  • Tutorials
  • JAX Examples

JAX Examples#

  • Segmented Polynomial in JAX

previous

Segmented Polynomial in PyTorch

next

Segmented Polynomial in JAX
NVIDIA NVIDIA
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2024-2026, NVIDIA Corporation & Affiliates.