Add support for PyTorch Lightning trainers hooked up to MPS on M1/M2 machines - Githubissues

ML4GLand / EUGENe

Elucidating the Utility of Genomic Elements with Neural Nets

MIT License

61 stars 4 forks source link

Add support for PyTorch Lightning trainers hooked up to MPS on M1/M2 machines #40

Open adamklie opened 9 months ago

adamklie commented 9 months ago

See #39

Also add to docs that this is currently not supported