GPU inference in Torchscript model; post layer norm

• Support GPU/CUDA inference in Torchscript model
• Support post layer norm
• Support custom init value for weight_c
• Add unit tests for GPU inference
• Add unit tests for backward()
• Add more unit tests for Torchscript