search

bunnech / cellot

Learning Single-Cell Perturbation Responses using Neural Optimal Transport
BSD 3-Clause "New" or "Revised" License
109 stars 9 forks source link

GPU acceleration #2

Closed yanwu2014 closed 6 months ago

yanwu2014 commented 1 year ago

Hi,

Thanks so much for the great tool! I was wondering if there's a way to use GPUs using the current training framework or if you were planning on adding GPU support in the near future?

Thanks! Yan

mksingh2104 commented 7 months ago

GPU acceleration would be greatly appreciated.

Sincerely,

bunnech commented 6 months ago

Yes, it was originally planned but did not happen yet. In case you would like to add a PR, let us know!