Add unroll argument to functions that use lax.scan.#150

carlosgmartin wants to merge 1 commit intogoogle-deepmind:maingoogle-deepmind/rlax:mainfrom

carlosgmartin:lax_scan_unrollcarlosgmartin/rlax:lax_scan_unrollCopy head branch name to clipboard