Adapt torch.jit.Traceable to RNN-cell like usage

[apaszke]

Right now it assumes that it encapsulates the whole model, or that it's a unique subgraph in the model. If a cell is used multiple times, we'll record the first forward run and compile a one-stage closure when it's reused. This means that the closure isn't differentiable even once.