herrmann / rustorchLinks
"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
Sorting:
- 👷 Build compute kernels☆74Updated last week
- Read and write tensorboard data using Rust☆21Updated last year
- ☆12Updated last year
- ☆23Updated 7 months ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11Updated last year
- A small python library to run iterators in a separate process☆10Updated last year
- Implementation of Hyena Hierarchy in JAX☆10Updated 2 years ago
- Make triton easier☆47Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- ☆39Updated 2 years ago
- Rust Implementation of micrograd☆52Updated last year
- ML/DL Math and Method notes☆61Updated last year
- Efficiently computing & storing token n-grams from large corpora☆24Updated 9 months ago
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆59Updated 3 years ago
- ☆130Updated last year
- Inference Llama 2 in one file of zero-dependency, zero-unsafe Rust☆38Updated last year
- Visualising Losses in Deep Neural Networks☆16Updated 11 months ago
- implement llava using candle☆15Updated last year
- Because it's there.☆16Updated 9 months ago
- Utilities for Training Very Large Models☆58Updated 9 months ago
- JAX bindings for the flash-attention3 kernels☆11Updated 11 months ago
- A dashboard for exploring timm learning rate schedulers☆19Updated 7 months ago
- 🤝 Trade any tensors over the network☆30Updated last year
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆14Updated last year
- Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code☆11Updated last year
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- ☆18Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- ☆11Updated 5 months ago
- This is a new metric that can be used to evaluate faithfulness of text generated by LLMs. The work behind this repository can be found he…☆31Updated last year