huggingface / accelerate
π A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
β8,645Updated this week
Alternatives and similar repositories for accelerate:
Users that are interested in accelerate are comparing it to the libraries listed below
- Fast and memory-efficient exact attentionβ17,032Updated this week
- Accessible large language models via k-bit quantization for PyTorch.β6,949Updated this week
- Train transformer language models with reinforcement learning.β13,373Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.β9,372Updated last week
- Ongoing research training transformer models at scaleβ12,118Updated this week
- π€ PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.β18,174Updated last week
- PyTorch extensions for high performance and large scale training.β3,306Updated 2 weeks ago
- Transformer related optimization, including BERT, GPTβ6,137Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)β8,861Updated 2 months ago
- An open source implementation of CLIP.β11,602Updated this week
- Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"β11,798Updated 4 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)β4,627Updated last year
- π Accelerate inference and training of π€ Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimizationβ¦β2,856Updated this week
- A concise but complete full-attention transformer with a set of promising experimental features from various papersβ5,234Updated last week
- LAVIS - A One-stop Library for Language-Vision Intelligenceβ10,480Updated 5 months ago
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalitiesβ21,097Updated last month
- QLoRA: Efficient Finetuning of Quantized LLMsβ10,394Updated 10 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorchβ8,625Updated 2 weeks ago
- Flax is a neural network library for JAX that is designed for flexibility.β6,493Updated this week
- π€ Evaluate: A library for easily evaluating machine learning models and datasets.β2,187Updated 3 months ago
- π€ Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.β28,700Updated this week
- β2,790Updated this week
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.β29,327Updated this week
- Unsupervised text tokenizer for Neural Network-based text generation.β10,816Updated 3 weeks ago
- [NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.β22,283Updated 8 months ago
- CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an imageβ28,622Updated 9 months ago
- Foundation Architecture for (M)LLMsβ3,070Updated last year
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.β31,359Updated 3 months ago
- BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)β7,344Updated last year
- An annotated implementation of the Transformer paper.β6,181Updated last year