3 ms·
Nope, changing graph shape requires recompilation: https://github.com/google/jax/discussions/17191 https://github.com/google/jax/discussions/17191
by logicchains 2y ago
Nope, changing graph shape requires recompilation: https://github.com/google/jax/discussions/17191 https://github.com/google/jax/discussions/17191