<p dir="auto">In jax/interpreters/xla.py,<br>
<code class="notranslate">if FLAGS.jax_debug_nans: check_nans(xla_call_p, out_buf)</code><br>
gives me<br>
<code class="notranslate">NameError: name 'out_buf' is not defined</code><br>
when <code class="notranslate">jax_debug_nans</code> is True.</p>
<p dir="auto">Versions 0.1.42+, <code class="notranslate">out_buf</code> referrence before assignment in line 371 of <code class="notranslate">xla.py</code>.</p>