GoMLX: ML with Go, vs Jax, TensorFlow, PyTorch

 GoMLX — Machine Learning for Go

An accelerated machine learning and math framework for Go.
A Python-free alternative to Jax, TensorFlow, or PyTorch.
Build JIT-compiled computation graphs and execute them at scale with an OpenXLA-accelerated backend, or deploy anywhere using the pure Go backend.