flax
596a6267 - Add JAX pytree key path API to tree_utils.

Commit

2 years ago

Add JAX pytree key path API to tree_utils. PiperOrigin-RevId: 511585331

References

Author

IvyZX

IvyZX

Committer

a-googler

a-googler

Parents

Loading