Skip to content

Latest commit

 

History

2 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch implementation of the percent point function (PPF) for a Gaussian distribution, i.e., inverse CDF of a Gaussian distribution.

The goal is to enable using the PPF function in CUDA to go back and forth from a Gaussian to a Uniform distribution.

Notes

The function emulates the behavior of scipy.special.ndtri but on either GPU or CPU.Tested on Google Colab with a T100, process got a 6x improvement on GPU (performance is worse on CPU):

Screenshot

About

Port of scipy's inverse Standard Gaussian CDF (ndtri.c) in Pytorch (c. 6x performance with CUDA).

Resources

Stars

1 star

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages