srush builds Jax-Lean transpiler to formally verify JAX tensor code
srush_nlp · x · 2026-10-09
- Sasha Rush upgrades his popular Tensor Puzzles to be "provably correct": a Jax-Lean transpiler translates JAX code into the Lean theorem prover, released as the srush/jax-lean library for proving properties of arbitrary JAX code.
- Key insight: proving arbitrary Python is nearly impossible, but JAX compiles code down to a minimal IR of numerical operations, where proofs become tractable.
- Inspired by Noether, Aeneas and Autodidax; a follow-up to Lean-Verified Transformers. Code and proofs written by AI, writing by human.
Related event: Sasha Rush Brings Lean Formal Verification to JAX Code(2 posts)→
More from coding & agent
- Agent Buddy: an open-source desk buddy that shows your AI coding agents' status — DanWahlin · 2026-10-09
- Developer's Grok Bot now natively integrates with X, no API or credits needed — daniel_mac8 · 2026-10-09
- AutoScientist's two-agent checklist loop auto-audits every training example — sarahookr · 2026-10-09
- Meta's KernelAgent uses multi-agent orchestration for 2.02x Triton kernel speedups — PyTorch · 2026-10-09
- Claude recovers lost 2019 build paths to recompile MakerDAO's DAI to an exact bytecode match — devanshmehta · 2026-10-09
- 6 models tested on real MCP servers: Opus 5.5 leads, open models cost 87% less per attempt — shensi · 2026-10-09