jax

19201f4b - Mention in docstring when function is not implemented (#4297)