Skip to content

Allow microbatching functions (optax.microbatching) to be called with dynamic shape dimensions or under jax.eval_shape without raising unhashable Tracer errors. - #1732

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_945840975
Open

Allow microbatching functions (optax.microbatching) to be called with dynamic shape dimensions or under jax.eval_shape without raising unhashable Tracer errors.#1732
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_945840975

Commits

Commits on Jul 22, 2026