Support custom PyTree metadata. Standardize naming of the "custom metadata" field (user-supplied metadata) as `custom_metadata`.#1461

Merged

copybara-service[bot] merged 1 commit intomainfrom test_704424560

Jan 21, 2025

+230-86

Commits on Jan 21, 2025

Support custom PyTree metadata. Standardize naming of the "custom metadata" field (user-supplied metadata) as custom_metadata.

cpgaffney1
authored and
Orbax Authors
committed