Bug Fix: Tracers do not always shave .sharding, use jax.typeof to get an abstract value.#28
Open
copybara-service[bot] wants to merge 1 commit into
Open
Bug Fix: Tracers do not always shave .sharding, use jax.typeof to get an abstract value.#28copybara-service[bot] wants to merge 1 commit into
.sharding, use jax.typeof to get an abstract value.#28copybara-service[bot] wants to merge 1 commit into