Six implementations of the same tiny GPT language model — from pure Python to PyTorch, JAX, and MLX. Inspired by Andrej Karpathy's microGPT. - View it on GitHub