Interface AotLinearizeOptions

Options for aotLinearize().

interface AotLinearizeOptions {
auxStore?: { value: any };
skipBackward?: boolean;
}

Properties

`Optional`auxStore

auxStore?: { value: any }

Mutable aux store from flattenFunWithAux. When provided, collectConcreteArrays(auxStore.value) is called AFTER the trace (which populates auxStore.value) to protect concrete aux arrays from PE intermediate disposal.

Must be passed as the store reference — NOT pre-computed arrays — because auxStore.value is only set during linearizeFlatUtil.

`Optional`skipBackward

skipBackward?: boolean

When true, skip building the backward jaxpr. The returned pullback will be a no-op stub. Use for linearizeFlat which only needs the forward jaxpr.