Skip to content

Add batch_shape argument to optax.tree_utils.tree_random_like.#1161

Open
carlosgmartin wants to merge 1 commit intogoogle-deepmind:mainfrom carlosgmartin:tree_random_like_batch_shape
+8-1

Commits