# SymJAX: symbolic CPU/GPU/TPU programming

arXiv: https://arxiv.org/abs/2005.10635 · HTML: https://mlsanity.com/paper/2005.10635/

## Key facts

- Cited by 1 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 21 May 2020.

## Built on by (most cited first, 1 of 1)

- [Spatial Transformer K-Means](https://mlsanity.com/paper/2202.07829/) (2022)

## Abstract (arXiv)

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.