LaurentMazare / tch-ext
Sample Python extension using Rust/PyO3/tch to interact with PyTorch
☆34Updated last year
Alternatives and similar repositories for tch-ext:
Users that are interested in tch-ext are comparing it to the libraries listed below
- Read and write tensorboard data using Rust☆19Updated last year
- ☆19Updated 5 months ago
- Experimentation using the xla compiler from rust☆90Updated 6 months ago
- A small python library to run iterators in a separate process☆10Updated 11 months ago
- A collection of optimisers for use with candle☆34Updated 3 months ago
- RAI: Rust ML framework with composable transformations like JAX.☆85Updated 7 months ago
- 8-bit floating point types for Rust☆46Updated last month
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆38Updated 11 months ago
- Experimental compiler for deep learning models☆28Updated 3 weeks ago
- ☆25Updated last year
- ☆12Updated last year
- ☆28Updated 3 months ago
- Einstein summation for Rust☆35Updated 3 years ago
- Rustic bindings for IREE☆18Updated 2 years ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆74Updated 2 months ago
- ☆21Updated 8 months ago
- A neural network inference library, written in Rust.☆60Updated 8 months ago
- A diffusers API in Burn (Rust)☆19Updated 8 months ago
- Graph model execution API for Candle☆13Updated 3 months ago
- Example of tch-rs on M1☆53Updated 11 months ago
- A rust port of pytorch dataloader☆26Updated 2 months ago
- ☆83Updated 2 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆40Updated 3 months ago
- implement llava using candle☆14Updated 9 months ago
- A Keras like abstraction layer on top of the Rust ML framework candle☆23Updated 8 months ago
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- Inference Llama 2 in one file of zero-dependency, zero-unsafe Rust☆37Updated last year
- A high-performance constrained decoding engine based on context free grammar in Rust☆47Updated 2 months ago
- Fast convolutions library implemented completely in Rust. Minimal depedencies required, and especially no external C libraries.☆26Updated 2 years ago