jax

e680304d - Remove warning suppression for tuple and list arguments to reductions. (#3545)