Patrick Nguyen

@tatatodd mind taking a look? @LordPython Do you have more information on how you arrived at this error, perhaps the script that generated the graph?

@LordPython is this issue still current?

To answer my own question: ``` def AssertShape(x: jnp.array, expected_shape) -> None: # Shape must be a python element with static size. xs = x.shape if len(xs) != len(expected_shape): raise...