Skip to content

Adopt jax.tree.* and rename backend.tree_util to backend.tree#231

Merged
pnkraemer merged 2 commits intomainfrom rename-tree-moduleJan 7, 2025

Commits

Commits on Jan 7, 2025