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

Allow microbatching functions (optax.microbatching) to be called with…

5a9fb0f
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar