sympy2jax
sympy2jax copied to clipboard
Rename jax.tree_map to jtu.tree_map
I also changed sympytorch to sympy2jax in the readme.
Closes: #14