微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Propel 可微分编程机器学习框架

程序名称:Propel

授权协议: Apache 2.0

操作系统: 跨平台

开发语言: TypeScript

Propel 介绍

Propel 是一个使用 JavaScript 做可微分编程的机器学习框架,既能在 Node 中使用,又能在浏览器中使用。在这两种环境中,Propel
都能够使用 GPU 硬件进行加速计算。在浏览器中,它能通过 deeplearn.js 使用
WebGL ;在 Node 上,它能使用 TensorFlow 的 C
API

Node 中使用:

npm install propel
import { grad } from "propel";

浏览器中使用:

<script src="https://unpkg.com/[email protected]"></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))

Propel 官网

http://propelml.org/

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐