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

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.1.tar.gz (4.7 MB view details)

Uploaded Source

File details

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

File metadata

  • Download URL: cudagrad-0.2.1.tar.gz
  • Upload date:
  • Size: 4.7 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.1.tar.gz
Algorithm Hash digest
SHA256 eee84673fbffaf03952b11020ce8264a0074c61c2987e89b484deebe59bc10de
MD5 fa9c0d528bcfaa4305c79b42055defae
BLAKE2b-256 8dea18a8ccfe3c5e14222c0de177ae152e5d60424280b79da6a5e3d68d72ba16

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