exojax icon indicating copy to clipboard operation
exojax copied to clipboard

Published 20 hours ago • verified publisher icon HajimeKawahara

vmap vs scan in xsmatrix

Open HajimeKawahara opened this issue 2 years ago • 0 comments

During implementation of PreMODIT, I found that jax.lax.scan in modit.xsmatrix can be replaced by vmap. But not sure everything is OK and/or vmap is better than scan. I will check this later.

HajimeKawahara avatar Jun 09 '22 07:06 HajimeKawahara