Something between Tinygrad and Micrograd
Project description
Robingrad
Something between Tinygrad and Micrograd.
Installation
To install the current release,
pip install robingrad==0.0.8
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
Release history Release notifications | RSS feed
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.8.tar.gz
(10.4 kB
view hashes)
Built Distribution
robingrad-0.0.8-py3-none-any.whl
(10.0 kB
view hashes)
Close
Hashes for robingrad-0.0.8-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3a1ca59be73e2428e80099ddb43a04c062c341891a2c5246928f3c7bc6d486fe |
|
MD5 | a9f1e403e9ebf82834e5ec3d17cb516a |
|
BLAKE2b-256 | 92671581c9323f4e70d5166b0ff3aff931eeedcb5eebe3d0cb8b25dfeeaf1534 |