论文标题
Symjax:符号CPU/GPU/TPU编程
SymJAX: symbolic CPU/GPU/TPU programming
论文作者
论文摘要
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.