When JIT hides your errors: Debugging JAX
Last week, Chex gave us a firewall against shape mismatches and NaN gradients. That firewall catches the bugs that announce themselves; a tensor with the wrong shape, a gradient that explodes into inf
Jul 18, 202613 min read31
