Links for jax-neuron