Traditional value assertions in jitted JAX lead to performance degradation. A new, not yet public, JAX API fixes this.