TFLite Task library is unfortunately used in our project, so there is no possibility to use nightly version.
We are working on native mobile development with kotlin and android studio but as you already pointed, output order is changed, so we are getting error with TFLite Task library, we are trying to solve it since yesterday but without any success.
So as I already said, we tried the solution with downgrade of tensorflow to version 2.5.0 but after that, the learning process is unusable, it takes so long.