Toy implementations of the following deep learning optimization algorithms from scratch in JAX -
- Stochastic Gradient Descent
- Momentum
- Adagrad
- RMSProp
- Adam
| Name | Name | Last commit date | ||
|---|---|---|---|---|
Toy implementations of the following deep learning optimization algorithms from scratch in JAX -