apd10 / RzLinear
A compressed alternative to matrix multiplication using state-of-the art compression ROBE-Z
☆9Updated last year
Alternatives and similar repositories for RzLinear:
Users that are interested in RzLinear are comparing it to the libraries listed below
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 11 months ago
- Unit Scaling demo and experimentation code☆16Updated 11 months ago
- Personal solutions to the Triton Puzzles☆17Updated 7 months ago
- ☆20Updated last year
- ☆21Updated 3 months ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆43Updated last week
- JAX implementations of RWKV☆19Updated last year
- extensible collectives library in triton☆83Updated 5 months ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆104Updated 2 months ago
- sigma-MoE layer☆18Updated last year
- PyTorch centric eager mode debugger☆46Updated 2 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 5 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 7 months ago
- Benchmarking PyTorch 2.0 different models☆21Updated last year
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 3 years ago
- ☆59Updated 2 weeks ago
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆45Updated last year
- RWKV model implementation☆37Updated last year
- Benchmarks to capture important workloads.☆29Updated 3 weeks ago
- Utilities for Training Very Large Models☆57Updated 4 months ago
- ☆11Updated last year
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated this week
- ☆100Updated 5 months ago
- ☆157Updated last year
- FlexAttention w/ FlashAttention3 Support☆26Updated 4 months ago
- A safetensors extension to efficiently store sparse quantized tensors on disk☆77Updated this week
- Implementation of Hyena Hierarchy in JAX☆10Updated last year
- ☆22Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆88Updated last year