flax
83e74665 - Replace deprecated API `jax.tree_map`

Commit

2 years ago

Replace deprecated API `jax.tree_map` PiperOrigin-RevId: 610428700

References

improve-tabulate-docs

#3715 - Replace deprecated API `jax.tree_map`

Author

JW1992

JW1992

Committer

a-googler

a-googler

Parents

Loading