Deprecate TF + JAX