yatorho / trchime

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hello Trchime!

What's Trchime?

Tiny deep-learning framework powered by python3 and numpy

  • examples --complete code examples of Trchime
  • documentation --Trchime development documentation and usage documentation

Dependencies

Name Version
Python 3.7.0+
Numpy 1.19.0+

Start

Compute $ y=kx +b $​ gradient with trchime, and output the gradient of $ x $​ .

The whole process is completed by graph calculation and automatic differentiation.

from trchime import Variable, Constant

k = Constant([[2, 3], [1, 1]])
b = Constant([[7], [3]])
x = Variable([[0], [0]])

y = k @ x + b
z = y.sum()
z.backward()

print(x.grad.data)

result:

[[3.],
 [4.]]

What can trchime do?

  • Graph Computing
  • Auto Gradient
  • Algebraic system
  • Gradient Descent
  • Neural Network API
  • Convolutional Neural Network

Connect us

Author yatorho
QQ 3227068950
E-mail 3227068950@qq.com

About

License:Apache License 2.0


Languages

Language:Python 100.0%