predictions = jnp.zeros((32,))
labels = jnp.zeros((32, 1))

diff = predictions - labels  # No error!
print(diff.shape)  # (32, 32) completely wrong, but silent

        __
        __