arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2005.10635cs.MScs.CV
收藏

SymJAX:符号CPU/GPU/TPU编程

SymJAX: symbolic CPU/GPU/TPU programming

Randall Balestriero

摘要:

SymJAX是JAX的符号编程版本,简化了图形输入/输出/更新,并为一般机器学习和深度学习应用程序提供了额外的功能。从用户的角度来看,SymJAX提供了一种la Theano体验,具有快速的图形优化/编译和广泛的硬件支持,以及类似Lasagne的深度学习功能。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑