google-deepmind/penzai

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

Training & fine-tuningPython Apache-2.0 70created 2024-04-04archived

Stars

1,899

1-day growth

7-day growth

—

30-day growth

—

Star history

Tracked since 2026-11-13

#fine-tuning#interpretability#jax#neural-networks#visualization

Top contributors

Not collected yet (contributors job).

google-deepmind/penzai · AI Repo Stats