6 ms·
Auto-research with codex: How I achieved a 232x Faster Kernel
- Almondsetat 2mo agoIn the last couple of days I wanted to try out the new definitive DeepSeek v4 releases. I gave it the repository of a semi-abandoned video compression codec and I told it to perform the usual benchmark -> profile -> verify -> research -> improve loop. I specifically chose this codec because the authors include a verifier for the bitstream to make sure you don't break stuff if you want to try your own implementation. I gave the agents access to the compiler's profiler and also Intel's VTune, which has fantastic output. In a couple of hours the LLM generated SSE and AVX implementations of the compression and decompression algorithms that almost doubled performance with a single core. Then I asked it to create a CUDA implementation using NVIDIA's NSIGHT profiler as a guide and it also started doing some good work. Personally, I believe that LLMs should be treated like an advanced version of Prolog or linear programming: you give the constraints, you have a way of verifying correctness, and you give it a clear goal. If the LLM can verify itself and course-correct you can basically leave it on autopilot
- rrhjm53270 2mo agoI tried kernel autoreasearch using DeepSeek-V4-Flash as well. It spent about 1-2 hours to complete the FlashAttention optimization job (https://github.com/fengwang/FA5090/tree/main/v7 https://github.com/fengwang/FA5090/tree/main/v7) and cost me only $0.2. I believe we are ready to offload a lot of this kind well-defined constrained optimization problems to AI Agent autoresearch.
- embedding-shape 2mo ago> I tried kernel autoreasearch using DeepSeek-V4-Flash as well. It spent about 1-2 hours to complete the FlashAttention optimization job Doing the same, re-implementing a lot of LLM/diffusion models in Rust+CUDA for my own usage, usually the initial implementation takes 1-2 days (of 100% autonomous work) then I put an agent to optimize the implementation which tends to get close to SOTA performance within another day or two. As long as you can point the agent at "This is the correct baseline, make sure any optimizations still pass this", seemingly you can leave them and they come back after N hours with a faster program that just works.
- josephg 2mo agoIf you have access to SOTA implementations, LLMs can often find, understand and copy their best optimisation ideas.
- _zoltan_ 2mo agoThis is exactly how I use it. I mean not on abandoned repos, but in a benchmark - profile - verify - research - improve loop.
- thebruce87m 2mo agoWhat’s the best way to loop it?
- _zoltan_ 2mo ago6 months ago I would have said the ralph loop plugin. with opus 5, I'm finding that asking itself to write such a loop skill that does benchmark - profile - verify - research - improve, keep logs for each iteration, etc. (just like autoresearch), and then using the skill, works very nicely.
- eterm 2mo agoI did something similar recently with Google's C# protobuf library. I had spotted I was getting CPU bound rather than memory bandwidth bound when doing streaming of uint32 buffers in dotnet gRPC. I then asked claude to compare the C#/.NET implementation in the library with the C++ version, and it quickly identified that the C# library was missing a couple of fairly cheap optimisations that were present in the C++ version. If I can help get a PR merged, then it'll be by far the biggest impact of any work I've ever done. I also compared the Rust version, it had this specific optimisation. The far more popular Tokio/Prost library did not. Given appropriate guardrails, LLMs are impossibly fast at iterating to find root causes and specific performance bottlenecks.
- Almondsetat 2mo agoYou presented another thing LLMs excel at: integrating something from a project that is not present in another one. I think they work so well at this because both the starting and ending points have an already existing structure, so the LLM can guide itself effectively. In your case it's even more egregious because we are talking about the same exact algorithm/functionality implemented in two different, but rather similar, programming languages. Could you have manually profiled and compared the execution paths? Sure. Could you have translated the C++ optimizations to C#? Sure. But in such an obvious case, the LLM managed itself.
- josephg 2mo agoYeah. LLMs are also great at reading and understanding thousands of lines of code quickly. It can find things like this - or compare disparate implementations. I’ve been using fable recently not to write code, but to write briefing notes on projects. “Read and compare these three libraries. What are the unique differences in their implementations? Why were those ideas chosen? What are the pros and cons of those designs? What ideas would you copy into a new project? Write a report on your findings.”
- qarl2 2mo agoI've had a lot of success decompiling old video game ROMs in exactly this way. Like you say - give it a way of verifying correctness - put it in a loop - and they are quite surprising. https://github.com/qarl/arcade-js https://github.com/qarl/arcade-js
- BlackRabbit1 2mo agoSame. I love reverse engineering embedded stuff. Even the cheap LLMs are great in doing the awful crud work in the beginning: finding offsets, firmware update file structures, brute forcing checksums, etc. It still produces a lot of crap in the later steps (understanding the implementation itself) but I'm happy doing this stuff myself then.
- qarl2 2mo ago> It still produces a lot of crap in the later steps (understanding the implementation itself) I've had success here by adding a phase called "grounding" that attempts to verify its "understanding" by creating tests that modify the running executable to ensure its made the right inference. Is this variable really MARIO_X? Change it and see if Mario moves. Etc. As an example in Donkey Kong - the system had trouble deciding if an array controlled barrels or fireballs. There was conflicting evidence. After many trips through the loop - it realized it does BOTH, depending on which level you're on. So the "understanding" grows with each iteration.
- kenerwin88 2mo agoOh wow, this is almost exactly what I’ve been doing with Zelda LTTP. I have it in rust now, but just finished the “first pass” you reference. Mine is still not really readable, second step is the modernizing the actual code. I’ve really struggled with needing to handhold it though, I’ll see if I can plagiarize from you!
- qarl2 2mo agoI'm working on getting the "handholding" down to zero. Frogger is almost done and I haven't had to intervene once.
- porridgeraisin 2mo agoThis is the way. Checkout the technique mentioned in the alphaevolve paper and see if it works well for your setting.
- Almondsetat 2mo agoThank you for the reference, I hope to be smart enough to try it out!
- porridgeraisin 2mo agoIt's a straightforward "prompting" + single evolutionary algorithm technique, The paper looks like well, a paper, but the actual thing is simple.
- tiahura 2mo agoI'm hoping to release a native Wine for MacOS w/ Win32 FEX support in the next few days. Watching claude and codex play winquake and age of empires, and debug support for Firefox 52 has been wild.
- poizan42 2mo agoI have used Opus 5 and some Fable 5 to finally get realtime transcoding of 4K 10-bit HEVC (to 1080p or smaller SDR AVC) working on a Raspberry Pi 4. It was very good at writing optimized NEON kernels. the Argon HEVC hardware decoder outputs SAND30 which is a tiled format that is annoying to work with and not really supported by anything else, the big performance issue has been with converting and scaling it, but as it turned out a lot of it was really with just moving memory around, so by fusing multiple steps into a single kernel it became fast enough. Experimenting with writing the NEON kernels for the different combinations would have taken forever by hand. I do wonder if auto-research would have reached something similar, it did take a significant amount of steering from me to get it to the point where it was working realtime. For anyone interested the ffmpeg is at https://github.com/poizan42/jellyfin-rpi-ffmpeg https://github.com/poizan42/jellyfin-rpi-ffmpeg and a shim for using it with stock jellyfin at https://github.com/poizan42/jellyfin-rpi-ffmpeg-shim https://github.com/poizan42/jellyfin-rpi-ffmpeg-shim
- worldthruword 2mo agoIs Mojo programming language useful in the age of AI?
- codetiger 2mo agoHad a similar experience with my Rust implementation for JSONLogic expression evaluation engine. As it has a full test suite with 1000s of cases and a benchmarking script, I was able to give some basic hints to try different optimization techniques and the end result was impressive. Reached from 1.6s to 200ms for a full benchmarking test. https://github.com/GoPlasmatic/datalogic-rs https://github.com/GoPlasmatic/datalogic-rs First 3 versions were hand written and maintained for 3yrs, and now 4th version came out in less than a month's time with impressive performance.
- alfiedotwtf 2mo ago> Personally, I believe that LLMs should be treated like an advanced version of Prolog or linear programming: you give the constraints, you have a way of verifying correctness, and you give it a clear goal. If the LLM can verify itself and course-correct you can basically leave it on autopilot This! People keep measuring how good AI is by one-shotting a problem, but I e found that a back and fourth via the Superpowers SKILL.md is how the model builds these constraints… then when it has all the holes, it gets to work
- michaelmrose 2mo ago[dead]
- 3abiton 2mo agoIt's very much how I've been using claude code in the past 5 month: brainstorm -> generate specs (constraints) -> generate exact plan -> implementation + review -> test / validate. The last stage is the most crucial one, and it's the most difficult to get it right for a complex solution (it's tough to cover every variation). But so far, it's been working great..
- Jackobrien 2mo agoDamn! If a solo engineer can do this, it makes the most around OAI/Anthropic start to look pretty weak.
- dzbarsky 2mo agoThis was nowhere near the top submission. But even if a solo engineer could get a top kernel, you don't think that having thousands of engineers, infinite tokens, and stronger models than are available to the public would give the labs a significant edge?
- genxy 2mo agoDoes the edge matter? I know you added significant as your hedge, but once you have feedback, your gain is largely irrelevant. Gain buys you bandwidth, so we are constructing systems run by the most powerful corporations where they are now optimizing for latency, as Archer says, do you want to flash crash civilization? This is how you do it.
- datakan 2mo agoI don't know. That just sounds like throwing money at a problem until it goes away. I'm not convinced that is the correct path forward.
- qeternity 2mo agoYou say this like that isn’t how the vast majority of problems are solved…
- dejavucoder 2mo ago1. labs have lots of inference capacity 2. they will have domain experts working on this so their efficiency is gonna be exponentially more (can direct LLM better, save money, reach same results faster)
- fooblaster 2mo ago
- tosh 2mo agoTraining material seems to be especially rich re GPU kernels and SIMD. I wonder if there is extra effort put into this because they are useful for the researchers working on the models or just a sub-domain that language models are a great fit for and humans have trouble with?
- dsign 2mo agoAnecdotally, I saw Opus 5 come up with a complicated loop unrolling technique when I asked it to implement a simple biquad in SIMD, missing a simpler solution. Maybe it was a downgraded session, who knows. That SIMD instruction set, the one for the ESP32-P4, is not very popular and all the documentation it has is a couple of blog posts. So I'm pretty sure it has at most seen some code for a predecessor during its training. However, the LLM was able to derive a full listing of the operations and their arguments from gcc to get us started, and that's why I was able to come up with my own implementation. Along the way, it also came up with insights about possible gotchas. Then, when implementing algorithms, it has been able to reason things out and get things working, despite the ISA not being extremely well known.
- msdz 2mo ago> Maybe it was a downgraded session, who knows. Off-topic, but imagine us collectively being okay with (or powerless to do anything about) this sentiment about any other software service provided like two years ago. How the times have changed…
- dejavucoder 2mo agowe live in exciting and scary times...
- soulofmischief 2mo agoWere other services so historically resource-constrained?
- msdz 2mo ago
- spacemanspiff01 2mo agoThis is really cool - I really like the beam search idea,
- oinoom 2mo agothis is the first time ive heard of beam search. i would have reached for a genetic algorithm of some sort, although it seems like some stochastic versions of beam search exist to avoid local minima. i wonder if there are any good frameworks for building these that agents can construct and use.
- dummydummy1234 2mo agoI think the question is how do you keep track of the ideas that the agent is pursuing - like I was working on this for implementing a fft, and doing the optimizations, but I held its hand and was like - hey let's go back and retry this older thing you discounted because of a 3 % slowdown.
- kimjune01 2mo agobeam search works when you have enumerable branches or some predefined dimension
- shken 2mo agoEvery step here has an oracle: wall-clock, the profile, pass or fail from the verifier. I had an agent-built app audited task by task, 10 came back done and 7 worked, and the three misses were the ones needing a credential or a setting on someone else's dashboard. Nothing in the loop could tell the agent it had failed, so it said done and moved on.
- phoghed 2mo agoI’ve tried it on simple UI tasks. Give it screenshot to work towards (or figma MCP), let it get screenshots from chrome to check its work. The existing models are surprisingly bad at it.
- iinnPP 2mo agoIt's really difficult to understand what your definition of surprisingly bad is. What was it continuously having problems with?
- phoghed 2mo agoJust reporting work as done that’s not even close mainly. Then if you manually screenshot both and ask if they are the same it’s like “yeah looks great boss”. The visual understanding of the models is just leagues worse than their text/language understanding. Or guessing colors rather than sampling from the image or pulling from figma is another stupid thing they do constantly.
- vickychijwani 2mo agoI imagine Gemini would do better on these types of tasks. Curious to try it out.
- ramon156 2mo agosome of these submissions seem to be omitting the actual rules. the #1 on edinh has a line that says "bypass ban check"
- dejavucoder 2mo agosubmission #2 by gau nernst is most numerically stable
- spwa4 2mo agoPeople are always going to hate auto-research and "loop engineering". Because it's got 2 properties: 1) it's the only way to get something out of models (or people for that matter) that they don't know yet. 2) it's harder to do with an LLM than without. Not easier. 3) and when you fuck it up, half the time the LLM (or other ML technique) makes a fool out of you and you spent $1000 to find the quickest way to get a robot leg on the ground is just to crash it into the ground.
- suddenlybananas 2mo agoWhat do you mean by 2?
- jurgenburgen 2mo agoLLMs will take shortcuts and cheat in ways that a human would realize are counterproductive.
- kilroy123 2mo agoI've seen this several times now. Disturbing. IMO, LLMs will be a dead end to anything close to AGI because of this and hallucinations. We're missing something in the mix, which I suspect is some kind of advanced JEPA model.
- iinnPP 2mo agoLLMs take shortcuts if allowed. Humans do it ignorantly. The LLMs will improve while average human IQ in the west dips closer and closer to the 80s on the global scale.
- _zoltan_ 2mo agoyou specify the goal. if the goal is achieved, it's achieved. the code the LLM writes will be read and maintained and developed further by LLMs. so it doesn't really matter what it produces as long as all the tests are green and it achieves exactly what you want it to achieve.
- myshapeprotocol 2mo ago[flagged]
- amarcheschi 2mo agoIsn't cholesky - used to substitute householder at a point - faster but less stable in some cases? I'm just recalling from memory since I had done a small project on qr decomposition with householder for an exam this year. I mean, if it is faster than the standard torch operation probably there are good reasons for which it is not the default standard torch operation. Might as well be wrong, I'm not sure
- cdavid 2mo agoFor once Cholesky is less general than QR: Cholesky only makes sense for positive matrices, while QR works for any matrix (including non square). Also QR is a primitive for operations like finding eigenvalues, and I don't think Cholesky can be used there.
- pranav_tech26 2mo ago[flagged]
- sqquima 2mo agoMeta commentary but it felt fresh to read a long wall of text that didn't seem to be AI generated. Thanks.
- dejavucoder 2mo agoauthor here! welcome! check out my featured section
- halJordan 2mo agoHow is this meta commentary when this or its negative version is present on literally every post.
- KuSpa 2mo agoFirst: Whataboutism, and second: Those are meta comments as well, they just don't declare themselves as one.
- RenSoft 2mo ago[flagged]
- lmeyerov 2mo agoIt's been fascinating doing a custom variant for GFQL, the first OSS embeddable Cypher property graph query engine for CPU+GPU - - accelerated launch of our new backends like polars, including a new lazy mode & planner, which are fundamentally new paths - while we initially aimed for top GPU benchmark scores, we now also maintain top CPU scores too! Long-term, more interesting to me is this opens rethinking what it means to be a query engine. Right now we are making it the fastest in general, especially on workloads from our own use, major industry benchmarks, and our users. At the same time, similar to jit and multistage computing, we're looking at new ahead-of-time optimization techniques users can do that are more interesting than plugging in custom indexes. Essentially, if our agents can do fast specializations, there should be safe hooks that we can expose to our user's agents too!
- deleted 2mo ago[deleted]
- huflungdung 2mo ago[dead]
- augment_me 2mo agoOne thing worth to note in the competition is that 8 out of the 10 top solutions, which all happened to be optimized this way completely broke at any other input than the competition ones. The only solutions that did not break when tested with OOD shapes were made by experts who know a lot about GPU programming and that did not create 25k lines of CUDA but followed and adjusted their solution in reasonable bounds. The takeaway from this is that these approaches will always solve for specificity, but it's a much harder task to steer the model into making general solutions. So if you're an inference provider for some specific model shape, fantastic, go for it. If you are a maintainer of a open-source library, this is not useful.
- josu 2mo agoOverfitting to the input is part of the meta in this type of challenges. The goal is not to create good, general or maintainable code. The only goal is to produce the fastest code.
- dejavucoder 2mo agofair argument
- piker 2mo agoNonetheless gives appropriate context to the speed up for folks like myself who were just skimming.
- dejavucoder 2mo agothis is true. in one of the later problems (cholesky decomposition), the organizer ran the submissions on a tiny training run to validate... and also provided code for same for our reference. most of the top solutions hit 4/8 or so. not very numerically stable. i found out that as i learnt more domain wise, i was (obviously) able to steer better. doing a re-write can also remove lots of slop and context rot (and subsequently make it easier for both human and LLM to make solution more numerically stable, less reward hackish)
- musebox35 2mo ago
- deleted 2mo ago[deleted]
- bla3 2mo agoMirrors my experience: LLMs are really good at optimizing, better than most humans. But also, they tend to not reach absolute peak performance where people made an effort to optimize something. Since most problems see fairly little optimization, that's still a big win most of the time.
- themeiguoren 2mo agoI've had pretty good luck with the following process for performance optimization loops: - Have an agent generate unit tests until it gets to 100% path (not just statement) coverage, with every numerical test asserting checks against golden values to prevent regressions - Let it rip on a performance improvement loop, for the widest E2E representative test case you have. Have it generate flamegraphs along the way so you can check in and steer it as necessary. - Optionally allow for 1 ULP changes in output values so that it doesn't kill itself getting bit-exact results. - Have it flag correctness errors as it goes, since your code probably isn't bug free. This is also how I've done language ports from python to rust, and having the ironclad test coverage protects you from drifting.
- spread2009 2mo ago[flagged]
- tizerluo 2mo ago[flagged]
- rldjbpin 2mo agoit is nice to see continued enthusiasm for kernel programming and optimization. however, it would be nice if they have more eyeballs at the rocm side of things. gpu mode does have some challenges for amd's platform, but it seems sparse in comparison.