18 projects
cuequivariance-torch
CUDA accelerated equivariant operations
cuequivariance-jax
CUDA accelerated equivariant operations
cuequivariance
CUDA accelerated equivariant operations
cuequivariance-ops-torch-cu13
cuequivariance-ops-torch - GPU Accelerated Torch Extensions for Equivariant Primitives
cuequivariance-ops-torch-cu12
cuequivariance-ops-torch - GPU Accelerated Torch Extensions for Equivariant Primitives
cuequivariance-ops-jax-cu13
cuequivariance-ops-jax - GPU Accelerated JAX Extensions for Equivariant Primitives
cuequivariance-ops-jax-cu12
cuequivariance-ops-jax - GPU Accelerated JAX Extensions for Equivariant Primitives
cuequivariance-ops-cu13
cuequivariance-ops - GPU Accelerated Extensions for Equivariant Primitives
cuequivariance-ops-cu12
cuequivariance-ops - GPU Accelerated Extensions for Equivariant Primitives
e3nn-jax
Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors.
e3nn
Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors.
cuequivariance-ops-torch-cu11
cuequivariance-ops-torch - GPU Accelerated Torch Extensions for Equivariant Primitives
cuequivariance-ops-cu11
cuequivariance-ops - GPU Accelerated Extensions for Equivariant Primitives
lie-learn
A python package that knows how to do various tricky computations related to Lie groups and manifolds (mainly the sphere S2 and rotation group SO3).
lie-nn
Tensor Products for Lie Groups.
rust-neighborlist
None
unique-with-tol
None
roundmantissa
Round mantissa of a float number
