Improved TorchPearl