add GPU functionality (speedups)

flatironinstitute / Cryo-EM-Heterogeneity-Challenge-1

The Inaugural Flatiron Institute Cryo-EM Heterogeneity Community Challenge

https://osf.io/8h6fz/

MIT License

2 stars 0 forks source link

Open geoffwoollard opened 2 weeks ago

geoffwoollard commented 2 weeks ago

see map2map distance functions in src/cryo_challenge/_map_to_map/map_to_map_distance.py
add option to yaml for device CPU/GPU:0 etc.
dist2dist (probably will not work on GPU)
- unsure if can get GPU speedups for python optimal transport ot
- unsure if any GPU support for CVXPY. I don't think so

geoffwoollard commented 2 weeks ago

Not blocking public release.