sympy2jax

https://github.com/patrick-kidger/sympy2jax

sympy2jax converts SymPy expressions into trainable JAX expressions as Equinox modules, enabling...

Tokens:4,464
Snippets:39
Trust Score:9.7
License:Apache-2.0
Update:2 months ago
Tokens:
Raw