Propel是一个使用JavaScript做可微分编程的机器学习框架,既能在Node中使用,又能在浏览器中使用。在这两种环境中,Propel都能够使用GPU硬件进行加速计算。在浏览器中,它能通过 deeplearn.js 使用WebGL;在Node上,它能使用TensorFlow的 CAPI 。
Node中使用:
npm install propelimport { grad } from "propel";浏览器中使用:
<script src="https://unpkg.com/propel@3.0.0"></script>与TensorFlow不同的是,Propel有一个命令式的 autograd 风格的API。运行过程中会随着追踪计算图——一种通用的梯度函数提供反向传播的简洁接口。
import { grad, linspace, plot } from "propel";f = x => x.tanh();x = linspace(-4, 4, 200);plot(x, f(x), x, grad(f)(x), x, grad(grad(f))(x), x, grad(grad(grad(f)))(x), x, grad(grad(grad(grad(f))))(x))
评论