cuequivariance-ops-torch

Introduction

cuequivariance_ops_torch provides CUDA kernels for the cuEquivariance project's PyTorch components. As such, it contains pytorch bindings to optimized kernels that cuEquivariance's operations map down to. In general, we advice that you access those kernels through cuEquivariance, but you may also find them useful on their own.

Installation

Please install using either pip install cuequivariance-ops-torch-cu12 or pip install cuequivariance-ops-torch-cu13 (depending on the CUDA toolkit you wish to use).

Documentation

For detailed usage information of the kernels, please refer to the doc-strings in their respective modules. For higher-level documentations, refer to cuEquivariance.

Pairformer inference primitives

The Torch package also exposes lower-level primitives for integrating Pairformer inference graphs:

attention_pair_bias_mask_from_normalized projects an already-normalized pair representation and applies an arbitrary Boolean [B,U,V] mask. The implementation is shape-generic; BF16 D=256, H=16, and square sequence lengths 384 and 512 are the performance-validated configurations. FP16 and BF16 inputs use native low-precision tensor-core products with FP32 accumulation; FP32 inputs retain the higher-accuracy TF32x3 path.
pairformer_combined_triangle_multiplication combines incoming and outgoing triangle multiplication after caller-provided non-affine LayerNorm with epsilon 1e-5. Its fused direction join applies the affine middle LayerNorm with epsilon 0.03. This operation intentionally fails closed outside BF16 inference with B=1, D=256, sequence length 384 or 512, and an SM100 GPU. The caller guarantees that the second normalized carrier is the separately materialized spatial transpose of the first; the hot path validates metadata and distinct storage but does not compare tensor contents. It returns the base update before any caller-owned peri-LayerNorm, output mask, dropout, or residual addition; model-level dropout must be disabled for inference. Use pack_pairformer_grouped_projection to convert grouped projection weights to the expected payload/gate layout.

Both operations are registered as torch.library custom operators with fake implementations for torch.compile tracing. See their Python docstrings for the complete input contracts.

Usage

You can import the library from python:

import cuequivariance_ops_torch

Kernels are primarily exposed as torch.nn.Module, but also provide a lower-level interface as torch.library operators. Generally, the module is responsible for proper input transformation and initialization, and the operator execute the kernel. This allows you to export models using this operations using torch.export, and running inference on them using TensorRT.

Support and Feedback

Please contact the cuEquivariance developers for any issues you might encounter.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (460.2 kB view details)

Uploaded Aug 7, 2026 CPython 3.14manylinux: glibc 2.27+ x86-64manylinux: glibc 2.28+ x86-64

cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl (455.6 kB view details)

Uploaded Aug 7, 2026 CPython 3.14manylinux: glibc 2.26+ ARM64manylinux: glibc 2.28+ ARM64

cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (460.3 kB view details)

Uploaded Aug 7, 2026 CPython 3.13manylinux: glibc 2.27+ x86-64manylinux: glibc 2.28+ x86-64

cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl (455.4 kB view details)

Uploaded Aug 7, 2026 CPython 3.13manylinux: glibc 2.26+ ARM64manylinux: glibc 2.28+ ARM64

cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (460.4 kB view details)

Uploaded Aug 7, 2026 CPython 3.12manylinux: glibc 2.27+ x86-64manylinux: glibc 2.28+ x86-64

cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl (455.4 kB view details)

Uploaded Aug 7, 2026 CPython 3.12manylinux: glibc 2.26+ ARM64manylinux: glibc 2.28+ ARM64

cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (460.5 kB view details)

Uploaded Aug 7, 2026 CPython 3.11manylinux: glibc 2.27+ x86-64manylinux: glibc 2.28+ x86-64

cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl (456.2 kB view details)

Uploaded Aug 7, 2026 CPython 3.11manylinux: glibc 2.26+ ARM64manylinux: glibc 2.28+ ARM64

cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (460.4 kB view details)

Uploaded Aug 7, 2026 CPython 3.10manylinux: glibc 2.27+ x86-64manylinux: glibc 2.28+ x86-64

cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl (456.1 kB view details)

Uploaded Aug 7, 2026 CPython 3.10manylinux: glibc 2.26+ ARM64manylinux: glibc 2.28+ ARM64

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Upload date: Aug 7, 2026
Size: 460.2 kB
Tags: CPython 3.14, manylinux: glibc 2.27+ x86-64, manylinux: glibc 2.28+ x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Algorithm	Hash digest
SHA256	`c1219828f61e93ab72dadfba617227148a27c360f30f960310d347767f45d30d`
MD5	`788c716e800ca4fd768708cf245f916e`
BLAKE2b-256	`662fe762d2a929d8e22d96b937ab94ccd39dd2e20b2cf9dbd7a4fc29feb90bd1`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Upload date: Aug 7, 2026
Size: 455.6 kB
Tags: CPython 3.14, manylinux: glibc 2.26+ ARM64, manylinux: glibc 2.28+ ARM64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp314-cp314-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Algorithm	Hash digest
SHA256	`4dbb5e5a54ff8ba6d43eb8de382c3380d4ce8d14b2a309cb12a6b16ae743e4e0`
MD5	`ce0fd4b3693ac95b1dca4598cc7a43b8`
BLAKE2b-256	`04190e4bf1bd8c5a9c6d3968d6b02dd0878967c1cc8bc887e1af4593e1239d69`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Upload date: Aug 7, 2026
Size: 460.3 kB
Tags: CPython 3.13, manylinux: glibc 2.27+ x86-64, manylinux: glibc 2.28+ x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Algorithm	Hash digest
SHA256	`589d92f136437b295f65819c1d2af11ecbed1db19558ed9ee482291d1c837aae`
MD5	`397a1e9793dee4646f8f03bf9d7521b8`
BLAKE2b-256	`b3a142c6e78a654c28ba03932e7c7bfb2918ae5828488aae30322ced83379a19`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Upload date: Aug 7, 2026
Size: 455.4 kB
Tags: CPython 3.13, manylinux: glibc 2.26+ ARM64, manylinux: glibc 2.28+ ARM64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp313-cp313-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Algorithm	Hash digest
SHA256	`57e8c2f100a4a0e97c7de3a40b7c64add9b36615eebfd9218fec752f1746d6f8`
MD5	`3542147f5a8f4bf4cc656b4c140fb7b1`
BLAKE2b-256	`fbbbb1c6ef2a8a474ebeb4a43ccc58852f2ee2db47552dbd40d4015c911588c1`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Upload date: Aug 7, 2026
Size: 460.4 kB
Tags: CPython 3.12, manylinux: glibc 2.27+ x86-64, manylinux: glibc 2.28+ x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Algorithm	Hash digest
SHA256	`57fa7d057ea3cfbc8c8f85ce8d24e6ff12cc49073759bf8ff789608d6cd7752d`
MD5	`fe4f8bef830b1954636037b5c19b5184`
BLAKE2b-256	`19f25ba170c6195e95e42ab9e2fade98a70a078a3663af14440a67b457208041`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Upload date: Aug 7, 2026
Size: 455.4 kB
Tags: CPython 3.12, manylinux: glibc 2.26+ ARM64, manylinux: glibc 2.28+ ARM64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp312-cp312-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Algorithm	Hash digest
SHA256	`0f51fa260d31ba5c9cfa78cfced4ccc5ce2ef94c22ec4f190397ab4063f8e5c8`
MD5	`ec42ff5fe3cc88144bc040224d281b2b`
BLAKE2b-256	`7e3392233dbca77d7aed37a5c92f416e2393898bd7b4fedfe02b4b0a09ed2646`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Upload date: Aug 7, 2026
Size: 460.5 kB
Tags: CPython 3.11, manylinux: glibc 2.27+ x86-64, manylinux: glibc 2.28+ x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Algorithm	Hash digest
SHA256	`bb5d055cf62c359440193bc88a5d056376372734b154798db9341ebafbce6eab`
MD5	`b4f0de03bdc1aeab98b99efa5d1c8e94`
BLAKE2b-256	`c7f56488a244db9ca54b414d6b98d18a98edd9f66911f05de7c8db1f39da351b`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Upload date: Aug 7, 2026
Size: 456.2 kB
Tags: CPython 3.11, manylinux: glibc 2.26+ ARM64, manylinux: glibc 2.28+ ARM64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp311-cp311-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Algorithm	Hash digest
SHA256	`d74bd3a55e82243ac79004c9547ad6892557d88303f51136e814bbf8e29f2c62`
MD5	`29a1c4af90545b0337314f8366bae4cf`
BLAKE2b-256	`89e9f0aad4178e04d2b1326ae3eefe891a44ecdf644f5f2b80a1c68098b0d0cc`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Upload date: Aug 7, 2026
Size: 460.4 kB
Tags: CPython 3.10, manylinux: glibc 2.27+ x86-64, manylinux: glibc 2.28+ x86-64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl
Algorithm	Hash digest
SHA256	`31a24c09b63cf484b7ac04b753a9a19135e1f7a4a3144cbfbc05a2dbc3761e9e`
MD5	`189ec6183a4f52b4193801c624e085ce`
BLAKE2b-256	`9e763a24f5c0496c919177374948620edc3453756b66aa09f14173ed7ea14725`

See more details on using hashes here.

File details

Details for the file cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl.

File metadata

Download URL: cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Upload date: Aug 7, 2026
Size: 456.1 kB
Tags: CPython 3.10, manylinux: glibc 2.26+ ARM64, manylinux: glibc 2.28+ ARM64
Uploaded using Trusted Publishing? No
Uploaded via: twine/6.2.0 CPython/3.9.6

File hashes

Hashes for cuequivariance_ops_torch_cu13-0.11.1-cp310-cp310-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl
Algorithm	Hash digest
SHA256	`eb96a9b20502ca89925c1cb55472984e2eff0e8ee30ec8f36b77c16ddd5cad08`
MD5	`e089d9316e68b6f93171c2cde3f87cdd`
BLAKE2b-256	`54f5adf3d93db173a27a2ff991a19ae66141eb799b8caf01b062842514361451`

See more details on using hashes here.

Release history Release notifications | RSS feed

This release

0.11.1

Aug 7, 2026

0.11.0

Aug 4, 2026

0.10.0

Apr 22, 2026

0.9.1

Mar 16, 2026

0.9.0

Feb 17, 2026

0.8.1

Jan 9, 2026

0.8.0

Dec 8, 2025

0.7.0

Oct 13, 2025

cuequivariance-ops-torch-cu13 0.11.1

cuequivariance-ops-torch

Introduction

Installation

Documentation

Pairformer inference primitives

Usage

Support and Feedback

Project links

Key dates

5 maintainers

Credits

License

Requires

Provides Extra

Classifiers

Download files

Source Distributions

Built Distributions

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

File details

File metadata

File hashes

Release history Release notifications | RSS feed