penzai icon indicating copy to clipboard operation
penzai copied to clipboard
verified publisher icon google-deepmind

Use `jax.tree_util.treedef_is_leaf` for readability

Open findmyway opened this issue 3 months ago • 0 comments

findmyway avatar Aug 19 '25 07:08 findmyway