Code

Add path_aware_map function#2371

Code

Merged

copybara-service[bot] merged 1 commit intogoogle:maingoogle/flax:mainfrom cgarciae:optax-partition-functioncgarciae/flax:optax-partition-functionCopy head branch name to clipboardAug 30, 2022

Commits

Commits on Aug 29, 2022

• add path_aware_map

  cgarciae

  committed
  acb7662

  Copy full SHA for acb7662