This repo translates jaxpr graphs into a SymPy elementwise expressions. This allows for symbolic analysis of the system. An example use case is finding sparsity pattern of jacobians and hessians of vector valued functions through symbolic differentiation, find example code under sparsity.py.
thomas-rkk/jax2sympy
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|