Show HN: I trained a 125M model to autocomplete piano on-device

434 points - today at 12:04 PM


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.

Source

Comments

tom_vidal today at 3:23 PM
This sort of “autocomplete” is actually fundamental to how classical composers were trained.

For anyone interested, I’d highly recommend reading Robert Gjerdingen’s article Gebrauchs-Formulas. https://www.researchgate.net/publication/259731561_Gebrauchs...

You can also listen to the transcript of four Russian composers, including Rachmaninoff, playing this pattern recognition and generation game at a dinner party in the late 1800’s: https://youtu.be/PlFPOWuwBHI?is=EKBK7QQkJs4MsTCU

Composers at the time could do this just by looking at sheet music and audiating, without using a piano.

jasonjmcghee today at 2:45 PM
I think this is a great project and very HN. Not sure why the comments are so focused on the deliverable- you learned way more and had a much more interesting experience.

One think I didn't see mentioned in the post- maybe I missed it- how large was the data? How many samples did you use to pretrain and post-train

joshuamerrill today at 6:16 PM
Classical pianist and software product designer here.

I see so much in common with this project and the numerous AI-based UX design tools out there. Whether it's music or UI, now that the "generation" portion of the work costs zero, all that remains is taste.

And so much of taste comes from exploring and killing off possibilities that turn out to be dead-ends. I love the idea that models like these will help us find the dead ends faster, or even produce a gem here and there.

P.S. if you want another uncanny version of Fur Elise, listen to Beethoven's own 1822 revision: https://www.youtube.com/watch?v=s24TtiGgb6k. His 1810 version that we all know was simpler and more balanced. But for what it's worth, Beethoven didn't publish either of them.

goda90 today at 1:34 PM
Reminds me of this project to generate every melody possible algorithmically in order to fight music copyright lawsuits. https://allthemusic.info/
karmelapple today at 3:09 PM
Hearing the start of FĂĽr Elise, and then it being taken in an incredibly different direction, is surprisingly disconcerting.
ghm2199 today at 3:15 PM
Interesting. I do feel letting a machine generating notes is taking the joy out of improvisation, is it not?

I think one of the great, early, joys of learning a piano is gaining the following intuitions: The seemingly harder path of learning sheet is actually faster. Your mind _should_ learn to think in two dimensions Spatial — where fingers go — and Time — pitch and tempo – ** when learning. The _internalization_ of Space and Time queues guide the fingers in a dance that is vastly satisfying. This skill leads you to the final part of the journey that is improvisation and the one more exciting than what i am on now.

---

**

Space: Your finger placement on keys right, e.g. knowing how to go from landmark/anchor notes(mid-c, G, F etc) and then go to the others above and below it. Crudely this is some what like typing from your landmark f and j qwerty keyboard

Time: The out singing/verbalizing of the notes/beats on a time measure as you play them(per the time measure). e.g. you can say out loud 1-2-3-4 for 4/4 measure, if the measure has quarter notes say out loud. And `1-e-and-a-2-e-and-a-3-e-and-a-4-e-and-a` for a 4/4 with 1/16th note granularity. Do this as you play the notes and you get a sense of tempo.

timmb today at 1:30 PM
Reminds me of Francois Pachet’s Continuator (all the way back in 2003, using hierarchical markov models)

https://www.francoispachet.fr/continuator/

hliyan today at 1:56 PM
Ah, MIDI files. The only type of music you could realistically download from the internet back in the day, and you had to wake up at ungodly hours so that your dialup modem would not rack up a massive phone bill.
heikkilevanto today at 1:15 PM
I would love something like that, except that I play the melody, and it produces proper 3-4 part accompaniment, preferably in good baroque style. Extra bonus if it could also write it into a file in a format suitable for music editing programs.
mare5x today at 4:58 PM
Really cool! It reminds me of a demo I saw live at NeurIPS: https://www.youtube.com/watch?v=8s3V922h3CU (paper at https://openreview.net/pdf?id=3yeBer3J5z).

If you like such creative AI work, I would recommend looking at the other "NeurIPS Creative AI Track 2025" submissions as well.

devonsolomon today at 1:25 PM
This is really fun. Scaler 3 starts with a chord progression and lets you break it down into musical performances and parts. Useful for ideation when producing.

Would be fun to get a midi clock going and play some chords on my piano and have my synth start jamming along with the bass and my keyboard doing some performance. Or any combination of the above.

sagebird today at 3:20 PM
FĂĽr Elsevier Journal Access Library pass out candy for safety this halloween your baby off milk shake it off.
rajivayyangar today at 6:00 PM
The results strike me as comparable or worse than you could get with a Markov model. I think it reveals the gap in understanding between LLMs and music. I think you need to either: - Set up a pipeline to decompose music into, say, harmonic sequences and melodic sequences, and then have the LLM work on some more fundamental or more high-level layer of musical composition and then re-translate it back into actual sounds. - Develop a better dataset and train the LLM more natively on musical examples.

Does anybody know of a project that has produced more convincing results?

axoltl today at 7:18 PM
Looks neat! I'd love for it to feed the MIDI notes back into my player piano instead of playing out of my iPhone's (comparatively tinny) speakers though.
evalystai today at 12:29 PM
The idea is awesome! :) However there's definitely much room for improvement, first of all rythm and composition (so there's some sense of musical form).
bravura today at 5:05 PM
You could also distill the fabulous Anticipatory Music Transformer from Stanford. https://crfm.stanford.edu/2023/06/16/anticipatory-music-tran...

Also consider checking their decisions about representation, etc

Tepix today at 2:46 PM
Oh, a cool idea! I just tried it, works pretty well. Kudos!

One feature request:

Instead of playing the AI-generated audio solely through the iPhone's speakers, add an option to send the audio as midi notes to a device (probably the same one you received the mini notes from).

butlike today at 4:08 PM
Question as a musician: Do you find satisfaction in the piece it created in-and-of-itself (subtracting the satisfaction from making the tool itself)?
davidajackson today at 1:37 PM
How would you expand this to support elements like attack ("velocity of the key-down" in piano speak), grace notes, timing etc. Would each of those be part of this model or another model? How would you model an arbitrary element (pedal, duration, etc...)
Ohentis today at 5:58 PM
It occurres to me that the relationship between pitches usually matters more than the pitches themselves. Perhaps the pitches could be encoded in a vector like what is typically done for position.
isoprophlex today at 12:32 PM
Amazing idea! Gonna hook this up to my little synthesizer and blast some square wave arpeggiated ML music!
dandaka today at 7:06 PM
I would love to see a jamming partner. So I could play along him on the same piece.
subhajeet2107 today at 1:21 PM
This is so amazing, can you improve the quality of generation at the cost of notes per seconds ? No one can play 108 notes/sec anyways, maybe you can train the model to do CoT for better quality
gaya3bollineni today at 2:09 PM
Running a 125M model on-device at that speed is impressive. How much did you have to optimize the model to get that performance on an iPhone?
pil0u today at 1:27 PM
Even after a few years deep into AI, I find your application absolutely magic. This is very inspiring, thank you for sharing.
leobg today at 1:31 PM
I don’t have MIDI. How about whistling or playing the piano via microphone? Sounds easy. Another 6 month rabbit hole? :)
oleggromov today at 4:44 PM
This is almost exactly what Jordan Rudess is doing now with some folks from MIT or Stanford right now. Google his interview with Rick Beato.
speedgoose today at 2:12 PM
Talking about AI music with some live human MIDI inputs, Magenta Realtime 2 was released a few weeks ago and is pretty fun.

https://magenta.withgoogle.com/magenta-realtime-2

h2aichat today at 4:19 PM
Congratulations! It is amazing!!! Can you do the same with a song and give different drums to see which one fits better?
dyzone today at 5:48 PM
I wonder if training SOTA models on this will make them more humane?
luciana1u today at 6:06 PM
next step is a model that autocompletes the part where you actually practice. mine just sits there judging my scales.
arkmm today at 1:51 PM
Very cool! Can you say a little bit about the size of the DPO training examples and how long training took?
theschwa today at 3:06 PM
Would you be willing to share about how much it cost to train a model like this?
yoouareperfect today at 12:26 PM
This is really awesome thanks for sharing
the_black_hand today at 2:20 PM
Cool work. I tried using LLMs to parse sheet music and they are really bad.
bilsbie today at 3:45 PM
Should tokens be multiple Midi notes? Why or why not?
deleted today at 3:38 PM
bubblegumcrisis today at 7:00 PM
This is disgusting. I'm sorry. All the positive comments. It's like Waymo's posts on HN. Incomprehensible to me.

I guess no one actually wants to learn about harmony, about voicing, and voice leading, spend the hours. No one wants to learn how to actually play an instrument. I guess no one is willing to do the work.

They want to just press some keys and declare that they made what the computer generated.

This is all so depressing for me.

I think in the near future, the world will divide, and all of the people who wish to think and do, will put up a wall separating themselves from the hu-bots that infest the rest of the planet. It's alright I guess. I hope there isn't a slaughter of one side or the other.

abidlabs today at 3:54 PM
Are the weights of the model available?
noman-land today at 3:43 PM
This is honestly astonishing and the first "AI music" I've heard that has the potential to sound beautiful. I always thought that MIDI would be a perfect format for this. Glad to see this person make it happen!
jwr today at 1:34 PM
> Eventually I used Gemini 3.5 Flash for pairwise evaluation

But, but… wouldn't that be… (gasp) DISTILLATION?

Fun project!

aghilmort today at 2:35 PM
really incredible work! great use case, impeccable learning strategy, congrats!
amarbirsingh today at 3:47 PM
Pretty Impressive!!
micromacrofoot today at 4:21 PM
I too am currently learning to play Sonatina in G Major, it's a fun one! the end is surprisingly tricky
mandeepj today at 2:40 PM
Gemma 4 E2B was too heavy for your needs?
deleted today at 1:35 PM
deleted today at 2:38 PM
lvbyte today at 1:30 PM
[dead]
qmmmur today at 12:26 PM
[flagged]