robingrad

Robingrad

Something between Tinygrad and Micrograd.

Installation

To install the current release,

pip install robingrad==0.1.0

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)