MiniGPT-JAX
This repository contains the model checkpoints for the MiniGPT-JAX project.
The model was built from scratch using:
- JAX
- Flax NNX
- Grain
- Optax
- Orbax
These checkpoints are intended for use with the corresponding GitHub repository:
GitHub
The project demonstrates the end-to-end implementation of a GPT-style language model, including data loading, training, checkpointing, inference, and interactive text generation.