Removes the TF dependency on the JaxModule and pushes TF dependencies down to the tensorflow_export. The serving_config processing now also occurs in the tensorflow_export class as it is strictly specific to TensorFlow. TF is still needed in the ExportManager since the public interface allows properties for getting the internal TF module and serving signatures.

[copybara-service[bot]]

Removes the TF dependency on the JaxModule and pushes TF dependencies down to the tensorflow_export. The serving_config processing now also occurs in the tensorflow_export class as it is strictly specific to TensorFlow. TF is still needed in the ExportManager since the public interface allows properties for getting the internal TF module and serving signatures.

[google-cla[bot]]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.