Registry / ai-ml / cuequivariance-ops-cu12

cuequivariance-ops-cu12

library0.10.0pypypi✓ verified 87d ago

GPU-accelerated operations for equivariant primitives, part of the cuequivariance ecosystem. Version 0.10.0, actively maintained. Provides CUDA 12.x kernels for spherical tensor operations.

ai-ml data

pip install cuequivariance-ops-cu12

package ↗homepage ↗docs ↗

INSTALL

IMPORT

SIG · CUEQUIVARIANCE-OPS

cuequivariance-ops-cu12

ai-mlpythonv0.10.0

harness data pending

Install & Compatibility

Where this runs

No compatibility data collected yet for this library.

Code

Verified usage

Verified import paths — ran on the pinned version, not inferred.

cuequivariance_ops

✓ import cuequivariance_ops as ceops

✗ import cuequivariance_ops_cu12

The package is installed as cuequivariance-ops-cu12 but the import name is cuequivariance_ops. Trying to import 'cuequivariance_ops_cu12' will fail with ModuleNotFoundError.

Basic import and environment check for cuequivariance-ops.

import torch import cuequivariance_ops as ceops device = 'cuda' if torch.cuda.is_available() else 'cpu' # Example: spherical harmonic transform irreps_out, irreps_in = "1x0o", "1x1e" # adapted to actual ops # Check available ops print(dir(ceops))

Debug

Known issues

breakingCUDA 12.x required: This package provides kernels compiled for CUDA 12.x only. Older CUDA versions will cause runtime errors.

fix

Install cu11 variant instead or update to CUDA 12.x.

affects: 0.10.0

gotchaImport name mismatch: Install as cuequivariance-ops-cu12 but import as cuequivariance_ops.

fix

Use 'import cuequivariance_ops as ceops' in code.

affects: all

gotchaRequires Python >=3.10. Using older Python will cause installation failure.

fix

Upgrade Python to 3.10 or higher.

affects: 0.10.0

Errors

Common errors & fixes

ModuleNotFoundError: No module named 'cuequivariance_ops_cu12'

Incorrect import: the package installs as cuequivariance-ops-cu12 but the importable module is cuequivariance_ops.

fix

Use 'import cuequivariance_ops' instead.

RuntimeError: CUDA error: no kernel image is available for execution on the device

Trying to run on a GPU that requires a different CUDA version (e.g., CUDA 11.x).

fix

Install the correct wheel for your CUDA version (e.g., cuequivariance-ops-cu11).

Upgrade

Version history

0.10.0latest on PyPI · released Apr 22, 2026

Audit

Dependencies

cuequivariancerequiredCore package for equivariance primitives

torchrequiredPyTorch tensor operations

Agent activity

13 hits · last 30 days

node

OpenAI (training)

Resources

packagehttps://pypi.org/project/cuequivariance-ops-cu12/ ↗

homepagehttps://developer.nvidia.com/cuequivariance ↗

docshttps://docs.nvidia.com/cuda/cuequivariance/ ↗