/gradient_descent

A script that explains the gradient descent algorithm with JAX package.

Primary LanguageJupyter Notebook

gradient_descent

A script that explains the gradient descent algorithm with JAX package.