equinox
equinox copied to clipboard
Unroll in eqx's internal `scan`
In the eqx.internal.scan
do we not have a way to unroll
the scan
? I'm not sure what the constraint on the "checkpointed" scan
since I'm not familiar with the algorithms used there, but I assumed people will want to use unroll
in the lax
styled scan
?