Any update on this? I’m in the same situation. Are there any other good substitutes for Tensorflow that could work?