search

karpathy / pytorch-normalizing-flows

Normalizing flows in PyTorch. Current intended use is education not production.
845 stars 98 forks source link

CUDA Version #2

Closed sadegh-aa closed 4 years ago

sadegh-aa commented 4 years ago

Thanks for sharing a fairly complete repo of NFs. Do you have plans to release the cuda version of the models (soon)?