Penzai: JAX research toolkit for building, editing, and visualizing neural netsgithub.com/google-deepmind261 pointsmccoyb2 years ago