cuequivariance-ops-jax
Introduction
cuequivariance_ops_jax provides CUDA kernels for the
cuEquivariance project's JAX
components. As such, it contains JAX 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 pip install cuequivariance-ops-jax-cu12 or
pip install cuequivariance-ops-jax-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.
Usage
You can import the library from python:
import cuequivariance_ops_jax
Support and Feedback
Please contact the cuEquivariance developers for any issues you might encounter.
6 maintainers
Metadata
6 maintainers
Release files for cuequivariance-ops-jax-cu13 0.13.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distributions (wheels)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| cuequivariance_ops_jax_cu13-0.13.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl | Python 3 | none | Linux glibc 2.24+ x86-64, Linux glibc 2.28+ x86-64 | Details |
| cuequivariance_ops_jax_cu13-0.13.0-py3-none-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl | Python 3 | none | Linux glibc 2.24+ ARM64, Linux glibc 2.28+ ARM64 | Details |
Total release size: 322.1 kB
Release files / cuequivariance_ops_jax_cu13-0.13.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
| Download URL | cuequivariance_ops_jax_cu13-0.13.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl |
|---|---|
| Size | 167.5 kB |
| Tags | Linux glibc 2.24+ x86-64 Linux glibc 2.28+ x86-64 Python 3 |
|
SHA-256 checksum How to use checksums |
26c2b77bee6086accd648db44c4980f6feee5c73bba8cb283b42ce732408fa82
|
|
BLAKE2b-256 checksum How to use checksums |
f1cc325b0b79f4a09e6836945ba566f5dd91422d3d5e6027b9e6dddf423e8d82
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.9.6
|
Release files / cuequivariance_ops_jax_cu13-0.13.0-py3-none-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl
| Download URL | cuequivariance_ops_jax_cu13-0.13.0-py3-none-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl |
|---|---|
| Size | 154.6 kB |
| Tags | Linux glibc 2.24+ ARM64 Linux glibc 2.28+ ARM64 Python 3 |
|
SHA-256 checksum How to use checksums |
da48bc39c80c225f3b1b58d9dd6d532a1de5c0721235157e0d6143b70ffc012c
|
|
BLAKE2b-256 checksum How to use checksums |
5789350b0753752c43e6870df4367eb86c45d6f2848e866d352732048f2d4f5a
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.9.6
|
