This is an example of a decoder-only Transformer LM. The only dependency is MLX.
Run the example on the GPU with:
python main.py --gpu
By default the dataset is the PTB corpus. Choose a different dataset with the --dataset option.
-
Notifications
You must be signed in to change notification settings - Fork 1
A decoder-only Transformer LM based on MLX specialized in evaluating logic expressions based on the BoLoCo datasets.
License
klusai/tiny-aristotle
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
A decoder-only Transformer LM based on MLX specialized in evaluating logic expressions based on the BoLoCo datasets.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published