Improve compilation times #23
Description
Compilation times are OK when using caustics on its own but they're completely unacceptable when caustics is incorporated within a numpyro model.
{{ message }}
Compilation times are OK when using caustics on its own but they're completely unacceptable when caustics is incorporated within a numpyro model.