To facilitate this, PyTorch provides data sets and data loader classes to feed data for training and testing, significantly speeding up the process.

Researchers have discovered a critical flaw in PyTorch’s distributed RPC system, allowing attackers to execute arbitrary commands on the OS and steal AI training data.