[Proposal] Remove jaxtyping dependency
jaxtyping was added as a dependency in https://github.com/cornellius-gp/linear_operator/pull/42. Since then I have not seen any updates that uses that functionality. But it caused some maintenance work both for this repo (https://github.com/search?q=repo%3Acornellius-gp%2Flinear_operator+jaxtyping&type=commits) and has also recently led to a breakage in our internal codebase b/c of some incompatibilities between jaxtypig and jax due to some numpy version issue. I have a strong sense that the added complexity and maintenance headache is not worth the trouble.
Proposal: Remove the jaxtyping dependency.
Options:
- Straightforward: Remove typing code and dependency outright
- Workaround: Manually create no-op
Float/Longtype classes to keep the jaxtyping syntax.
@gpleiss, @corwinjoy, any thoughts?
@gpleiss, @corwinjoy - speak now or forever hold your peace.
Removing this sounds reasonable to me. When I added these checks I was hoping for much stronger typing on the shapes and sizes. Unfortunately, many of the routines have quite variable output which blocks strong type checks. Also, the errors are not that easy to read. So, if you feel it is better removed, I understand.