revert changes + update docstring

15 jobs for fixup-JaxLinearOperator in 21 minutes and 10 seconds (queued for 2 seconds)