jax

c56e794a - doc: fix description of logsumexp axis