Skip to main content

raxis-cuda12

The CUDA 12 backend of raxis, a readable re-implementation of the core of PyTorch (ATen + autograd) in C++20.

pip install "raxis[cuda12]"   # or pip install raxis-cuda12

installs raxis together with its GPU kernels, caching allocator and cuBLAS bindings; import raxis loads them, and tensors move to the GPU with .cuda() / .to("cuda") as in PyTorch. It needs an NVIDIA driver (the CUDA runtime is linked statically, cuBLAS comes from NVIDIA's wheel).

raxis-cuda12 runs on drivers r525 and newer and on GPUs from Volta (sm_70) on; raxis-cuda13 needs a driver r580 or newer and a GPU from Turing (sm_75) on. Install only one of them.

Metadata

Release files for raxis-cuda12 0.4.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Built distributions (wheels)

Table of built distributions (wheels) for raxis-cuda12 0.4.0
File Interpreter ABI Platform Actions
raxis_cuda12-0.4.0-py3-none-win_amd64.whl Python 3 none Windows x86-64 Details for raxis_cuda12-0.4.0-py3-none-win_amd64.whl
raxis_cuda12-0.4.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl Python 3 none Linux glibc 2.28+ x86-64, Linux glibc 2.24+ x86-64 Details for raxis_cuda12-0.4.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl

Total release size: 72.1 MB

Release files / raxis_cuda12-0.4.0-py3-none-win_amd64.whl

Download URL raxis_cuda12-0.4.0-py3-none-win_amd64.whl
Size 36.4 MB
Tags Python 3 Windows x86-64
SHA-256 checksum
How to use checksums
6463b0ac37a74923917d3e98ed4b45d3650d812e8e17c42a59148b32ef7e9a99
BLAKE2b-256 checksum
How to use checksums
a0198383071a040a89ab1b43841504b2be919a7027c157ca2c491e59b63f4690
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/7.0.0 CPython/3.13.16

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Nov 1, 2026.

Transparency log

Release files / raxis_cuda12-0.4.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl

Download URL raxis_cuda12-0.4.0-py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl
Size 35.7 MB
Tags Linux glibc 2.24+ x86-64 Linux glibc 2.28+ x86-64 Python 3
SHA-256 checksum
How to use checksums
ed4fd68de6c5ca7deac06ebabc1d8db883a78651ba32c7631541bfd1f1599889
BLAKE2b-256 checksum
How to use checksums
40b5dde935758c5fc3f13e781da749d126993addc58fd32b093fb48e0a4ca1b2
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/7.0.0 CPython/3.13.16

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Nov 1, 2026.

Transparency log

Security for raxis-cuda12 0.4.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Full provenance

All 2 files in this release have been uploaded with provenance metadata.

Select from 2 files

Attestation for raxis_cuda12-0.4.0-py3-none-win_amd64.whl

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Nov 1, 2026.

Transparency log

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

0.4.0 This release

2 release files Full provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page