6 Repos
Techniques for processing multiple text sequences in parallel to optimize hardware utilization during model training.
Distinct from Stateful Sequence Batching: Closest candidates focus on specific matrix utilities or stateful batching, not general parallel sequence training for LLMs.
Explore 6 awesome GitHub repositories matching artificial intelligence & ml · Batch Sequence Training. Refine with filters or upvote what's useful.
GPT-Neo is an open-source distributed training framework designed for scaling GPT-2 and GPT-3-style language models across multiple devices using mesh-tensorflow for model parallelism. It provides the infrastructure to train transformer-based language models with billions of parameters across distributed computing environments, making large-scale language model research accessible outside of proprietary systems. The framework supports training both autoregressive GPT-style models and masked language models like BERT or RoBERTa, with configurable masking strategies and token handling. It inclu
Groups variable-length sequences into padded batches to maximize hardware utilization during training.
GPT2-Chinese is a Chinese language model implementation based on the GPT-2 architecture. It provides a causal language model trainer and a natural language generation tool designed for training and generating human-like Chinese text sequences. The system integrates a BERT tokenizer to process Chinese corpora into manageable units for machine learning. It enables the development of predictive text models that can generate specific patterns, such as news or poetry, through prompt-based text completion. The project covers a full workflow including text tokenization, model training using a trans
Implements parallel processing of text segments to optimize GPU utilization during training.
Analyzes how batch sizes and sequence lengths affect hardware FLOPs utilization.
xtuner ist eine umfassende Trainings-Engine für Large Language Models und bietet ein Toolkit für Pre-Training, Supervised Fine-Tuning und die Optimierung von vision-sprachlichen multimodalen Modellen. Sie dient als verteilter Trainingsbeschleuniger und spezialisiertes Framework zur Skalierung von Mixture-of-Experts-Modellen sowie zur Ausrichtung von Modellverhalten durch Reinforcement Learning from Human Feedback. Das Projekt zeichnet sich durch fortgeschrittene Speicher- und Rechenoptimierungen aus, wie Sequence-Parallelism für ultra-lange Kontextfenster und Interleaved-Pipeline-Parallelism zur Reduzierung von GPU-Idle-Zeiten. Es bietet eine dedizierte Suite für Preference-Optimization und implementiert Techniken wie Group Relative Policy Optimization und Direct Preference Optimization, um Modell-Policies und Belohnungssysteme zu verfeinern. Breite Funktionsbereiche decken verteiltes Modelltraining über mehrere Knoten hinweg, multimodale Datensatzvorbereitung und die Verwaltung von Adapter-basiertem Fine-Tuning ab. Die Engine enthält zudem Tools für Modellevaluation, Weight-Merging und den Export trainierter Parameter in Inferenz-Engines. Das Training wird über standardisierte Konfigurationsdateien und verteilte Launcher verwaltet, um konsistente Ergebnisse über Rechencluster hinweg sicherzustellen.
Processes extended text datasets by distributing long sequences across hardware to overcome VRAM limits.
Dieses Projekt ist ein PyTorch-Tutorial für Sentiment-Analyse und eine Deep-Learning-Implementierung zur Textanalyse. Es bietet eine NLP-Pipeline (Natural Language Processing) zur Sequenzklassifizierung, die darauf ausgelegt ist, Textdaten zu bereinigen und neuronale Netze für die Kategorisierung von Wortfolgen zu trainieren. Der Fokus liegt auf der Anpassung vortrainierter Sprachmodelle für spezifische Textklassifizierungsaufgaben unter Verwendung benutzerdefinierter Datensätze. Es beinhaltet einen Prozess zum Fine-Tuning großer Sprachmodelle sowie die Implementierung rekurrenter Netze und Transformer zur Erkennung emotionaler Tonalität. Das Projekt deckt ein breites Spektrum der Textsequenzklassifizierung und PyTorch-Textverarbeitung ab. Dies beinhaltet Workflows zur Vorbereitung von Rohtext-Datensätzen mit der TorchText-Bibliothek sowie den Aufbau von Deep-Learning-Modellen zur Kategorisierung von Texten.
Groups multiple text sequences into tensors to optimize hardware utilization during model training.
FastVideo is a comprehensive system for accelerated video generation, serving as a video generation inference engine, a video diffusion training framework, and a modular pipeline orchestrator. It provides a distributed transformer optimizer and a distillation toolkit designed to reduce denoising steps and model complexity to increase frame rates. The project distinguishes itself through specialized acceleration techniques, including joint distillation and sparse attention training. It implements low-step video generation and weight quantization to FP8 or FP4 precision to increase throughput a
Distributes long video sequences across multiple GPUs using tensor-parallel layers to handle large workloads.