google-deepmind/penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

语言：Python · GitHub 星标：1,899 · Fork：69

话题：fine-tuning · interpretability · jax · neural-networks · visualization

在 GitHub 上查看原仓库