CustomGetAttrKey# class penzai.core.tree_util.CustomGetAttrKey[source]# Bases: object Subclass-friendly variant of jax.tree_util.GetAttrKey. Methods __init__(name) Attributes name