robingrad

ロビングラード

TinygradとMicrogradの間の何か。

インストール

現在のリリースをインストールするには、

 pip install robingrad==0.1.0

ソースから

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

例

• 例フォルダーでは、 Robingrad Libraryを使用してトレーニングされたモデルの例を見つけることができます。
• Robingradを使用したMLPネットの宣言例：

 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 )