orbax icon indicating copy to clipboard operation
orbax copied to clipboard
verified publisher icon google

[JAX] Fix confusion between a pytree and a PyTreeDef in orbax.

Open copybara-service[bot] opened this issue 2 years ago • 0 comments

[JAX] Fix confusion between a pytree and a PyTreeDef in orbax.

This causes pytype errors under an upcoming JAX change.

copybara-service[bot] avatar Mar 04 '23 14:03 copybara-service[bot]