Add test case for jit compile of vmap on gpu#25980

Open

codinglover222 wants to merge 4 commits intojax-ml:mainfrom codinglover222:jit-vmap-compile-test

+27

Commits on Jan 18, 2025

Add test case for jit compile of vmap on gpu.
codinglover222
committed

Commits on Jan 20, 2025