Sasha Rush
@srush_nlp
@sakshjn Currently the Lean jaxpr doesn’t yet support symbolic dims, but mostly for simplicity. That specific proof and section came from the previous blog which did it length agnostically.
Designwise, I would like it if people had clean length agnostic specs and then on trace it also
