3 ms·

Doesn't JAX support dynamic graphs as well?

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