Week 1# Introduction to JAX One application : gradient based optimization Root finding methods Homework 1 - Solution