Add torch-ffi import