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:
- Experimentation using the xla compiler from rust☆96Updated last year
- 8-bit floating point types for Rust☆59Updated 2 weeks ago
- RAI: Rust ML framework with composable transformations like JAX.☆92Updated last year
- A collection of optimisers for use with candle☆39Updated last week
- ☆24Updated 4 months ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆40Updated last year
- ☆90Updated 7 months ago
- ☆20Updated 10 months ago
- Low rank adaptation (LoRA) for Candle.☆154Updated 4 months ago
- A rust port of pytorch dataloader☆28Updated 8 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆47Updated 3 weeks ago
- Experimental compiler for deep learning models☆68Updated 3 months ago
- ☆27Updated 2 years ago
- Read and write tensorboard data using Rust☆21Updated last year
- ☆33Updated 9 months ago
- Locality Sensitive Hashing in Rust with Python bindings☆117Updated 2 years ago
- Rust client for the huggingface hub aiming for minimal subset of features over `huggingface-hub` python package☆219Updated 2 months ago
- A neural network inference library, written in Rust.☆64Updated last year
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆108Updated 2 years ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆75Updated this week
- GPU based FFT written in Rust and CubeCL☆23Updated 2 months ago
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle☆74Updated last year
- Example of tch-rs on M1