Sasha Rush Brings Lean Formal Verification to JAX Code

Princeton NLP researcher Sasha Rush released Provably Correct Tensor Puzzles, building a Jax-to-Lean transpiler that translates JAX code into the Lean theorem prover for formal verification of tensor puzzles and arbitrary ML code.

2026-10-09 ~ 2026-10-09 · 2 related posts

• srush builds Jax-Lean transpiler to formally verify JAX tensor code — srush_nlp · 2026-10-09
• srush's Provably Correct Tensor Puzzles: verifying ML code with a Jax-to-Lean transpiler — srush_nlp · 2026-10-09