JTan comments

Repositories
Issues
Comments

Results 33 comments of


                                            JTan

Using `vmap` for root finding with a vector of parameters

> Hello! Could you send a minimal code example? I'm happy to investigate that. Which method are you using? Hi, I'm running into two different errors, with and w/o `vmap`....

Using `vmap` for root finding with a vector of parameters

Thanks both, any pointers on how I can batch a problem where I have multiple root finding problems involving a vector of parameters in Jax? I've tried treating it as...

Using `vmap` for root finding with a vector of parameters

I managed to get it to work, but it's still rather slow. I'll try 'pmap'ing it over cores to see if that helps. I couldn't get it to work with...