Improve `Graph` geometry

[michalk8]

Use jax.scipy.cho_{factor,solve}