revalo / tree-diffusion
Diffusion on syntax trees for program synthesis
☆417Updated 4 months ago
Related projects ⓘ
Alternatives and complementary repositories for tree-diffusion
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆313Updated 4 months ago
- LLM verified with Monte Carlo Tree Search☆249Updated 2 months ago
- Visualizing the internal board state of a GPT trained on chess PGN strings, and performing interventions on its internal board state and …☆192Updated 5 months ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆332Updated last week
- Reasoning Computers. Lambda Calculus, Fully Differentiable. Also Neural Stacks, Queues, Arrays, Lists, Trees, and Latches.☆233Updated this week
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆176Updated 5 months ago
- Draw more samples☆174Updated 4 months ago
- The boundary of neural network trainability is fractal☆161Updated 8 months ago
- Simple Transformer in Jax☆115Updated 4 months ago
- Open weights language model from Google DeepMind, based on Griffin.☆606Updated 4 months ago
- a curated list of data for reasoning ai☆111Updated 3 months ago
- A compositional diagramming and animation library as an eDSL in Python☆209Updated 4 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆55Updated 3 months ago
- a small code base for training large models☆264Updated last week
- The history files when recording human interaction while solving ARC tasks☆94Updated this week
- Reverse Engineering the Abstraction and Reasoning Corpus☆184Updated last month
- DiscoGrad - automatically differentiate across conditional branches in C++ programs☆204Updated last month
- ☆197Updated 3 months ago
- ☆99Updated 3 months ago
- Gradient descent is cool and all, but what if we could delete it?☆101Updated this week
- ☆235Updated 7 months ago
- Create feature-centric and prompt-centric visualizations for sparse autoencoders (like those from Anthropic's published research).☆157Updated last month
- Domain Specific Language for the Abstraction and Reasoning Corpus☆206Updated 3 weeks ago
- ☆105Updated 3 months ago
- A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…☆270Updated last week
- Stop messing around with finicky sampling parameters and just use DRµGS!☆317Updated 5 months ago
- Synthesizing Graphics Programs for Scientific Figures and Sketches with TikZ☆483Updated this week
- A library for making RepE control vectors☆478Updated last month
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆332Updated 3 months ago