jaxfit icon indicating copy to clipboard operation
jaxfit copied to clipboard

Published 20 hours ago • verified publisher icon Dipolar-Quantum-Gases

Using vmap

Open felavila opened this issue 1 year ago • 2 comments

Hi, I want to perform an vmap iteration over an array, but the "TracerArrayConversionError" appears. I want to ask if it is possible to use this kind of operation or not. Thanks for your answer.

felavila avatar Jun 26 '23 14:06 felavila

Currently, the curve fit function includes control code which is not JIT compatible. This was a design choice to maintain all of the same functionality from SciPy's curve_fit function

In the future a simplified JIT compatible version will be added.

lucashofer avatar Jul 03 '23 12:07 lucashofer

Thanks for you answer. Cheers

felavila avatar Jul 05 '23 20:07 felavila