Skip to content
ci

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

Sign in to view logs

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

Annotations

2 errors

The logs for this run have expired and are no longer available.