The hasktorch FFI bindings to the ATen library - View it on GitHub