jax
fe683acd - Use np.eye instead of onp.eye in quickstart notebook.

Commit

6 years ago

Use np.eye instead of onp.eye in quickstart notebook. JAX supports np.eye, so no need to fall back to onp.

References

Author

Justin Lebar

Parents