BUGFix: Wrap roll as OpFromGraph with custom JAX dispatch to jnp.roll#1900
Open
ayulockedin wants to merge 1 commit intopymc-devs:mainfrom
Open
BUGFix: Wrap roll as OpFromGraph with custom JAX dispatch to jnp.roll#1900ayulockedin wants to merge 1 commit intopymc-devs:mainfrom
ayulockedin wants to merge 1 commit intopymc-devs:mainfrom
