huggingface / optimum-tpu
Google TPU optimizations for transformers models
☆74Updated this week
Related projects ⓘ
Alternatives and complementary repositories for optimum-tpu
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆40Updated this week
- ☆106Updated this week
- ☆155Updated this week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆228Updated this week
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆171Updated 3 months ago
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆152Updated this week
- ☆64Updated 2 years ago
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆58Updated 3 months ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆250Updated 3 weeks ago
- Manage scalable open LLM inference endpoints in Slurm clusters☆237Updated 3 months ago
- experiments with inference on llama☆105Updated 5 months ago
- ☆116Updated 2 months ago
- ☆91Updated last month
- ☆49Updated 7 months ago
- ☆200Updated 9 months ago
- ☆96Updated last month
- Comprehensive analysis of difference in performance of QLora, Lora, and Full Finetunes.☆81Updated last year
- QLoRA with Enhanced Multi GPU Support☆36Updated last year
- Automated Identification of Redundant Layer Blocks for Pruning in Large Language Models☆194Updated 6 months ago
- Set of scripts to finetune LLMs☆36Updated 7 months ago
- ☆115Updated last week
- ☆20Updated last year
- A safetensors extension to efficiently store sparse quantized tensors on disk☆46Updated this week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆83Updated last week
- 🕹️ Performance Comparison of MLOps Engines, Frameworks, and Languages on Mainstream AI Models.☆132Updated 3 months ago
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆154Updated 3 weeks ago
- code for training & evaluating Contextual Document Embedding models☆92Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆190Updated 2 weeks ago
- A toolkit for fine-tuning, inferencing, and evaluating GreenBitAI's LLMs.☆72Updated 3 weeks ago
- ☆106Updated 2 weeks ago