scico
scico copied to clipboard
Coherent policy needed for use of `jax.jit`
A coherent policy for the use of jax.jit
needs to be formulated and implemented. At present, its usage seems somewhat arbitrary and inconsistent; for example:
-
Operator
defaults tojit=False
whileLinearOperator
defaults tojit=True
- For the most part there is no jitting of
Functional
classes, with the exception ofL0Norm.prox
. - Refactoring of methods in
scico.optimize
classes to support jitting should be explored.