tch-rs
tch-rs copied to clipboard
Rust bindings for the C++ api of PyTorch.
I'm on the fence about doing it this way. Partly what I'd like with this is easier m1 MacBook support and maybe iOS support. The issue is that the build...
Hi, I'm not super familiar with the design of the library, but I've been trying to use it in a project of mine. In doing so I found that `Tensor::f_copy`...
Add ```no_grad``` to the sampling function to improve inference efficiency.
This is similar to #330 but the idea is to enforce the proper link flags on the final binaries. This will only work for the binaries/examples/tests from the `tch` crate...
When `cargo build` my project with only two deps: ``` [dependencies] tch = "0.10.3" tokenizers = "0.13.2" ``` This error came out: `= note: libtorch_sys-6c1049f81ce711d6.rlib(torch_api.o) : error LNK2038: mismatch detected...
We tried to update tch to 0.16 which adds support for pytorch 2.3 in [burn](https://github.com/tracel-ai/burn) and we get a compilation error on Windows: ``` INTEL MKL ERROR: The specified module...
This is all I've done in terminal ``` wget https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.3.0%2Bcpu.zip zip libtorch-shared-with-deps-2.3.0+cpu.zip export LIBTORCH=/home/test/libtorch export LIBTORCH_INCLUDE=$LIBTORCH/include:$LIBTORCH/include/torch/csrc/api/include export LIBTORCH_LIB=$LIBTORCH/lib export LIBTORCH_STATIC=1 export LIBTORCH_BYPASS_VERSION_CHECK=1 ``` Then I made a rust-bert project that...
Is it possible to publish version 0.16 of pyo3-tch to crates.io? I see that 0.16 of tch-rs and torch-sys are available but the latest version of pyo3-tch is 0.15.
Hi, is it possible to get this in rust? in python: ``` import torch x = torch.tensor(2., requires_grad=True) a = x ** 2 a.retain_grad() b = x ** 3 b.retain_grad()...