SymJAX: symbolic CPU/GPU/TPU programming

2020-05-21Code Available1· sign in to hype

Randall Balestriero

Code Available — Be the first to reproduce this paper.

Code

github.com/RandallBalestriero/SymJAX
OfficialIn paperjax★ 130

Abstract

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.

Tasks

BIG-bench Machine Learning CPU Deep Learning GPU

SymJAX: symbolic CPU/GPU/TPU programming

Code

Abstract

Tasks

Reproductions