jax

798a2648 - [jax2tf] Fix bug in population count and move expect_tf_exception (#4214)

[jax2tf] Fix bug in population count and move expect_tf_exception (#4214) into correctness stats. The code was using `tf.bitcast` instead of `tf.cast`, but using `expect_tf_exception` in every case was hiding the errors.