jax

a45e2837 - add back a full_lower, dropped in #3491 (#3530)