SymJAX:符号CPU/GPU/TPU编程
SymJAX: symbolic CPU/GPU/TPU programming
摘要:
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分析而来,可能存在错误,请以论文正文为准。