GitHub repository: patrick-kidger/sympy2jax
Primary language: Python
Description: Turn SymPy expressions into trainable JAX expressions.