xxcheng0708 / pytorch-model-train-template
pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用
☆92Updated last year
Alternatives and similar repositories for pytorch-model-train-template:
Users that are interested in pytorch-model-train-template are comparing it to the libraries listed below
- DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练)☆151Updated last year
- DeepSpeed Tutorial☆95Updated 7 months ago
- 主要记录大语言大模型(LLMs) 算法(应用)工程师多模态相关知识☆161Updated 10 months ago
- 多模态 MM +Chat 合集☆247Updated last month
- Reading notes about Multimodal Large Language Models, Large Language Models, and Diffusion Models☆317Updated this week
- ☆292Updated last month
- ☆36Updated last month
- 和李沐一起读论文☆176Updated 2 months ago
- ☆182Updated 5 months ago
- Efficient Multimodal Large Language Models: A Survey☆327Updated 3 weeks ago
- Yet another PyTorch Trainer and some core components for deep learning.☆214Updated 10 months ago
- 包含程序员面试大厂面试题和面试经验☆123Updated 2 months ago
- The pure and clear PyTorch Distributed Training Framework.☆276Updated last year
- My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"☆224Updated last month
- ☆100Updated 8 months ago
- Keras implement of Finite Scalar Quantization☆71Updated last year
- ☆189Updated last year
- SimVQ: Addressing Representation Collapse in Vector Quantized Models with One Linear Layer☆227Updated 2 months ago
- Personal Project: MPP-Qwen14B & MPP-Qwen-Next(Multimodal Pipeline Parallel based on Qwen-LM). Support [video/image/multi-image] {sft/conv…☆425Updated last week
- [Survey] Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Survey☆393Updated 2 months ago
- [NeurIPS 2024] Classification Done Right for Vision-Language Pre-Training☆202Updated this week
- WWW2025 Multimodal Intent Recognition for Dialogue Systems Challenge☆118Updated 4 months ago
- Lion and Adam optimization comparison☆60Updated 2 years ago
- Explore the Limits of Omni-modal Pretraining at Scale☆96Updated 6 months ago
- an implementation of transformer, bert, gpt, and diffusion models for learning purposes☆151Updated 5 months ago
- pytorch distribute tutorials☆117Updated last month
- ☆64Updated 2 years ago
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆158Updated 10 months ago
- PyTorch Dataset Rank Dataset☆42Updated 4 years ago
- mllm-npu: training multimodal large language models on Ascend NPUs☆91Updated 6 months ago