Skip to content

Support tf2jax.convert inside of tf.functions (e.g. when a JAX model is being traced for export as a TF saved model via orbax.export). #2208

Support tf2jax.convert inside of tf.functions (e.g. when a JAX model is being traced for export as a TF saved model via orbax.export).

Support tf2jax.convert inside of tf.functions (e.g. when a JAX model is being traced for export as a TF saved model via orbax.export). #2208

Triggered via pull request October 30, 2025 03:09
Status Failure
Total duration 3m 5s
Artifacts

ci.yml

on: pull_request
Matrix: build-and-test
Fit to window
Zoom out
Zoom in

Annotations

3 errors
Python 3.11 on ubuntu-latest
Process completed with exit code 1.
Python 3.12 on ubuntu-latest
The strategy configuration was canceled because "build-and-test._3_11_ubuntu-latest" failed
Python 3.12 on ubuntu-latest
The operation was canceled.