https://github.com/google/jax
Raw File
Tip revision: 28626a30d495913398ac291c5a51cd29b22b046b authored by Mark Sandler on 22 March 2024, 04:50:34 UTC
[JAX] Fix lax.full_like to preserve sharding in jitted code.
Tip revision: 28626a3
.bazelversion
6.1.2
back to top