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:
- Read and write tensorboard data using Rust☆21Updated last year
- A collection of optimisers for use with candle☆36Updated last month
- Experimentation using the xla compiler from rust☆93Updated 10 months ago
- ☆20Updated 8 months ago
- ☆26Updated last year
- 8-bit floating point types for Rust☆46Updated 3 months ago
- A small python library to run iterators in a separate process☆10Updated last year
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆39Updated last year
- ☆23Updated 2 months ago
- Experimental compiler for deep learning models☆67Updated last month
- RAI: Rust ML framework with composable transformations like JAX.☆90Updated 11 months ago
- ☆30Updated 7 months ago
- ☆12Updated last year
- ☆87Updated 5 months ago
- Example of tch-rs on M1☆53Updated last year
- A Rust Library for High-Performance Tensor Exchange with Python☆47Updated 2 weeks ago
- Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle☆72Updated last year
- A neural network inference library, written in Rust.☆63Updated 11 months ago
- A rust port of pytorch dataloader☆27Updated 6 months ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆74Updated 2 weeks ago
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- Low rank adaptation (LoRA) for Candle.☆150Updated 2 months ago
- GPU based FFT written in Rust and CubeCL☆23Updated 2 weeks ago
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆106Updated last year
- Port of Andrej Karpathy's minbpe to Rust☆23Updated last year
- Rustic bindings for IREE☆18Updated 2 years ago
- Python tqdm in Rust.☆45Updated last year
- A curated collection of Rust projects related to neural networks, designed to complement "Are We Learning Yet."☆54Updated 2 weeks ago
- 🧮 alphatensor matrix breakthrough algorithms + simd + rust.☆61Updated 2 years ago
- Graph model execution API for Candle☆13Updated 7 months ago