Go will sadly never be viable for ML tasks due to its C interop. Maybe calling an LLM api but often you have a mix of models. Rust also struggles here a bit due to its poor C++ interop.
Written but someone who has built ML libraries from scratch in Go
Written but someone who has built ML libraries from scratch in Go