Mathematics of Machine Learning
Solving Optimization Problems with JAX
A comprhensive tutorial explaining different beginner optimization algorithms. I derived all the algorithms from scratch and showed how to implement them in python using JAX. The article was shared by many members of JAX team on twitter.