penzai: JAX research toolkit for building, editing, and visualizing neural nets
Comments ⌘ Read more