Will there be a PyTorch version of the implementation?

yang-song / score_inverse_problems

Official repo for "Solving Inverse Problems in Medical Imaging with Score-Based Generative Models"

214 stars 26 forks source link

Will there be a PyTorch version of the implementation? #8

Open obiyoag opened 1 year ago

obiyoag commented 1 year ago

Hi,

I am very interested in your work and want to reproduce your results. But I am not very familiar with JAX. Wonder will there be a PyTorch version of the implementation.

Best regards