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.4.tar.gz (4.7 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.4-cp310-cp310-manylinux2014_x86_64.whl (441.5 kB view details)

Uploaded CPython 3.10

File details

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

File metadata

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

File hashes

Hashes for cudagrad-0.2.4.tar.gz
Algorithm Hash digest
SHA256 89e1288b3afc8470fb42756fc026010373b7f2fc51fc1df4aedda2ada77b72db
MD5 cb1a19dc6bcd2c7d22906e0c37cd85cb
BLAKE2b-256 4f848d53a574d55744cb7b6caf8c36e9eb8a89e7c9b5f693a09aef98d0860ed3

See more details on using hashes here.

File details

Details for the file cudagrad-0.2.4-cp310-cp310-manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for cudagrad-0.2.4-cp310-cp310-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 ad33e11fb7a4782b85bcf500dee640b2df2340f9ec847d5d96d6e1ae045d1c93
MD5 f4aa2ec2012d2ed183a9025bd2696948
BLAKE2b-256 7ce817d416e7a1f9eb172bd9082d05558f7bc15b642fd8dee7352fab170fe14a

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