compute
TangentLinearOperator
lx.trace
diagonal(ComposedLinearOperator)
_identity_dtype
in_dtype
project