Use jax.numpy consistently (iris_optimize.py) (#372)
Use jax.numpy consistently (iris_optimize.py) (#372)

* use_jax.numpy_consistently_iris_optimize

* Use jax.numpy consistently (iris_optimize.py)

* Use jax.numpy consistently (iris_optimize.py)

Co-authored-by: Dristi <adffd619eaf1b3df14b444e15e3a3422a41d26ed@Dristis-MacBook-Air.local>
Co-authored-by: Kevin P Murphy <511ae2a02bba6d592ebf47bddd29765615f30fdc@gmail.com>
probml/pyprobml,probml/pyprobml,probml/pyprobml,probml/pyprobml