cuEquivariance
cuEquivariance is an NVIDIA Python library designed to facilitate the construction of high-performance geometric neural networks using segmented polynomials and triangular operations. cuEquivariance provides a comprehensive API for describing segmented polynomials made out of segmented tensor products and optimized CUDA kernels for their execution. Additionally, cuEquivariance offers bindings for both PyTorch and JAX, ensuring broad compatibility and ease of integration.
Equivariance is the mathematical formalization of the concept of "respecting symmetries." Robust physical models exhibit equivariance with respect to rotations and translations in three-dimensional space. Artificial intelligence models that incorporate equivariance are often more data-efficient.
Documentation
Please refer to the project documentation for more information https://docs.nvidia.com/cuda/cuequivariance/.
Installation
# Choose the frontend you want to use
pip install cuequivariance-jax
pip install cuequivariance-torch
pip install cuequivariance # Installs only the core non-ML components
# CUDA kernels
pip install cuequivariance-ops-jax-cu12 # or -cu13
pip install cuequivariance-ops-torch-cu12 # or -cu13
License
All files hosted in this repository are subject to the Apache 2.0 license.
Disclaimer
cuEquivariance is in a Beta state. Beta products may not be fully functional, may contain errors or design flaws, and may be changed at any time without notice. We appreciate your feedback to improve and iterate on our Beta products.
7 maintainers
Metadata
7 maintainers
Release files for cuequivariance-jax 0.13.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| cuequivariance_jax-0.13.0.tar.gz | 214.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| cuequivariance_jax-0.13.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 420.9 kB
Release files / cuequivariance_jax-0.13.0.tar.gz
| Download URL | cuequivariance_jax-0.13.0.tar.gz |
|---|---|
| Size | 214.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
078a293813d95c7d79c38e2977df326300abd96daf4c4c09f9370080524a0326
|
|
BLAKE2b-256 checksum How to use checksums |
6fc7fb44f5c812c648bafa4f46c9aece756048ed8cd8bda41820bda153dea3e5
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
Release files / cuequivariance_jax-0.13.0-py3-none-any.whl
| Download URL | cuequivariance_jax-0.13.0-py3-none-any.whl |
|---|---|
| Size | 206.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
097d6376c7ea8b39b306a8e63eb0e7f331f5c256c27f72e285dbe0e37b62f6f1
|
|
BLAKE2b-256 checksum How to use checksums |
4d1e1e35363334e8a5a6fb55c1131c9a0b6048eeff4ef8641d1e19dd33ebb5c8
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.8
|
