Package compute provides a modular API for defining and executing multidimensional computation graphs with pluggable backends.

GoMLX: An Accelerated Machine Learning Framework For Go

ONNX Runtime based compute backend for GoMLX

Conveniently download files, models, tokenizers from HuggingFace Hub

Documentation Website for GoMLX projects

ONNX / GoMLX Conversion

OpenXLA API Bindings for Go (StableHLO and PJRT)

XLA build actions for PJRT CPU and binary releases for GoMLX and go-xla.

Darwin backends for GoMLX

GoMLX L