Add more linear algebra ops

[marcelluethi]

Dimwit currently lacks many important linear algebra operations, such as svd, cholesky, ...
These would be simple to add, as the underlying jax implementation already supports them.