Skip to content

Add optax.tree_utils.tree_batch_shape.#1161

Merged
copybara-service[bot] merged 1 commit intogoogle-deepmind:mainfrom carlosgmartin:tree_random_like_batch_shapeFeb 3, 2025

Commits

Commits on Jan 28, 2025