Tweeted By @PyTorch
Fast custom-RNNs using TorchScript and torch.jit
— PyTorch (@PyTorch) May 1, 2019
- describes how to write custom RNNs in PyTorch that run close to CuDNN speeds, powered by our torch.jit compiler
- details of how we do the optimization in the compilerhttps://t.co/Jlwn2Nel6X pic.twitter.com/kzCmQZSCj5