michaelg/dev: links for jax-lightning