Facebook's extensions to torch/nn