GoMLX: Accelerating Machine Learning with Go, GPUs, and TPUs

📝 Quick Summary: GoMLX is an accelerated machine learning and generic math framework for Go, designed to be a PyTorch/Jax/TensorFlow equivalent for the Go ecosystem. It supports a pure Go backend for broad compatibility and an optimized backend usin...