graphcore-research / unit-scaling-demo
Unit Scaling demo and experimentation code
☆16Updated last year
Alternatives and similar repositories for unit-scaling-demo:
Users that are interested in unit-scaling-demo are comparing it to the libraries listed below
- Repository for CPU Kernel Generation for LLM Inference☆26Updated last year
- Flexible simulator for mixed precision and format simulation of LLMs and vision transformers.☆49Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- QuIP quantization☆52Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆73Updated 8 months ago
- ☆68Updated last month
- Implementation of Hyena Hierarchy in JAX☆10Updated 2 years ago
- ☆104Updated 8 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆69Updated 10 months ago
- Make triton easier☆47Updated 10 months ago
- ☆68Updated 3 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 9 months ago
- ☆11Updated last month
- Vocabulary Parallelism☆19Updated 2 months ago
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆40Updated last week
- Using FlexAttention to compute attention with different masking patterns☆43Updated 7 months ago
- Memory Optimizations for Deep Learning (ICML 2023)☆64Updated last year
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆36Updated 3 weeks ago
- Intel Gaudi's Megatron DeepSpeed Large Language Models for training☆13Updated 4 months ago
- extensible collectives library in triton☆86Updated last month
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆48Updated last year
- FlexAttention w/ FlashAttention3 Support☆26Updated 7 months ago
- Best practices for testing advanced Mixtral, DeepSeek, and Qwen series MoE models using Megatron Core MoE.☆10Updated 2 weeks ago
- ☆25Updated 5 months ago
- ☆158Updated last year
- Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆81Updated 5 months ago
- ☆20Updated 3 weeks ago
- Quantized Attention on GPU☆45Updated 5 months ago
- ☆13Updated last month
- FBI-LLM: Scaling Up Fully Binarized LLMs from Scratch via Autoregressive Distillation☆48Updated 9 months ago