LaurentMazare / tch-extLinks
Sample Python extension using Rust/PyO3/tch to interact with PyTorch
☆40Updated 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☆100Updated last year
- 8-bit floating point types for Rust☆61Updated this week
- Read and write tensorboard data using Rust☆24Updated last year
- A Rust Library for High-Performance Tensor Exchange with Python☆50Updated last month
- Experimental compiler for deep learning models☆74Updated 4 months ago
- A collection of optimisers for use with candle☆45Updated last month
- A rust port of pytorch dataloader☆30Updated last year
- RAI: Rust ML framework with composable transformations like JAX.☆95Updated last year
- ☆97Updated 2 months ago
- ☆19Updated 3 weeks ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆40Updated last year
- Low rank adaptation (LoRA) for Candle.☆169Updated 9 months ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆77Updated 5 months ago
- ☆28Updated 2 years ago
- ☆33Updated this week
- Locality Sensitive Hashing in Rust with Python bindings☆121Updated 2 years ago
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆111Updated 2 years ago
- ☆40Updated last year
- GPU based FFT written in Rust and CubeCL☆28Updated last month
- A neural network inference library, written in Rust.☆68Updated last year
- Port of Andrej Karpathy's minbpe to Rust☆30Updated last year
- 🧮 alphatensor matrix breakthrough algorithms + simd + rust.☆63Updated 3 years ago
- Automatic differentiation in Rust with WGPU support☆24Updated 4 years ago
- Rust client for the huggingface hub aiming for minimal subset of features over `huggingface-hub` python package☆254Updated last week
- Tiny Autograd engine written in Rust☆60Updated last year
- Dataflow is a data processing library, primarily for machine learning.☆24Updated 2 years ago
- ☆135Updated last year
- Rustic bindings for IREE☆18Updated 2 years ago
- Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle☆76Updated last year
- A Deep Learning and preprocessing framework in Rust with support for CPU and GPU.☆133Updated 2 years ago