Repository info

penzai

• Summary: a summary of penzai
• Description: A JAX research toolkit for building, editing, and visualizing neural networks.
• Stars: 1803
• Number of forks: 66