论文标题

Symjax:符号CPU/GPU/TPU编程

SymJAX: symbolic CPU/GPU/TPU programming

论文作者

Balestriero, Randall

论文摘要

Symjax是JAX的符号编程版本,简化了图形输入/输出/更新,并为通用机器学习和深度学习应用程序提供了其他功能。从用户角度来看,Symjax提供了La Theano的体验,并具有快速的图形优化/编译和广泛的硬件支持,以及类似宽面条的深度学习功能。

SymJAX is a symbolic programming version of JAX simplifying graph input/output/updates and providing additional functionalities for general machine learning and deep learning applications. From an user perspective SymJAX provides a la Theano experience with fast graph optimization/compilation and broad hardware support, along with Lasagne-like deep learning functionalities.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源