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

• Pytorch >=1.0

---

Documentation

https://david-leon.github.io/Pytorch_Ext/