TFP features since a while (experimentally) the possibility to use automatically composite tensors, e.g. by sticking to a contract, a complex object will be able to pass through a tf.function.
TensorFlow also introduced something similar now, the ExtensionType.
Since we would like to use this features ourselves in a library and were wondering:
what is the future of these? Is the auto composite function of TFP meant to stay? Or should we rather rely on the ExtensionTypes?
What is about JAX, will it adopt one or the other as well? (in case you know)
Many thanks in advance,