25 Jan
2022
25 Jan
'22
11:14 a.m.
21: def chunk_scanner(idx): idx comes from elements of jnp.arange() on line 31, which generates a vector of ascending integers starting at 0.