LaurentMazare / tch-extLinks
Sample Python extension using Rust/PyO3/tch to interact with PyTorch
☆36Updated 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:
- Read and write tensorboard data using Rust☆21Updated last year
- Experimentation using the xla compiler from rust☆93Updated 9 months ago
- ☆19Updated 7 months ago
- A collection of optimisers for use with candle☆36Updated 2 weeks ago
- ☆26Updated last year
- Experimental compiler for deep learning models☆67Updated last week
- 8-bit floating point types for Rust☆45Updated 2 months ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆39Updated last year
- ☆23Updated last month
- Dataflow is a data processing library, primarily for machine learning.☆21Updated last year
- ☆12Updated last year
- ☆86Updated 4 months ago
- RAI: Rust ML framework with composable transformations like JAX.☆89Updated 10 months ago
- A small python library to run iterators in a separate process☆10Updated last year
- A Rust Library for High-Performance Tensor Exchange with Python☆45Updated this week
- ☆30Updated 6 months ago
- Low rank adaptation (LoRA) for Candle.☆147Updated last month
- Simplistic API for deep learning tensor operations☆30Updated last month
- implement llava using candle☆15Updated 11 months ago
- Einstein summation for Rust☆38Updated 4 years ago
- A neural network inference library, written in Rust.☆62Updated 10 months ago
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆104Updated last year
- Example of tch-rs on M1☆53Updated last year
- Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle☆71Updated last year
- A rust port of pytorch dataloader☆26Updated 5 months ago
- Python tqdm in Rust.☆45Updated last year
- GPU based FFT written in Rust and CubeCL☆22Updated 2 months ago
- ☆129Updated last year
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆73Updated last week
- GGML bindings that aim to be idiomatic Rust rather than directly corresponding to the C/C++ interface☆19Updated last year