An "eXtension" to the PyTorch Deep Learning Framework. It contains "eXtra" features and functionality that I wish were in the official PyTorch framework.
TorchX is set up as a pip installable Python package:
pip install -U git+https://github.com/antoniojkim/TorchX.git
The primary documentation for this repository can be found at https://torch-ext.readthedocs.io/en/latest/