LaurentMazare / tch-extLinks
Sample Python extension using Rust/PyO3/tch to interact with PyTorch
☆37Updated last year
Alternatives and similar repositories for tch-ext
Users that are interested in tch-ext are comparing it to the libraries listed below
Sorting:
- ☆23Updated 3 months ago
- 8-bit floating point types for Rust☆48Updated 2 weeks ago
- Experimentation using the xla compiler from rust☆95Updated 11 months ago
- Read and write tensorboard data using Rust☆21Updated last year
- ☆20Updated 9 months ago
- A collection of optimisers for use with candle☆37Updated last week
- ☆90Updated 6 months ago
- Experimental compiler for deep learning models☆68Updated 2 months ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆40Updated last year
- A rust port of pytorch dataloader☆27Updated 7 months ago
- RAI: Rust ML framework with composable transformations like JAX.☆92Updated last year
- Low rank adaptation (LoRA) for Candle.☆152Updated 3 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆47Updated 3 weeks ago
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆108Updated 2 years ago
- Locality Sensitive Hashing in Rust with Python bindings☆116Updated 2 years ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆74Updated 3 weeks ago
- Rust client for the huggingface hub aiming for minimal subset of features over `huggingface-hub` python package☆219Updated last month
- A neural network inference library, written in Rust.☆63Updated last year
- Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle☆73Updated last year
- ☆31Updated 8 months ago
- GPU based FFT written in Rust and CubeCL☆23Updated last month
- ☆27Updated last year
- ☆130Updated last year
- Inference Llama 2 in one file of pure Rust 🦀☆233Updated last year
- Dataflow is a data processing library, primarily for machine learning.☆24Updated 2 years ago
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- Tiny Autograd engine written in Rust☆57Updated 11 months ago
- Rustic bindings for IREE☆18Updated 2 years ago
- Port of Andrej Karpathy's minbpe to Rust☆25Updated last year
- Example of tch-rs on M1☆54Updated last year