This repository accompanies the NeurIPS paper “HYSYNTH: Context-Free LLM Approximation for Guiding Program Synthesis.”
HYSYNTH learns a probabilistic context-free grammar (PCFG) that approximates the distribution of programs produced by an LLM and uses those weights to guide search.
- ARC DSL grammar —
dsl/v0_3/dsl.lark
- PCFG estimation (ARC) —
compute_pcfg.py
- ARC synthesis runner —
run_synthesize.py
- TF-Coder PCFG estimation —
tf_coder/tfcoder_pcfg_modified.py
- Probe PCFG — Probe