ComingUp ComingUp
Training a 125M-parameter Model to Autocomplete Piano

Training a 125M-parameter Model to Autocomplete Piano

Aug 20, 2026 AI & Machine Learning
autocomplete generative_ai midi music generation transformer

Gallery

Training a 125M-parameter Model to Autocomplete Piano

About

I trained a 125M-parameter transformer to autocomplete piano performances in real time (~108 notes/sec on an iPhone 15).The idea is basically GitHub Copilot or Tabnine, except instead of prompting it with code, you prompt it by playing a few notes on a MIDI piano. The model then continues what you played, entirely on-device.The app is free if anyone wants to try it. Happy to answer questions about the model, training, Core ML, or the many things that didn't work.

Comments (3)

Georgiana Lowe Georgiana Lowe 4 days ago

125M params hitting 108 notes/sec on mobile hardware is wild

Sonya Kris Sonya Kris 4 days ago

what dataset did u use for the piano training

Dahlia Will Dahlia Will 2 days ago

market for piano autocomplete seems pretty niche tbh