def _lu(operand: TfVal, _in_avals, _out_aval):
    return _convert_jax_impl(lax_linalg._lu_python)(operand, _in_avals=_in_avals, _out_aval=_out_aval)
def _lu(operand: TfVal, _in_avals, _out_aval):
    return _convert_jax_impl(lax_linalg._lu_python)(operand, _in_avals=_in_avals, _out_aval=_out_aval)