TFLite не нашел операцию для встроенного кода операции «SELECT» версии «3»

Question

У меня есть модель TFLite, преобразованная из Jax:

      converter = tf.lite.TFLiteConverter.experimental_from_jax(
    [serving_func], [[("encoder_inputs", encoder_inputs), ("decoder_inputs", decoder_inputs), ("primings", primings)]])
converter.target_spec.supported_ops = [
    tf.lite.OpsSet.TFLITE_BUILTINS, tf.lite.OpsSet.SELECT_TF_OPS
]
converter.optimizations = [tf.lite.Optimize.DEFAULT]
tflite_model = converter.convert()

Когда я пытаюсь загрузить его в Python, я получаю

      ValueError: Didn't find op for builtin opcode 'SELECT' version '3'. An older version of this builtin might be supported. Are you using an old TFLite binary with a newer model?
Registration failed.

Когда я пытаюсь загрузить модель на устройство в Kotlin, она заканчивается

      java.lang.IllegalArgumentException: Internal error: Cannot create interpreter: Didn't find op for builtin opcode 'SELECT' version '3'. An older version of this builtin might be supported. Are you using an old TFLite binary with a newer model?

Что мне попробовать сделать по-другому? Если я попытаюсь удалить что-либо изtarget_spec.supported_ops, конвертер не будет конвертировать.