jax

053cd5aa - [jax2tf] Clean up test_dynamic_slice. (#4236)

[jax2tf] Clean up test_dynamic_slice. (#4236) * [jax2tf] Clean up test_dynamic_slice. With the XLA nested compilation bug fixed, this should now work fine.