pytorch_ext

Extension to Pytorch, including new modules, new functionals, bug fixes, etc.

Install

Install from source to get the up-to-date version:

pip install git+https://github.com/david-leon/Pytorch_Ext.git

Dependency