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 to jit=False while LinearOperator defaults to jit=True
- For the most part there is no jitting of
Functional classes, with the exception of L0Norm.prox.
- Refactoring of methods in
scico.optimize classes to support jitting should be explored.
A coherent policy for the use of
jax.jitneeds to be formulated and implemented. At present, its usage seems somewhat arbitrary and inconsistent; for example:Operatordefaults tojit=FalsewhileLinearOperatordefaults tojit=TrueFunctionalclasses, with the exception ofL0Norm.prox.scico.optimizeclasses to support jitting should be explored.