jax

850e8a75 - fix typo in linearize docstring