k-diffusion
k-diffusion copied to clipboard
Use `torch.special.expm1`
This function provides greater precision than exp(x) - 1 for small values of x.
Found with TorchFix https://github.com/pytorch-labs/torchfix/