pytorch_ext An extension library for PyTorch that fills the gaps we encountered. Under development, not stable.