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 0
License
johnviljoen/jax2sympy
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published