Generate ONNX/PB for TAEF1 and load in jax2tf
Original: https://huggingface.co/madebyollin/taef1 License: MIT
