search

fbartolic / caustics

Differentiable microlensing powered by JAX
https://fbartolic.github.io/caustics/
MIT License
13 stars 2 forks source link

Experiment with the level at which @jit decorators are placed #12

Closed fbartolic closed 2 years ago

fbartolic commented 2 years ago

I currently use jit at the level of smallest functions. Performance could be improved by exploring the effect of JIT at higher level functions.

fbartolic commented 2 years ago

Doesn't improve performance at all but I've removed the JIT decorators from some functions anyway.