├── .gitignore ├── LICENSE ├── README.md ├── assets └── baby_gpt2_shakespeare_pretrain_loss.png ├── configs ├── train_gpt2_owt.py └── train_gpt2_shakespeare.py ├── configurator.py ├── data ├── openwebtext │ ├── prepare.py │ └── readme.md └── shakespeare │ ├── prepare.py │ └── readme.md ├── environment.yaml ├── model.py ├── optimizer.py ├── sample.py ├── tboard_utils.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/README.md -------------------------------------------------------------------------------- /assets/baby_gpt2_shakespeare_pretrain_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/assets/baby_gpt2_shakespeare_pretrain_loss.png -------------------------------------------------------------------------------- /configs/train_gpt2_owt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/configs/train_gpt2_owt.py -------------------------------------------------------------------------------- /configs/train_gpt2_shakespeare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/configs/train_gpt2_shakespeare.py -------------------------------------------------------------------------------- /configurator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/configurator.py -------------------------------------------------------------------------------- /data/openwebtext/prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/data/openwebtext/prepare.py -------------------------------------------------------------------------------- /data/openwebtext/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/data/openwebtext/readme.md -------------------------------------------------------------------------------- /data/shakespeare/prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/data/shakespeare/prepare.py -------------------------------------------------------------------------------- /data/shakespeare/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/data/shakespeare/readme.md -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/environment.yaml -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/model.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/optimizer.py -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/sample.py -------------------------------------------------------------------------------- /tboard_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/tboard_utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vithursant/nanoGPT_mlx/HEAD/train.py --------------------------------------------------------------------------------