Add MPS as a possible computing backend

jeshraghian / snntorch

Deep and online learning with spiking neural networks in Python

https://snntorch.readthedocs.io/en/latest/

MIT License

1.28k stars 217 forks source link

Closed laurentperrinet closed 11 months ago

laurentperrinet commented 11 months ago

These changes just add a quick test to know if the MPS device from the Apple Silicon chips is available. Was tested on tutorials with no problem.

jeshraghian commented 11 months ago

Love your work, this is perfect.

laurentperrinet commented 11 months ago

You are too kind!

My first PR to snntorch and I ❤️ it!