(*args, **kwargs)
return backend()['remat'](*args, **kwargs)
Recompute everything in the backward pass to same memory.
Recompute everything in the backward pass to same memory.
Recompute everything in the backward pass to same memory .
def remat(*args, **kwargs):
  """Recompute everything in the backward pass to same memory."""
  return backend()['remat'](*args, **kwargs)
def remat ( * args , * * kwargs ) : return backend ( ) [ 'remat' ] ( * args , * * kwargs )