Penzai 0.2.1
New features:
- Added support for the Gemma 2 pretrained models.
Improvements:
- Adjusted colors for
TransformerBlock
andAttention
when rendered with Treescope. - Fixed an outdated link in the README.
TransformerBlock
and Attention
when rendered with Treescope.