Closed copybara-service[bot] closed 2 months ago
Updated pl.BlockSpec argument order following google/jax#22209
Updated pl.BlockSpec argument order following google/jax#22209