PyTorch

"When you're training a SOTA generative model with two people the giant wealth of existing GitHub Issues threads/reference codebases/etc is essential."

"Empowers project with efficient model development and deployment, including ONNX integration."

"ML hardware acceleration using CUDA for Nvidia GPUs on Windows, and via MPS and MLX frameworks for Apple silicon on macOS!"