This repository was archived by the owner on Feb 7, 2026. It is now read-only.

This convenience method is copied from flax.struct.

PiperOrigin-RevId: 621690005