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.
-
Notifications
You must be signed in to change notification settings - Fork 2
Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.
License
johnviljoen/jax2sympy
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published