7 ms·
Flux 2 Klein pure C inference
- reactordev 9mo agoThis is both awesome and scary. Yes, now we can embed image gen in things like game engines and photoshop or build our own apps. On the other hand, we can include image gen in anything…
- nusl 9mo agoThis was possible before, though
- rvz 9mo agoYes, it was always possible. It's almost as if this is the first time many have seen something built in C with zero dependencies which makes this easily possible. Since they are used to languages with package managers adding 30 package and including 50-100+ other dependencies just before the project is able to build.
- snarfy 9mo agorip 1425 https://xkcd.com/1425/ https://xkcd.com/1425/
- d_watt 9mo agoRegarding the meta experiment of using LLMs to transpile to a different language, how did you feel about the outcome / process, and would you do the same process again in the future? I've had some moments recently for my own projects as I worked through some bottle necks where I took a whole section of a project and said "rewrite in rust" to Claude and had massive speedups with a 0 shot rewrite, most recently some video recovery programs, but I then had an output product I wouldn't feel comfortable vouching for outside of my homelab setup.
- deleted 9mo ago[deleted]
- antirez 9mo agoIt depends on the situation. In this case the agent worked only using the reference code provided by Flux's Black Forest Labs which is basically just the pipeline implemented as a showcase. The fundamental way for this process to work is that the agent can have a feedback to understand if it is really making progresses, and to debug failures against a reference implementation. But then all the code was implemented with many implementation hints about what I wanted to obtain, and without any reference of other minimal inference libraries or kernels. So I believe this just is the effect of putting together known facts about how Transformers inference works plus an higher level idea of how software should appear to the final user. Btw today somebody took my HNSW implementation for vector sets and translated it to Swift (https://github.com/jkrukowski/swift-hnsw https://github.com/jkrukowski/swift-hnsw). I'm ok with that, nor I care of this result was obtained with AI or not. However it is nice that the target license is the same, given the implementation is so similar to the C one.
- rcarmo 9mo agoThis is pretty great. I’ve gone and hacked your GTE C inference project to Go purely for kicks, but this one I will look at for possible compiler optimizations and building a Mac CLI for scripting…
- kubb 9mo agoThis repo has Swift wrappers, not a rewrite of hnsw.c, which apparently you weren't the only author of.
- antirez 9mo agoThanks,I thought it was a complete rewrite of the same logic and algorithms.
- jhatemyjob 9mo agoWhen I first saw the OP, panic started to set in that I am fucked and Chat-Completions/LLMs/AI/whatever-you wanna-call-it will soon be able to create anything and eat away at my earning potential. And I will spend my elder years living with roommates, with no wife or children because I will not be able to provide for them. But upon reading that you used a reference implementation, I've realized that you simply managed to leverage it as the universal translator apenwarr believes is the endgame for this new technology [1]. So, now I feel better. I can sleep soundly tonight knowing my livelihood is safe, because the details still matter. [1] https://apenwarr.ca/log/20251120 https://apenwarr.ca/log/20251120
- antirez 9mo agoSomething that may be interesting for the reader of this thread: this project was possible only once I started to tell Opus that it needed to take a file with all the implementation notes, and also accumulating all the things we discovered during the development process. And also, the file had clear instructions to be taken updated, and to be processed ASAP after context compaction. This kinda enabled Opus to do such a big coding task in a reasonable amount of time without loosing track. Check the file IMPLEMENTATION_NOTES.md in the GitHub repo for more info.
- lukebechtel 9mo agoVery cool! Yep, a constantly updated spec is the key. Wrote about this here: https://lukebechtel.com/blog/vibe-speccing https://lukebechtel.com/blog/vibe-speccing I've also found it's helpful to have it keep an "experiment log" at the bottom of the original spec, or in another document, which it must update whenever things take "a surprising turn"
- ctoth 9mo agoHonest question: what do you do when your spec has grown to over a megabyte? Some things I've been doing: - Move as much actual data into YML as possible. - Use CEL? - Ask Claude to rewrite pseudocode in specs into RFC-style constrained language? How do you sync your spec and code both directions? I have some slash commands that do this but I'm not thrilled with them? I tend to have to use Gemini for actually juggling the whole spec. Of course it's nice and chunked as much as it can be? but still. There's gonna need to be a whole new way of doing this. If programming languages can have spooky language at a distance wait until we get into "but paragraph 7, subsection 5 of section G clearly defines asshole as..." What does a structured language look like when it doesn't need mechanical sympathy? YML + CEL is really powerful and underexplored but it's still just ... not what I'm actually wanting.
- lukebechtel 9mo agoSharding or compaction, both possible with LLMs. Sharding: Make well-named sub-documents for parts of work. LLM will be happy to create these and maintain cross references for you. Compaction: Ask the LLM to compact parts of the spec, or changelog, which are over specified or redundant.
- throwaway2027 9mo agoIf I asked Claude to do the same can I also just put MIT license on it with my name? https://github.com/black-forest-labs/flux2 https://github.com/black-forest-labs/flux2 uses Apache License apparently. I know it doesn't matter that much and as long as it's permissive and openly available people don't care it's just pedantics but still.
- antirez 9mo agoThe reference code shows how to setup the inference pipeline. It does not implement 99% of what the C code does. That is, the inference kernels, the transformer and so forth.
- netdur 9mo agoi would love if you took the time to instruct claude to re-implement inference in c/c++, and put an mit license on it, it would be huge, but only if it actually works
- badsectoracula 9mo agoFWIW stable-diffusion.cpp[0] (which implements a lot more than just stable diffusion, despite the name) is already a MIT licensed C++ library. [0] https://github.com/leejet/stable-diffusion.cpp/ https://github.com/leejet/stable-diffusion.cpp/
- eikenberry 9mo agoAssuming this was done in a US jurisdiction it doesn't matter what license you put on it as it is public domain and it needs no license. The US copyright office has ruled that anything AI generated is not covered by copyright.
- jacquesm 9mo agoCorrection: it has ruled that anything AI generated is not copyrightable. That's a very important little difference and it does not mean that the production of the AI is not covered by copyright, it may well be (though proving that is going to be hard in most cases).
- neomantra 9mo agoThanks for sharing this — I appreciate your motivation in the README. One suggestion, which I have been trying to do myself, is to include a PROMPTS.md file. Since your purpose is sharing and educating, it helps others see what approaches an experienced developer is using, even if you are just figuring it out. One can use a Claude hook to maintain this deterministically. I instruct in AGENTS.md that they can read but not write it. It’s also been helpful for jumping between LLMs, to give them some background on what you’ve been doing.
- antirez 9mo agoIn this case, instead of a prompt I wrote a specification, but later I had to steer the models for hours. So basically the prompt is the sum of all such interactions: incredibly hard to reconstruct to something meaningful.
- enriquto 9mo agoThis steering is the main "source code" of the program that you wrote, isn't it? Why throw it away. It's like deleting the .c once you have obtained the .exe
- minimaxir 9mo agoIt's more noise than signal because it's disorganized, and hard to glean value from it (speaking from experience).
- neomantra 9mo agoI wasn’t exactly suggesting this. The source code (including SVG or DOCX or HTMl+JS for document work) is the primary ground truth which the LLM modifies. Humans might modify it too. This ground truth is then rendered (compiled, visualized) to the end product. The PROMPTS.md is communication metadata. Indeed, if you fed the same series of prompts freshly, the resultant ground truths might not make sense because of the stochastic nature of LLMs. Maybe “ground truth” isn’t exactly the right word, but it is the consistent, determined basis which formed from past work and will evolve with future work.
- csto12 9mo agoAs someone who doesn’t code in C and does more analytics work (SQL), is the code generated here “production grade?” One of the major criticisms I hear about llms is they tend to generate code that you wouldn’t want to maintain, is that the case here?
- chrsw 9mo agoIt's not bad. Skimming the code I'd say it's not enterprise quality but it's definitely better than an amateur throwaway project.
- keyle 9mo agoClassic. non-enterprise C quality.
- minimaxir 9mo agoThose statements are mostly out of date and symptomatic of pre-agent-optimized LLMs. Opus 4.5 with clarifying rules in the CLAUDE.md does a good job at following idiomatic best practices in my experience. That said, I'm mixed on agentic performance for data science work but it does a good job if you clearly give it the information it needs to solve the problem (e.g. for SQL, table schema and example data)
- hirako2000 9mo agoNot my experience. All frontier models I constantly test, agentic or not, produce code less maintainable than my (very good) peers and myself (on a decent day). Plus they continue to introduce performance blunders. Crying wolves, on day maybe there will be a wolf and I may be the last of us to check whether that's true.
- adefa 9mo agoI ran a similar experiment last month and ported Qwen 3 Omni to llama cpp. I was able to get GGUF conversion, quantization, and all input and output modalities working in less than a week. I submitted the work as a PR to the codebase and understandably, it was rejected. https://github.com/ggml-org/llama.cpp/pull/18404 https://github.com/ggml-org/llama.cpp/pull/18404 https://huggingface.co/TrevorJS/Qwen3-Omni-30B-A3B-GGUF https://huggingface.co/TrevorJS/Qwen3-Omni-30B-A3B-GGUF
- antirez 9mo agoThe refusal because often AI writes suboptimal GGML kernels looks very odd, to me. It means that who usually writes manually GGML kernels, could very easily steer the model into writing excellent kernels, and even a document for the agents can be compiled with the instructions on how to do a great work. If they continue in this way, soon a llama.cpp fork will emerge that will be developed much faster and potentially even better: it is unavoidable.
- rjh29 9mo agoThe refusal is probably because OP said "100% written by AI" and didn't indicate an interest in actually reviewing or maintaining the code. In fact, a later PR comment suggests that the AI's approach was needlessly complicated.
- hirako2000 9mo agoAlso because it's a large PR. Also because the maintainer has better things to do than taking longer and more energy to review than the author spent to write it, just to find that multiple optimisations will be requested, which the author may not be able to take on. the creator of llama.cc can hardly be suspected to be reluctant or biased towards GenAI.
- adefa 9mo agoAbsolutely -- it's perfectly understandable. I wanted to be completely upfront about AI usage and while I was willing and did start to break the PR down into parts, it's totally OK for the maintainers to reject that too. I wanted to see if Claude Code could port the HF / MLX implementation to llama.cpp and it was successful -- in my mind that's wild! I also learned a ton about GPU programming, how omni models work, and refined my approach to planning large projects with automated end to end integration tests. The PR was mostly to let people know about the code and weights, since there are quite a few comments requesting support: https://github.com/ggml-org/llama.cpp/issues/16186 https://github.com/ggml-org/llama.cpp/issues/16186
- yunnpp 9mo ago> I believe that inference systems not using the Python stack (which I do not appreciate) are a way to free open models usage and make AI more accessible. What you're saying here is that you do not appreciate systems not using the Python stack, which I think is the opposite of what you wanted to say.
- tomashubelbauer 9mo agoI am an ESL speaker but I don't see why the sentence fragment in parentheses couldn't be parsed as relating only to "Python stack" as opposed to "systems not using the Python stack". I read it that way, but again, as an ESL speaker, I might be missing intuition or actual grammatical knowledge that would tick off a native speaker such as, presumably, yourself.
- zipy124 9mo agoIt is based upon context, you are correct that it is ambiguious, as is the problem of most natural language. -I believe that <inference systems not using the Python stack> (which I do not appreciate) are a way to free open models usage and make AI more accessible. This reading of the text would lead one to believe they don't appreciate inferences systems not written in python. Given the inference system produced by the author is also not using the python stack (it is in C), we can assume this is not the correct reading. -I believe that inference systems not using the <Python stack> (which I do not appreciate) are a way to free open models usage and make AI more accessible. This reading says that the author does not like the python stack for inference, which given the author has produced this inference in C, would support the statement. That is we have to take both readings and think which one fits the context around it, hopefully this helps :)
- treksis 9mo agohow fast is this compare to python based?
- rcarmo 9mo agoThe Python libraries are themselves written in C/C++, so what this does performance-wise is, at best, cutting through some glue. Don't think about this as a performance-driven implementation.
- antirez 9mo agoVery slow currently, I added the benchmarks in the README. To go faster it needs to implement inference faster than the current float32-only kernels.
- throwaway314155 9mo agoPyTorch MPS is about 10x faster per the README.md.
- ChrisArchitect 9mo agoRelated: FLUX.2 [Klein]: Towards Interactive Visual Intelligence https://news.ycombinator.com/item?id=46653721 https://news.ycombinator.com/item?id=46653721
- llmidiot 9mo ago[flagged]
- antirez 9mo agoOne of the most important thing to do right now to redistribute something to the society, is to use AI to write free software: more free software than ever. If AI will be hard to access in the future, the more software it is released free, the better. If instead things go well (as I hope), there will be just a multiplication of the effect of OSS using today and tomorrow AI. In any way, writing free software using AI is a good idea, IMHO. I believe LLMs are the incarnation of software democratization, which aligns very well with why I used to write OSS. LLMs "steal" ideas, not verbatim code, you can force them to regurgitate some verbatim stuff, but most of it is ideas, and we humans also re-elaborate things we saw and we avoid (like LLMs are able to do) to emit the same stuff verbatim. But the software can't be patented for very good reasons, and LLMs capture all this value that is not subject to intellectual property, and provides it to the people that don't have the right tools and knowledge. And, it allows people that can code, to code 100x more.
- hollowturtle 9mo ago> redistribute something to the society with a proprietary black box tool you pay a subscription for? that's nonsense
- airstrike 9mo agoYou can always run models locally? Local models will become cheaper and faster over time. Don't panic just yet
- hollowturtle 9mo agoWill this will that and never consider will not. As of now, observation made on evidence, it looks way more the latter to me.
- dkdcio 9mo ago
- holografix 9mo agoNo cuBLAS?
- re 9mo ago> I wanted to see if, with the assistance of modern AI, I could reproduce this work in a more concise way, from scratch, in a weekend. I don't think it counts as recreating a project "from scratch" if the model that you're using was trained against it. Claude Opus 4.5 is aware of the stable-diffusion.cpp project and can answer some questions about it and its code-base (with mixed accuracy) with web search turned off.
- antirez 9mo agoThe two projects have literally nothing in common. Not a line of code, not the approach, nor the design. Nothing. LLMs are not memorization machines that recall every project in the cut & paste terms you could think of.
- deleted 9mo ago[deleted]
- falloutx 9mo agoI dont understand, so its just to generate the pic using a model. Isn't that trivial, whats the advantage of doing it in C? Is the model running in C? Readme is overly verbose and It seems like a project that just does one task and it costed the author $80.
- fabmilo 9mo agobecause of the principle: you only understand what you can create. You think you know something until you have to re-create it from scratch.
- gbalduzzi 9mo agoYes, the model runs in C, you just provide the model weights to the program. The main advantage is that you don't need the python interpreter to run the program. While not revolutionary, it is definitely not trivial and its main purpose is to demonstrate Claude code abilities in a low level, non trivial task.
- lovasoa 9mo agoThe author of this project is also the author of redis. He knows what he is doing. Running inference for a model, even when you have all the weights, is not trivial.
- deleted 9mo ago[deleted]
- antirez 9mo agoWhy it costed me $80? I pay the monthly subscription, and during that time I use it for multiple things. I spent one day of Claude Max on that so it costed me 2 euros and 60 cents.
- falloutx 9mo agoI may have misread that. Anyway, its 2.60 euros plus value of your time unless it was done with ralph agents.
- abecedarius 9mo agoA suggestion born of experience: besides printing the seed for an image, add it to the image file as metadata. Otherwise, if you're me, you'll lose it.
- ThrowawayTestr 9mo agoComfyui does that automatically
- antirez 9mo agoGood idea, noted in the TODO.
- mmastrac 9mo agoIs it just my connection or is the huggingface downloader completely broken? It was saturating my internet connection without making any progress whatsoever. EDIT: https://github.com/bodaay/HuggingFaceModelDownloader https://github.com/bodaay/HuggingFaceModelDownloader seems to be making progress.
- jabedude 9mo agoSalvatore, how did you pick up the requisite background knowledge on this subject? IIRC this is your first OSS project in the ML domain, just curious if/how much Claude was helpful with providing you with domain expertise while building this engine
- antirez 9mo agoHello, I always used to play with AI. I wrote this, some time ago, just to make an example: https://github.com/antirez/gguf-tools https://github.com/antirez/gguf-tools And I have a YouTube channel mostly about AI (in Italian language) where I regularly post videos and often read papers that I then explain in the channel. I have a long time passion about AI, I wrote my first NN implementation in 2003 (used here, many years ago, as a showcase of Redis modules https://github.com/antirez/neural-redis https://github.com/antirez/neural-redis), and never stopped since there to implement, for fun, small GPT models and things like that, using PyTorch or C. Also my work at Redis Vector Sets, in the latest year, exposed me more to working with models (especially text embedding models of many kinds, but also other models). So while Claude was fundamental to get the implementation fast, I had background to have idea about what was happening in the different stages. I believe it is a very interesting question to understand if this kind of work can be made with programming background and near-zero AI background. My feeling is that you ned more time, more back and forth, maybe to provide the agent with more examples, but eventually it will do something working.
- bakkoting 9mo agoNeat! I wonder how slow this would be running in wasm. In my dream world it would also use WebGPU but that's a much bigger lift.
- jitl 9mo agoREADME says it’s optimized for Metal, if it really is using metal compute shader, apparently the programming model is fairly similar to WebGPU. You could try asking Claude to translate it :)
- khimaros 9mo agohttps://github.com/leejet/stable-diffusion.cpp https://github.com/leejet/stable-diffusion.cpp
- kristianp 9mo agoNote that the original FLUX.2 [klein] model [1] and python code was only released about 3 days ago (inexact without knowing the times and timezones involved.) Discussed at [2] [1] https://bfl.ai/blog/flux2-klein-towards-interactive-visual-intelligence https://bfl.ai/blog/flux2-klein-towards-interactive-visual-i... [2] https://news.ycombinator.com/item?id=46653721 https://news.ycombinator.com/item?id=46653721
- p1esk 9mo agoI wonder how long it would have taken antirez without opus
- filipstrand 9mo agoReally cool project! Impressive to see this being done in pure C. I'm the maintainer of MFLUX (https://github.com/filipstrand/mflux https://github.com/filipstrand/mflux) which does a similar thing, but at a higher level using the MLX framework optimised for Apple Silicon. I just merged Flux 2 Klein support as well and was happy to see this discussion :) I started out doing this type of work roughly 1.5 years ago when FLUX.1 was released and have been doing it off and on ever since with newer models, trying to use more and more AI over time. At one point, I vibe-coded a debugger to help the coding agent along. It worked OK but as models have gotten stronger, this doesn't really seem to be needed anymore. My latest version simply has a SKILL.md that outlines my overall porting strategy (https://github.com/filipstrand/mflux/blob/main/.cursor/skills/mflux-model-porting/SKILL.md https://github.com/filipstrand/mflux/blob/main/.cursor/skill...). Somewhat surprisingly, this actually works now with Cursor + Codex-5.2, with little human intervention. > Even if the code was generated using AI, my help in steering towards the right design, implementation choices, and correctness has been vital during the development. This definitely resonates! Curious to hear more about what worked/didn't for you. A couple of things I've found useful: - Porting the pipeline backwards: This is the way I did it personally before using any coding models. The typical image generation flow is the following: 1.Text_encodings (+ random_noise_latent) 2.Transformer loop 3.VAE decoding I found that by starting with the VAE first (by feeding it pre-loaded tensors from the reference extracted at specific locations) it was the quickest way to get to an actual generated image. Once the VAE is done and verified, only then proceed backwards the chain and handle the Transformer, etc. I still prefer to do it this way and I like to manually intervene between step 3,2 and 1, but maybe this won't actually be needed soon? - Also, with the VAE, if you care about implementing the encoding functionality (e.g to be used with img2img features), the round-trip test is a very quick way to verify correctness: image_in -> encode -> decode -> image_out : compare(image_in, image_out) - Investing in a good foundation for weight handling, especially when doing repeat work across similar models. Earlier coding models would easily get confused about weight assignment, naming conventions etc. A lot of time could be wasted because weight assignment failed (sometimes silently) early on.
- antirez 9mo agoThanks, super interesting remarks. Funny enough I also went the route of implementing it backward :D Initially I totally skipped the text encoding part. Used Python to generate binary blogs. Also the easy 1 step test with image output correctness that you can implement immediately, with fixed seed and max pixel difference bound to floating point approximation errors, is super useful to provide the model with an immediate correctness feedback after a change.
- DamianLewis 9mo ago[dead]
- cboyardee 9mo ago[dead]
- kurtis_reed 9mo agoConfusing to have project name same as file name
- antirez 9mo agoThis is to follow the naming of llama.cpp
- imranq 9mo agoJust because it is in C, doesn't mean you will get C like performance. Just look at the benchmarks, it is 8x slower than just using PyTorch... while I get its cool to use LLMs to generate code at this level, getting super high performing optimized code is very much out of the domain of current frontier LLMs
- jrk 9mo agoThe PyTorch version is using the GPU (with Metal Performance Shaders); this C version is currently using (in the docs I saw) a single CPU core, with AMX (via Apple Accelerate BLAS) but not yet with OpenMP for parallelism. It’s not slow because LLM code is bad, but because it’s not running on the same hardware. That said, it’s also not as fast as it is because of the LLM—all the critical code is in kernel libraries it calls (the same as for PyTorch).
- nbardy 9mo agoNo it’s not. I have written cuda kernels and 8bit optimizers with this. They’re actually very good at speed optimization and can iterate very quickly taking notes on trials and failures and benchmarks. I’ve had it write 10 different attempts in around an hour and benchmark them all then merge and beat very strong baselines in torch
- antirez 9mo agoAbsolutely true, but now I'll focus on making it fast and I believe it will be possible to go much faster. I left the agent working in the night with a specification and now I'm going to see the progresses and restart the work.
- fulafel 9mo agoInteresting that OpenBLAS and MPS are reportedly nearly the same speed although the README sounds like only MPS uses the GPU.
- antirez 9mo agoI think that this is because the current code does a terrible job at taking the activations in the GPU and fusing the kernels. This is the next thing to fix in this implementation indeed.
- 1vuio0pswjnm7 9mo ago"I believe that inference systems not using the Python stack (which I do not appreciate) are a way to free open models usage and make AI more accessible."
- antirez 9mo agoUpdates: 1. Now it is much faster, and the Python benchmarks were re-done correctly (the benchmark didn't account for model loading, and did warm-up before starting the actual inference, while the C code was tested exactly in the reverse way). 2. Now there is --mmap support to run on Linux with blas target with 16GB of RAM. Inference is viable on my old-ish Dell Latitude i5. 3. Seed now part of the PNG metadata. 4. Many other improvements, check the README.
- scottmf 9mo agoI independently did the same with an MLX implementation on Sunday (also with Claude Code). I expected this C implementation to be notably faster, but my M3 Max (36GB) could barely make it past the first denoising step before OOMing (at 512x512) Am I doing something wrong? The MLX implementation takes ~1/sec per step with the same model and dimensions: https://x.com/scottinallcaps/status/2013187218718753032 https://x.com/scottinallcaps/status/2013187218718753032
- thasso 9mo ago> I believe that inference systems not using the Python stack (which I do not appreciate) are a way to free open models usage and make AI more accessible. What's wrong with the Python stack? I have never much used it or any other ML stack so I'm genuinely curious.
- peter_d_sherman 9mo ago>"This program generates images from text prompts [...] using the [data from] FLUX.2-klein-4B [...] and is implemented entirely in C, with zero external dependencies beyond the C standard library." You had me at image generation! Pure C with zero external dependencies -- is just an extra added bonus... No, actually, pure C with zero external dependencies -- is quite awesome in its own right! Well done!