Penzai 0.1.5

New features:

• Added a gradient checkpointing combinator Checkpointed in penzai.experimental.v2.toolshed.gradient_checkpointing. This can reduce memory usage when taking gradients through a model.

Bug fixes and improvements:

• Fixed an issue where rendering logic would run out of order in some situations (in particular, using a treescope rendering inside an iframe inside another treescope rendering).

Documentation changes:

• Added documentation of gradient checkpointing on the "how-to" guide page.