This was tensorflow btw which has Go bindings support.
It is a smart & worthwhile move, we also needed to drop python for performance/cost gains.
Actually the docs around this weren’t great. Took the train-in-python & inference-in-go approach. And only for versions greater than tf2