Sample Python extension using Rust/PyO3/tch to interact with PyTorch - View it on GitHub