Skip to content
Permalink

Comparing changes

Choose two branches to see what’s changed or to start a new pull request. If you need to, you can also or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: jax-ml/jax
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: 0eec4e7
Choose a base ref
...
head repository: jax-ml/jax
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: 28dc694
Choose a head ref
  • 1 commit
  • 5 files changed
  • 1 contributor

Commits on Feb 7, 2024

  1. Add jax.tree_util.tree_traverse 

    This provides the same API as tree.traverse from google-deepmind/tree
    @jakevdp
    jakevdp committed Feb 7, 2024
    Configuration menu
    Copy the full SHA
    28dc694 View commit details
    Browse the repository at this point in the history
Loading