PyTorch
To speed up your pytorch model use CUDA :
https://youtu.be/QxlvD_yEgz8?si=Vdf0fi4Vt8qp7VL5&t=318