Skip to main content

Something between Tinygrad and Micrograd

Project description

Robingrad


Maintenance stability-wip

Something between Tinygrad and Micrograd.

Installation

To install the current release,

pip install robingrad==0.0.6

From source

git clone https://github.com/marcosalvalaggio/robingrad.git
cd robingrad
./build.sh

Examples

  • In the examples folder, you can find examples of models trained using the Robingrad library.
  • A declaration example of an MLP net using Robingrad:
from robingrad import Tensor, draw_dot
import robingrad.nn as nn

class RobinNet:
    def __init__(self):
        self.l1 = nn.Linear(5,16)
        self.l2 = nn.Linear(16,1)
    def __call__(self, x):
        x = self.l1(x)
        x = x.relu()
        x = self.l2(x)
        return x
        
model = RobinNet()
res = model(X_train[0].reshape((1,5)))
draw_dot(res)

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

robingrad-0.0.6.tar.gz (9.6 kB view hashes)

Uploaded Source

Built Distribution

robingrad-0.0.6-py3-none-any.whl (9.1 kB view hashes)

Uploaded Python 3

Supported by

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