vix.ing · top · new · best · stats · spec

SymJAX: symbolic CPU/GPU/TPU programming

2020/05/21 by Randall Balestriero, Balestriero, Randall · 1 citation
Computer Science · #Computer Vision and Pattern Recognition (cs.CV) #Evolutionary Algorithms and Applications #FOS: Computer and information sciences #Mathematical Software (cs.MS) #Music and Audio Processing #Neural Networks and Applications #cs.CV #cs.MS

paper · pdf · doi:10.48550/arxiv.2005.10635

arxiv created 2020/05/21 · openalex publication_date 2020/05/21 · arxiv updated 2020/05/22 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

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.

Cited by

Related