Skip to main content

A tensor-valued autograd engine for Python

Project description

cudagrad

CUDA C++ strided float tensor automatic differentiation engine with Python bindings

Install

Available on PyPI as a source distribution, requires cmake and optionally nvcc if available:

pip install cudagrad

Examples

The following examples were written purely in Python using only cudagrad.Tensor for learning:

OR

0.52 seconds (59.5% faster than torch)

/benchmarks/_cudagrad/or.py

XOR

4.5 seconds (39.2% faster than torch)

/benchmarks/_cudagrad/xor.py

MOONS

14.25 seconds (5.8% slower than torch)

/benchmarks/_cudagrad/moons.py

Project details


Download files

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

Source Distribution

cudagrad-0.2.3.tar.gz (4.8 MB view details)

Uploaded Source

Built Distribution

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

cudagrad-0.2.3-cp311-cp311-macosx_14_0_arm64.whl (308.7 kB view details)

Uploaded CPython 3.11macOS 14.0+ ARM64

File details

Details for the file cudagrad-0.2.3.tar.gz.

File metadata

  • Download URL: cudagrad-0.2.3.tar.gz
  • Upload date:
  • Size: 4.8 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.11.9

File hashes

Hashes for cudagrad-0.2.3.tar.gz
Algorithm Hash digest
SHA256 78c673137e79667d28a3747d9fb23b152fec4629cb1a78dc08c35497cdc1cefc
MD5 ca69c23fb43aa710ef0b57b44e23d9fc
BLAKE2b-256 578e165bc5d3530bdd39b833d2398c4ac56c473a579d1514f38530ca11f4821e

See more details on using hashes here.

File details

Details for the file cudagrad-0.2.3-cp311-cp311-macosx_14_0_arm64.whl.

File metadata

File hashes

Hashes for cudagrad-0.2.3-cp311-cp311-macosx_14_0_arm64.whl
Algorithm Hash digest
SHA256 d26cdb03827d358a84c3f66290bc9fa561c9271eb1f1cedd9cdb13d3e1567ee7
MD5 f2a6d2feb0298c92eecd0f728f841498
BLAKE2b-256 3712ffdfc12896f3c1b12c354821a22e680fa3cff8468bc16b0f286c1a8f9ea7

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page