While this library is for now standalone, the goal is to get both the mathematical function as well as the distributions into torch core package. See also pytorch/pytorch#108948. The torchlambertw ...