xxcheng0708 / pytorch-model-train-templateLinks
pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用
☆127Updated 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
Sorting:
- DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练)☆183Updated 2 years ago
- ☆42Updated 10 months ago
- 一款便捷的抢占显卡脚本☆384Updated last week
- DeepSpeed Tutorial☆104Updated last year
- 青稞Talk☆179Updated last week
- 主要记录大语言大模型(LLMs) 算法(应用)工程师多模态相关知识☆253Updated last year
- 多模态 MM +Chat 合集☆279Updated 4 months ago
- Yet another PyTorch Trainer and some core components for deep learning.☆222Updated last year
- [COLM 2025] Open-Qwen2VL: Compute-Efficient Pre-Training of Fully-Open Multimodal LLMs on Academic Resources☆294Updated 3 months ago
- Lion and Adam optimization comparison☆64Updated 2 years ago
- ☆216Updated 3 weeks ago
- ☆200Updated 2 years ago
- ☆73Updated 3 years ago
- The pure and clear PyTorch Distributed Training Framework.☆274Updated last year
- pytorch distribute tutorials☆161Updated 6 months ago
- The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink…☆673Updated this week
- 这是一个DiT-pytorch的代码,主要用于学习DiT结构。☆84Updated last year
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆338Updated 10 months ago
- An awesome gpu tasks scheduler. 轻量好用的GPU机群任务调度工具。觉得有用可以点个star☆194Updated 3 years ago
- My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"☆270Updated last month
- Efficient Multimodal Large Language Models: A Survey☆377Updated 7 months ago
- Simple tutorials on Pytorch DDP training☆285Updated 3 years ago
- Implementation of "Attention Is Off By One" by Evan Miller☆197Updated 2 years ago
- 和李沐一起读论文☆213Updated 6 months ago
- [Survey] Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Survey☆465Updated 11 months ago
- Pytorch Lightning入门中文教程,转载请注明来源。(当初是写着玩的,建议看完MNIST这个例子再上手)☆229Updated 5 years ago
- Reading notes about Multimodal Large Language Models, Large Language Models, and Diffusion Models☆818Updated 2 weeks ago
- Collect the awesome works evolved around reasoning models like O1/R1 in visual domain☆49Updated 5 months ago
- Implementation of FlashAttention in PyTorch☆176Updated 11 months ago
- Implementation of Denoising Diffusion Probabilistic Model in MindSpore☆44Updated 3 years ago