search

google-deepmind / torch-distributions

Probability distributions, wrapped for Torch.
BSD 3-Clause "New" or "Revised" License
61 stars 25 forks source link

GPU Support #44

Closed chenyuntc closed 7 years ago

chenyuntc commented 8 years ago

Hi,it's a wonderful module, but it seems doesn't support GPU. Do you plan to add GPU support ? I come from scipy.stats, and looking for GPU accelerate on certain math function.