Recent commits
Default branch activity.
Update README with Model Specifications (#27)
Show more
This repository contains JAX example code for loading and running the Grok-1 open-weights model.
Make sure to download the checkpoint and place the ckpt-0 directory in checkpoints - see Downloading the weights
Then, run
pip install -r requirements.txt
python run.py
Update README with Model Specifications (#27)
This repository contains JAX example code for loading and running the Grok-1 open-weights model.
Make sure to download the checkpoint and place the ckpt-0 directory in checkpoints - see Downloading the weights
Then, run
pip install -r requirements.txt
python run.py
Update README with Model Specifications (#27)