jax

5d6413ce - Added debug_callback to the list of exclusions in jax2tf/tests/primitives_test.py