def _threefry2x32_jax_impl(*args: TfVal, _in_avals, _out_aval):
    res = _convert_jax_impl(functools.partial(jax._src.random._threefry2x32_lowering, use_rolled_loops=False), multiple_results=True)(*args, _in_avals=_in_avals, _out_aval=_out_aval)
    return res
def _threefry2x32_jax_impl(*args: TfVal, _in_avals, _out_aval):
    res = _convert_jax_impl(functools.partial(jax._src.random._threefry2x32_lowering, use_rolled_loops=False), multiple_results=True)(*args, _in_avals=_in_avals, _out_aval=_out_aval)
    return res