herrmann / rustorch
"PyTorch in Rust"
☆16Updated last year
Alternatives and similar repositories for rustorch:
Users that are interested in rustorch are comparing it to the libraries listed below
- Read and write tensorboard data using Rust☆20Updated last year
- Tensor library for Zig☆11Updated 5 months ago
- A small python library to run iterators in a separate process☆10Updated last year
- A collection of optimisers for use with candle☆34Updated 5 months ago
- ☆21Updated 4 months ago
- Make triton easier☆47Updated 10 months ago
- Because it's there.☆16Updated 7 months ago
- ☆18Updated last year
- Inference Llama 2 in one file of zero-dependency, zero-unsafe Rust☆38Updated last year
- Training hybrid models for dummies.☆20Updated 3 months ago
- JAX bindings for the flash-attention3 kernels☆11Updated 8 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆34Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 11 months ago
- ☆27Updated 9 months ago
- Personal solutions to the Triton Puzzles☆18Updated 9 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆49Updated last week
- Rust crate for some audio utilities☆22Updated last month
- My explorations into editing the knowledge and memories of an attention network☆34Updated 2 years ago
- Learn CUDA with PyTorch☆20Updated 2 months ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11Updated 11 months ago
- 👷 Build compute kernels☆35Updated this week
- Implementation of a holodeck, written in Pytorch☆17Updated last year
- ML/DL Math and Method notes☆60Updated last year
- Minimum Description Length probing for neural network representations☆19Updated 2 months ago
- ☆19Updated 2 years ago
- Repository containing the SPIN experiments on the DIBT 10k ranked prompts☆24Updated last year
- A small rust-based data loader☆24Updated 4 months ago
- Port of Andrej Karpathy's minbpe to Rust☆22Updated 11 months ago
- ☆11Updated 2 months ago
- HyPe: Better Pre-trained Language Model Fine-tuning with Hidden Representation Perturbation [ACL 2023]☆14Updated last year