PyTorch

See also JAX (PyTorch is dead. Long live JAX).

For multiple GPUs, use distributed data parallel

Related libraries

Tutorials

Tips