Open Leoningel opened 1 year ago
Using this can make it trivial to implement the SGD: https://github.com/patrick-kidger/sympy2jax
Using this can make it trivial to implement the SGD: https://github.com/patrick-kidger/sympy2jax