Examples developed using GoMLX

🚀 NEW 🚀 KaLM-Gemma3 12B parameters: Tencent’s top-ranked sentence encoder for RAGs, using go-huggingface to load the model and tokenizer, and GoMLX to execute it.

🚀 NEW 🚀 Gemma 3 270M: Demonstrates ONNX-converted text generation (LLM) using the onnx-community/gemma-3-270m-it-ONNX model with GoMLX. It uses the gomlx/onnx-gomlx package to convert the model, and gomlx/go-huggingface to download the model and run the tokenizer.

🚀 NEW 🚀 GPT-2: Demonstrates text generation using the new (experimental) transformer and generator packages.

🚀 NEW 🚀 BERT-base-NER: A BERT-base model fine-tuned for Named Entity Recognition. It’s also an ONNX-converted model from the dslim/bert-base-NER model on HuggingFace.

🚀 NEW 🚀 MixedBread Reranker v1: A cross-encoder reranking example. See HuggingFace MixedBread Reranker v1 page. It uses the gomlx/onnx-gomlx package to convert the model, and gomlx/go-huggingface to download the model and run the tokenizer.

Adult/Census model

How do KANs learn?

Cifar-10 demo

MNIST demo (library and command-line only)

Dogs & Cats classifier demo

IMDB Movie Review demo

Diffusion model for Oxford Flowers 102 dataset (generates random flowers)

Flow Matching Study Notebook based on Meta’s “Flow Matching Guide and Code”.

GNN model for OGBN-MAG (experimental)

Last, a trivial synthetic linear model, for those curious to see a barebones simple model.

Neural Style Transfer 10-year Celebration: see a demo written using GoMLX of the original paper.

Triplet Losses: various negative sampling strategies as well as various distance metrics.

AlphaZero AI for the game of Hive: It uses a trivial GNN to evaluate positions on the board. It includes a WASM demo (runs GoMLX in the browser!) and a command-line UI to test your skills!