TODO: PyTorch interoperability
==============================