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.0.tar.gz (789.4 kB view details)

Uploaded Source

File details

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

File metadata

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

File hashes

Hashes for cudagrad-0.2.0.tar.gz
Algorithm Hash digest
SHA256 716102ffe35414ffd156c61ca65603e467a8017fff5ba966303eded650ab975c
MD5 27328d7313de2b25f4d509638f873404
BLAKE2b-256 daef8911dff270afe055499a636b95e752a8ccb0cd0c8c74374c7865a09ca429

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