Skip to content

Commit

Permalink
Add an example of jax.tree_util.register_static to the jax 101 pytree…
Browse files Browse the repository at this point in the history 
… docs.

PiperOrigin-RevId: 559896201
  • Loading branch information
jax authors committed Aug 24, 2023
1 parent d452eea commit ec95862
Show file tree
Hide file tree
Showing 2 changed files with 201 additions and 65 deletions.

0 comments on commit ec95862

Please sign in to comment.