linear_operator icon indicating copy to clipboard operation
linear_operator copied to clipboard

[Proposal] Remove jaxtyping dependency

Open Balandat opened this issue 2 months ago • 2 comments

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/Long type classes to keep the jaxtyping syntax.

@gpleiss, @corwinjoy, any thoughts?

Balandat avatar Oct 08 '25 00:10 Balandat

@gpleiss, @corwinjoy - speak now or forever hold your peace.

Balandat avatar Oct 18 '25 21:10 Balandat

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.

corwinjoy avatar Oct 18 '25 22:10 corwinjoy