Show HN: Z80-μLM, a 'Conversational AI' That Fits in 40KB

(github.com)

495 points by quesomaster9000 2 days ago

120 comments

How small can a language model be while still doing something useful? I wanted to find out, and had some spare time over the holidays.

Z80-μLM is a character-level language model with 2-bit quantized weights ({-2,-1,0,+1}) that runs on a Z80 with 64KB RAM. The entire thing: inference, weights, chat UI, it all fits in a 40KB .COM file that you can run in a CP/M emulator and hopefully even real hardware!

It won't write your emails, but it can be trained to play a stripped down version of 20 Questions, and is sometimes able to maintain the illusion of having simple but terse conversations with a distinct personality.

--

The extreme constraints nerd-sniped me and forced interesting trade-offs: trigram hashing (typo-tolerant, loses word order), 16-bit integer math, and some careful massaging of the training data meant I could keep the examples 'interesting'.

The key was quantization-aware training that accurately models the inference code limitations. The training loop runs both float and integer-quantized forward passes in parallel, scoring the model on how well its knowledge survives quantization. The weights are progressively pushed toward the 2-bit grid using straight-through estimators, with overflow penalties matching the Z80's 16-bit accumulator limits. By the end of training, the model has already adapted to its constraints, so no post-hoc quantization collapse.

Eventually I ended up spending a few dollars on Claude API to generate 20 questions data (see examples/guess/GUESS.COM), I hope Anthropic won't send me a C&D for distilling their model against the ToS ;P

But anyway, happy code-golf season everybody :)

nineteen999 2 days ago

This couldn't be more perfectly timed .. I have an Unreal Engine game with both VT100 terminals (for running coding agents) and Z80 emulators, and a serial bridge that allows coding agents to program the CP/M machines:

https://i.imgur.com/6TRe1NE.png

Thank you for posting! It's unbelievable how someone sometimes just drops something that fits right into what you're doing. However bizarre it seems.

  • quesomaster9000 2 days ago

    Oh dear, it seems we've... somehow been psychically linked...

    I developed a browser-based CP/M emulator & IDE: https://lockboot.github.io/desktop/

    I was going to post that instead, but wanted a 'cool demo' instead, and fell down the rabbit hole.

    • stevekemp 2 days ago

      That is beautiful.

      I wrote a console-based emulator, and a simple CP/M text-adventure game somewhat recently

      https://github.com/skx/cpmulator/

      At some point I should rework my examples/samples to become a decent test-suite for CP/M emulators. There are so many subtle differences out there.

      It seems I could even upload a zipfile of my game, but the escape-codes for clearing the screen don't work, sadly:

      https://github.com/skx/lighthouse-of-doom

    • jaak 2 days ago

      I've been playing the Z80-μLM demos in your CP/M emulator. Works great! However, I have yet to guess a correct answer in GUESS.COM! I'm not sure if I'm just not asking the right questions or I'm just really bad at it!

    • nineteen999 a day ago

      Haha I love it. Just imagine if instead of DOS-based Windows, a CP/M based alternative evolved and took over the PC industry. Nice one!

  • sixtyj 2 days ago

    Connections: Alternative History of Technology by James Burke documents these "coincidences".

    • TeMPOraL 2 days ago

      Those "coincidences" in Connections are really no coincidence at all, but path dependence. Breakthrough advance A is impossible or useless without prerequisites B and C and economic conditions D, but once B and C and D are in place, A becomes obvious next step.

      • embedding-shape 2 days ago

        Some of those really are coincidences, like "Person A couldn't find their left shoe and ended up in London at a coffee house, where Person B accidentally ended up when their carriage hit a wall, which lead to them eventually coming up with Invention C" for example.

        Although from what I remember from the TV show, most of what he investigates/talks about is indeed path dependence in one way or another, although not everything was like that.

      • sixtyj 2 days ago

        That’s why I’ve put the word in parentheses :)

  • [removed] a day ago
    [deleted]
  • simonjgreen 2 days ago

    Super intrigued but annoyingly I can’t view imgur here

    • abanana 2 days ago

      Indeed, part of me wants to not use imgur because we can't access it, but a bigger part of me fully supports imgur's decision to give the middle finger to the UK after our government's censorship overreach.

      • homebrewer 2 days ago

        It blocks many more countries than just the UK because it's the lowest effort way of fighting "AI" scrapers.

        imgur was created as a sort of protest against how terrible most image hosting platforms were back then, went down the drain several years later, and it's now just like they were.

        • supern0va 2 days ago

          It turns out that running free common internet infrastructure at scale is both hard and expensive, unfortunately. What we really need is a non-profit to run something like imgur.

      • wizzwizz4 2 days ago

        It was a really clever move on Imgur's part. Their blocking the UK has nothing to do with the Online Safety Act: it's a response to potential prosecution under the Data Protection Act, for Imgur's (alleged) unlawful use of children's personal data. By blocking the UK and not clearly stating why, people assume they're taking a principled stand about a different issue entirely, so what should be a scandal is transmuted into positive press.

rahen 2 days ago

I love it, instant Github star. I wrote an MLP in Fortran IV for a punched card machine from the sixties (https://github.com/dbrll/Xortran), so this really speaks to me.

The interaction is surprisingly good despite the lack of attention mechanism and the limitation of the "context" to trigrams from the last sentence.

This could have worked on 60s-era hardware and would have completely changed the world (and science fiction) back then. Great job.

  • noosphr 2 days ago

    Stuff like this is fascinating. Truly the road not taken.

    Tin foil hat on: i think that a huge part of the major buyout of ram from AI companies is to keep people from realising that we are essentially at the home computer revolution stage of llms. I have a 1tb ram machine which with custom agents outperforms all the proprietary models. It's private, secure and won't let me be motetized.

    • Zacharias030 2 days ago

      how so? sound like you are running Kimi K2 / GLM? What agents do you give it and how do you handle web search and computer use well?

giancarlostoro 2 days ago

This is something I've been wondering about myself. What's the "Minimally Viable LLM" that can have simple conversations. Then my next question is, how much can we push it so it can learn from looking up data externally, can we build a tiny model with an insanely larger context window? I have to assume I'm not the only one who has asked or thought of these things.

Ultimately, if you can build an ultra tiny model that can talk and learn on the fly, you've just fully localized a personal assistant like Siri.

  • fho 2 days ago

    You might be interested in RWKV: https://www.rwkv.com/

    Not exactly "minimal viable", but a "what if RNNs where good for LLMs" case study.

    -> insanely fast on CPUs

    • giancarlostoro a day ago

      My personal idea revolves around "can I run it on a basic smartphone, with whatever the 'floor' for basic smartphones under lets say $300 is for memory (let's pretend RAM prices are normal).

      Edit: The fact this runs on a Smartphone means it is highly relevant. My only thing is, how do we give such a model an "unlimited" context window, so it can digest as much as it needs. I know some models know multiple languages, I wouldnt be surprised if sticking to only English would reduce the model size / need for more hardware and make it even smaller / tighter.

  • qingcharles 2 days ago

    I think what's amazing to speculate is how we could have had some very basic LLMs in at least the 90s if we'd invented the tech previously. I wonder what the world would be like now if we had?

  • Dylan16807 2 days ago

    For your first question, the LLM someone built in Minecraft can handle simple conversations with 5 million weights, mostly 8 bits.

    I doubt it would be able to make good use of a large context window, though.

Dwedit 2 days ago

In before AI companies buy up all the Z80s and raise the prices to new heights.

andrepd 2 days ago

We should show this every time a Slack/Teams/Jira engineer tries to explain to us why a text chat needs 1.5GB of ram to start up.

  • dangus 2 days ago

    > It won't write your emails, but it can be trained to play a stripped down version of 20 Questions, and is sometimes able to maintain the illusion of having simple but terse conversations with a distinct personality.

    You can buy a kid’s tiger electronics style toy that plays 20 questions.

    It’s not like this LLM is bastion of glorious efficiency, it’s just stripped down to fit on the hardware.

    Slack/Teams handles company-wide video calls and can render anything a web browser can, and they run an entire App Store of apps, all from a cross-platform application.

    Including Jira in the conversation doesn’t even make logical sense. It’s not a desktop application that consumes memory. Jira has such a wide scope that the word “Jira” doesn’t even describe a single product.

    • ben_w 2 days ago

      > Slack/Teams handles company-wide video calls and can render anything a web browser can, and they run an entire App Store of apps, all from a cross-platform application.

      The 4th Gen iPod touch had 256 meg of RAM and also did those things, with video calling via FaceTime (and probably others, but I don't care). Well, except "cross platform", what with it being the platform.

      • dangus 2 days ago

        Group FaceTime calls didn’t exist at the time. That wasn’t added until 2018 and required iOS 12.

        Remember that Slack does simultaneous multiple participants screen sharing plus annotations plus HD video feeds from all participants plus the entirety of the rest of the app continues to function as if you weren’t on a call at all simultaneously.

        It’s an extremely powerful application when you really step back and think about it. It just looks like “text” and boring business software.

    • messe 2 days ago

      > can render anything a web browser can

      That's a bug not a feature, and strongly coupled to the root cause for slack's bloat.

      • dangus 2 days ago

        One person’s “bloat” is another person’s “critical business feature.”

        The app ecosystem of Slack is largely responsible for its success. You can extend it to do almost anything you want.

        • spopejoy a day ago

          > app ecosystem of Slack is largely responsible for its success.

          Is that true? Slack was one of the first private chats that was not painful to use, circa 2015. I personally hate the integrations and wish they'd just fix the bugs in their core product.

    • andrepd 2 days ago

      My Pentium 3 in 2005 could do chat and video calls and play chess and send silly emotes. There is no conceivable user-facing reason why in 20 years the same functionality takes 30× as many resources, only developer-facing reasons. But those are not valid reasons for a professional. If a bridge engineer claims he now needs 30× as much concrete to build the same bridge as he did 20 years ago, and the reason is his/her own conveinence, that would not fly.

      • ben_w 2 days ago

        > If a bridge engineer claims he now needs 30× as much concrete to build the same bridge as he did 20 years ago, and the reason is his/her own conveinence, that would not fly.

        By itself, I would agree.

        However, in this metaphor, concrete got 15x cheaper in the same timeframe. Not enough to fully compensate for the difference, but enough that a whole generation are now used to much larger edifices.

      • dangus 2 days ago

        I have great doubts that you were doing simultaneous screen sharing from multiple participants with group annotation plus HD video in your group calls, all while supporting chatting that allowed you to upload and view multiple animated gifs, videos, rich formatted text, reactions, slash command and application automation integrations, all simultaneously on your Pentium 3.

        I would be interested to know the name of the program that did all that within the same app during that time period.

        For some reason Slack gets criticism for being “bloated” when it basically does anything you could possibly imagine and is essentially a business communication application platform. Nobody can actually name a specific application that does everything Slack does with better efficiency.

vedmakk 2 days ago

If one would train an actual secret (e.g. a passphrase) into such a model, that a user would need to guess by asking the right questions. Could this secret be easily reverse engineered / inferred by having access to models weights - or would it be safe to assume that one could only get to the secret by asking the right questions?

  • Kiboneu 2 days ago

    I don’t know, but your question reminds me of this paper which seems to address it on a lower level: https://arxiv.org/abs/2204.06974

    “Planting Undetectable Backdoors in Machine Learning Models”

    “ … On the surface, such a backdoored classifier behaves normally, but in reality, the learner maintains a mechanism for changing the classification of any input, with only a slight perturbation. Importantly, without the appropriate "backdoor key", the mechanism is hidden and cannot be detected by any computationally-bounded observer. We demonstrate two frameworks for planting undetectable backdoors, with incomparable guarantees. …”

  • ronsor 2 days ago

    > this secret be easily reverse engineered / inferred by having access to models weights

    It could with a network this small. More generally this falls under "interpretability."

roygbiv2 2 days ago

Awesome. I've just designed and built my own z80 computer, though right now it has 32kb ROM and 32kb RAM. This will definitely change on the next revision so I'll be sure to try it out.

  • wewewedxfgdf 2 days ago

    RAM is very expensive right now.

    • wickedsight 2 days ago

      I just removed 128 megs of RAM from an old computer and am considering listing it on eBay to pay off my mortgage.

    • tgv 2 days ago

      We're talking kilobytes, not gigabytes. And it isn't DDR5 either.

      • boomlinde 2 days ago

        Yeah, even an average household can afford 40k of slow DRAM if they cut down on luxuries like food and housing.

gcanyon 2 days ago

So it seems like with the right code (and maybe a ton of future infrastructure for training?) Eliza could have been much more capable back in the day.

  • antonvs 2 days ago

    The original ELIZA ran on an IBM 7094 mainframe, in the 1960s. That machine had 32K x 36-bit words, and no support for byte operations. It did support 6-bit BCD characters, packed 6 per word, but those were for string operations, and didn't support arithmetic or logical operations.

    This means that a directly translated 40 KB Z80 executable might be a tight squeeze on that mainframe, because 40K > 32K, counting words, not bytes. Of course if most of that size is just 2-bit weight data then it might not be so bad.

    ELIZA running on later hardware would have been a different story, with the Z80 - released in 1976 - being an example.

orbital-decay 2 days ago

Pretty cool! I wish free-input RPGs of old had fuzzy matchers. They worked by exact keyword matching and it was awkward. I think the last game of that kind (where you could input arbitrary text when talking to NPCs) was probably Wizardry 8 (2001).

gwern 2 days ago

So if it's not using attention and it processes the entire input into an embedding to process in one go, I guess this is neither a Transformer nor a RNN but just a MLP?

Peteragain 2 days ago

There are two things happening here. A really small LLM mechanism which is useful for thinking about how the big ones work, and a reference to the well known phenomenon, commonly dismissively referred to as a "trick", in which humans want to believe. We work hard to account for what our conversational partner says. Language in use is a collective cultural construct. By this view the real question is how and why we humans understand an utterance in a particular way. Eliza, Parry, and the Chomsky bot at http://chomskybot.com work on this principle. Just sayin'.

  • nrhrjrjrjtntbt 2 days ago

    MAYBE

    • cwmoore 2 days ago

      Universally correct reply, although honestly a bit vague.

      • Peteragain a day ago

        Fair. The background reading is the EMCA stuff - conversation analysis cf Sacks etc at, and Ethnomethods (Garfunkel). And Vygotsky cf Kozulin. People such as Robert Moore at IBM and Lemon at Herriot-Watt work in this space but there is no critical mass in the face of LLM mania.

      • Peteragain a day ago

        And the Chomskybot analysis is quite enlightening..

bartread 2 days ago

This is excellent. Thing I’d like to do if I had time: get it running on a 48K Spectrum. 10 year old me would have found that absolutely magical back in the 1980s.

  • tomduncalf 2 days ago

    This was my first thought too haha. That would be mind blowing

    • bartread 2 days ago

      Yeah, very WarGames.

      EDIT: Actually thinking about it some more…

      - Imagine what you could do with 16-bit games of the era with one or more of these models embedded. Swap the model depending on the use case within the game. Great for adventures, RPGs, strategy, puzzle, and trading games (think Elite). With 512K or 1MB of RAM, plus 2 - 4 floppies (which became increasingly common as the era wore on), you could probably do a lot, especially if the outcomes of conversations can result in different game outcomes

      - Back in the day nobody was really trying to do anything serious with AI on 8 or even most 16-bit machines, because nobody thought they were powerful enough to do anything useful with. Now the thinking has changed to how much somewhat useful intelligence can I cram into the least powerful device, even if that’s only for fun?

      - Imagine showing this running on a CP/M machine, like the C128, to a serious AI researcher working back in the 1980s. Minds blown, right?

      - Now spool forward 10 years into the 1990s and think what PC hardware of that era would have been capable of with these limited language models. I wonder what that era might have looked like with something that seems like somewhat useful conversational AI? A sort of electro-steampunk-ish vibe maybe? People having really odd conversations with semi-capable home automation running via their PCs.

Zee2 2 days ago

This is super cool. Would love to see a Z80 simulator set up with these examples to play with!

  • dmd 11 hours ago

    https://3e.org/private/z80ulmweb/

    It's just one-shot AI slop - literally, the prompt was 'make a web based version of [github url of this project]' and it spat this out. It appears to work fine.

    I'll keep it up for a couple of months and then it'll be auto-deleted, no sense in keeping it around longer than that.

MagicMoonlight 2 days ago

What I really want is a game where each of the NPCs has a tiny model like this, so you can actually talk to them.

  • GuB-42 15 hours ago

    I thought about this, chatbots existed well before LLMs (Eliza: 1966!) and the only time I have seen a commercially successful game with a (very simple) chatbot was Quake III Arena!

    Quake 3 is probably the last game where you would expect a chatbot, as there are few games where storytelling matters less and it is a very little known feature, but Quake 3 bots can react to what you say in the chat, in addition to the usual taunts.

    But that's the thing, Quake 3 can do it because it is inconsequential, in a story-driven game like a RPG, NPCs have a well defined spot in the story and gameplay, they tell you exactly what you need to know, as to not disrupt the flow of the story. Tell you too much, and they will spoil the big reveal, tell you too little, and you don't know what to do, tell you irrelevant details and you get lost chasing them. It has to be concise and to the point, so that those who don't really care know what to do to advance the story, but with enough flavor to make the world alive. It is really hard to find the right balance, and if in addition, you have to incorporate a chatbot, it borders on impossible.

    It looks like a good idea on the surface, but it most likely isn't, unless it is clearly not part of the main gameplay loop, as in Quake 3.

    Some people had some success using a (big) LLM as a DM in D&D, which I think is easier since it can make up the story as it advances, it is much harder to make up game elements in a computer RPG that are not programmed in.

vatary 2 days ago

It's pretty obvious this is just a stress test for compressing and running LLMs. It doesn't have much practical use right now, but it shows us that IoT devices are gonna have built-in LLMs really soon. It's a huge leap in intelligence—kind of like the jump from apes to humans. That is seriously cool.

  • acosmism 2 days ago

    i'll echo that practicality only surfaces once it is apparent what can be done. yea this feels like running "DOOM on pregnancy test devices" type of moment

anonzzzies 2 days ago

Luckily I have a very large amount of MSX computers, zx, amstrad cpc etc and even one multiprocessor z80 cp/m machine for the real power. Wonder how gnarly this is going to perform with bankswitching though. Probably not good.

alfiedotwtf 2 days ago

An LLM in a .com file? Haha made my day

  • teaearlgraycold 2 days ago

    SLM

    • quesomaster9000 2 days ago

      All the 'Small' language models and the 'TinyML' scene in general tend to bottom out at a million parameters, hence I though 'micro' is more apt at ~150k params.

jacquesm 2 days ago

Between this and RAM prices Zilog stock must be up! Awesome hack. Now apply the same principles to a laptop and take a megabyte or so, see what that does.

a_t48 2 days ago

Nice - that will fit on a Gameboy cartridge, though bank switching might make it super terrible to run. Each bank is only 16k. You can have a bunch of them, but you can only access one bank at a time (well, technically two - bank 0 is IIRC always accessible).

  • ColonelPhantom 2 days ago

    Each layer of the LM is also at most 16 KiB, so if you want to minimize bank switching, I think making sure each layer is in one bank would be enough? Bank switching shouldn't give much overhead anyway unless it complicates an inner loop, which would be avoided if no layers are split across banks.

  • ant6n 2 days ago

    You have 32KB of ROM, plus 8 Kb of ram on original game boy. Game boy color has more. Bank switching is super fast, as well. Given that models are likely streamed, I doubt the bank switching is a problem.

    Biggest pain point is likely the text input.

jasonjmcghee 2 days ago

For future projects and/or for this project, there are many LLMs available more than good enough to generate that kind of synthetic data (20 Qs) with permissive terms of use. (So you don’t need to stress about breaking TOS / C&D etc)

Zardoz84 2 days ago

Meanwhile, Eliza was ported to BASIC and was run on many home computers in the 80s.

coolius 2 days ago

This is impressive, those are some very restrictive requirements. I wonder what we are able to run on more powerful hardware such as ESP32 or RP2040, has anyone tried this?

magicalhippo 2 days ago

As far as I know, the last layer is very quantization-sensitive, and is typically not quantized, or quantized lightly.

Have you experimented with having it less quantized, and evaluated the quality drop?

Regardless, very cool project.

  • kouteiheika 2 days ago

    (Not OP)

    It depends on the model, but from my experiments (quantizing one layer of a model to 2-bit and then training the model with that layer in 2-bit to fix the damage) the first layer is the most sensitive, and yes, the last layer is also sensitive too. The middle layers take the best to quantization.

    Different components of a layer also have a different sensitivity; e.g. the MLP downscale block damages the model the most when quantized, while quantizing the Q projection in self attention damages the model the least.

pdyc 2 days ago

interesting, i am wondering how far can it go if we remove some of these limitations but try to solve some extremely specific problem like generating regex based on user input? i know small models(270M range) can do that but can it be done in say < 10MB range?

  • Waterluvian 2 days ago

    Generate an LLM that is designed to solve one extremely specific problem: answering the ultimate question of life, the universe, and everything.

    Even with modern supercomputing the computation would be outpaced by the heat death of the universe, so token output must be limited to a single integer.

DrNosferatu 2 days ago

Awesome! Anyone for a port to the MSX?

A web version would also be cool.

Y_Y 2 days ago

Very cool. Did you consider using sparse weights?

lostmsu 21 hours ago

Did you train the model with quantization awareness? How?

codetiger 2 days ago

Imagine, this working on a Gameboy, in those days. Would've sounded like magic

  • Sharlin 2 days ago

    I don’t think this could beat an ELIZA-style bot in how magical it feels, given the extreme terseness of its replies.

  • lodovic 2 days ago

    I love these thought experiments. Looking at the code size, it would have been possible for someone to come up with this back in the days, similar to the idea of a million monkeys on a typewriter eventually producing Shakespeare.

  • alfiedotwtf 2 days ago

    And would have lasted 3 minutes.

    Speaking of - I remember my first digital camera (Fujitsu 1Mb resolution using SmartMedia)… it used so much power that you could take 20-30 photos and then needed to replace all 4 batteries lol

  • numpad0 2 days ago

    Flip phones had predictive texts since forever. LLMs are just* supercharged predi[ctive text algorithms are computer algorithms that are]

  • qingcharles 2 days ago

    "Look, my Game Boy passes the Turing Test!"

    *burns you at the stake*