This repo is developed based on nanoGPT_mlx but will be more like the original version of nanoGPT.
It is mainly designed for inferencing models trained by nanoGPT. You can get the pytorch model from here and use pt2mlxnpz to convert the .pt model to mlx model to inference.
It is also the start of my CS498MLS's project UMA.