Fix faulty interaction between jax.vmap
and validate_args=True
#1686
Merged
fehiepsi merged 2 commits intopyro-ppl:masterfrom pierreglaser:vmap-validate-argsNov 25, 2023
+21-2
jax.vmap
and validate_args=True
#1686