revalo / tree-diffusion
Diffusion on syntax trees for program synthesis
☆442Updated 7 months ago
Alternatives and similar repositories for tree-diffusion:
Users that are interested in tree-diffusion are comparing it to the libraries listed below
- LLM verified with Monte Carlo Tree Search☆264Updated 2 weeks ago
- Reasoning Computers. Lambda Calculus, Fully Differentiable. Also Neural Stacks, Queues, Arrays, Lists, Trees, and Latches.☆246Updated 3 months ago
- Visualizing the internal board state of a GPT trained on chess PGN strings, and performing interventions on its internal board state and …☆200Updated 3 months ago
- A compositional diagramming and animation library as an eDSL in Python☆217Updated 2 months ago
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…☆604Updated 2 months ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆391Updated this week
- Bootstrapping ARC☆100Updated 3 months ago
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆358Updated 8 months ago
- Simple Transformer in Jax☆136Updated 7 months ago
- Sparsify transformers with SAEs and transcoders☆461Updated this week
- a small code base for training large models☆286Updated 2 months ago
- The history files when recording human interaction while solving ARC tasks☆97Updated this week
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆342Updated 6 months ago
- DiscoGrad - automatically differentiate across conditional branches in C++ programs☆203Updated 5 months ago
- Open weights language model from Google DeepMind, based on Griffin.☆620Updated 7 months ago
- Visualize the intermediate output of Mistral 7B☆339Updated 3 weeks ago
- Domain Specific Language for the Abstraction and Reasoning Corpus☆236Updated 4 months ago
- Tensor Network Library with Autograd☆256Updated this week
- a curated list of data for reasoning ai☆128Updated 6 months ago
- Puzzles for exploring transformers☆332Updated last year
- Reverse Engineering the Abstraction and Reasoning Corpus☆229Updated 4 months ago
- ☆211Updated 7 months ago
- Stop messing around with finicky sampling parameters and just use DRµGS!☆343Updated 8 months ago
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆185Updated 8 months ago
- A library for making RepE control vectors☆551Updated last month
- Mistral7B playing DOOM☆127Updated 7 months ago
- ☆239Updated 11 months ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆250Updated last year
- a categorical deep learning compiler☆194Updated 9 months ago