Wrapper for C++ torch::jit::Module with methods, attributes, and parameters.| pytorch.org
A wrapper for sharding module parameters across data parallel workers.| pytorch.org
Learn important machine learning concepts hands-on by writing PyTorch code.| www.learnpytorch.io
Base class for all neural network modules.| pytorch.org