revalo / tree-diffusion
Diffusion on syntax trees for program synthesis
☆451Updated 9 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☆270Updated last month
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…☆605Updated last week
- A library for making RepE control vectors☆562Updated 2 months ago
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆366Updated 9 months ago
- Reasoning Computers. Lambda Calculus, Fully Differentiable. Also Neural Stacks, Queues, Arrays, Lists, Trees, and Latches.☆249Updated 4 months ago
- Simple Transformer in Jax☆136Updated 9 months ago
- The history files when recording human interaction while solving ARC tasks☆103Updated this week
- run paligemma in real time☆131Updated 10 months ago
- Code to train and evaluate Neural Attention Memory Models to obtain universally-applicable memory systems for transformers.☆300Updated 5 months ago
- Sparsify transformers with SAEs and transcoders☆499Updated this week
- ☆123Updated 3 weeks ago
- Visualizing the internal board state of a GPT trained on chess PGN strings, and performing interventions on its internal board state and …☆203Updated 4 months ago
- Visualize the intermediate output of Mistral 7B☆348Updated 2 months ago
- Getting crystal-like representations with harmonic loss☆180Updated last month
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆399Updated 2 weeks ago
- Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…☆281Updated last month
- A compositional diagramming and animation library as an eDSL in Python☆216Updated 4 months ago
- Stop messing around with finicky sampling parameters and just use DRµGS!☆347Updated 10 months ago
- ☆106Updated 3 months ago
- DiscoGrad - automatically differentiate across conditional branches in C++ programs☆204Updated 6 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…☆283Updated 3 months ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆250Updated last year
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆187Updated 10 months ago
- ☆242Updated last year
- Bootstrapping ARC☆105Updated 4 months ago
- Open weights language model from Google DeepMind, based on Griffin.☆629Updated last month
- ☆65Updated 2 weeks ago
- Puzzles for exploring transformers☆335Updated last year
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆342Updated 8 months ago
- a categorical deep learning compiler☆199Updated last month