Support any pytorch-known devices

LSSTDESC / parsnip

Deep generative modeling of astronomical transient light curves

MIT License

7 stars 10 forks source link

Support any pytorch-known devices #7

Closed hombit closed 2 years ago

hombit commented 2 years ago

Sorry, posted it by a mistake. BTW, I have some small things to improve (you practically could see them above), please let me know if you are interested in these contributions

kboone commented 2 years ago

Sure, these look like some good changes. I'd be happy to merge them in if you want to send them as individual PRs.