jax

06d41fba - Added a note about squared distances