4 个仓库
Techniques that increase training throughput through packing, compilation, and memory reduction strategies.
Distinct from Training Memory Optimizers: Distinct from Training Memory Optimizers: focuses on throughput improvements (packing, compilation) in addition to memory reduction, not just memory alone.
Explore 4 awesome GitHub repositories matching data & databases · Training Speed Optimizations. Refine with filters or upvote what's useful.
该项目是一个实现 Faster R-CNN 架构的 PyTorch 目标检测框架。它作为一种视觉模型,用于预测图像和实时视频流中多个目标周围的精确边界框。 该系统针对多 GPU 训练进行了优化,以减少模型收敛所需的时间。它利用 GPU 加速设计来处理复杂检测网络的训练和推理。 该框架涵盖了完整的目标检测生命周期,包括针对静态图像和实时视频流的自定义网络训练和推理。它包括使用标准化数据集进行模型性能验证的功能,以及诸如基于长宽比的分组和分布式工作负载等训练优化。
Implements throughput improvements through aspect-ratio-based grouping and multi-image batching to optimize training speed.
Tensorpack 是一个高性能 TensorFlow 训练框架和分布式深度学习工具包。它提供了一套用于构建和训练神经网络的工具,重点在于执行速度和架构灵活性。 该项目作为一个神经网络优化套件,实现了高效的执行模式以减少训练开销。它作为一个并行数据加载流水线,利用自动化并行化在处理大数据集时最大化吞吐量。 该工具包涵盖了使用数据并行策略在多个 GPU 和计算集群上进行的分布式训练。其功能包括大规模数据集处理和性能优化,以提高训练吞吐量。
Reduces training overhead and increases throughput using high-efficiency execution patterns.
Torchtune is a PyTorch-native library for fine-tuning, aligning, and quantizing large language models. It provides a configurable training pipeline orchestrated through YAML recipes, with CLI overrides and component swapping, distributed training via FSDP2, memory optimizations, and parameter-efficient fine-tuning methods like LoRA, DoRA, and QLoRA. The library distinguishes itself through its YAML-driven configuration system that defines all training parameters and instantiates components from config files, with full CLI override capability for any field or component at launch time. It suppo
Combines packing, compilation, and activation checkpointing to reduce memory and increase training throughput.
FlagAI is a distributed deep learning framework and platform designed for the end-to-end lifecycle of large-scale foundation models. It provides a toolkit for training, fine-tuning, and deploying large language models and multi-modal systems across multi-node computing clusters. The project features hardware-agnostic compute abstractions to ensure consistent execution across different accelerators. It includes a dedicated library for parameter-efficient fine-tuning, allowing large neural networks to be adapted to specific tasks with minimal parameter updates and reduced computational overhead
Decreases total training time for large-scale architectures through specialized attention and parallel processing.