This repository contains JAX example code for loading and running the Grok-1 open-weights model.