def _nest_transpose_rule(*args, **kwargs): return ad.call_transpose(nest_p, *args, **kwargs)
def rule(*args, **kwargs): return ad.call_transpose(prim.subcall('transpose'), *args, **kwargs)
def _transpose_rule(*args, **kwargs): return ad.call_transpose(self.subcall('transpose'), *args, **kwargs)