hkproj / pytorch-transformer-distributed
Distributed training (multi-node) of a Transformer model
☆43Updated 7 months ago
Related projects ⓘ
Alternatives and complementary repositories for pytorch-transformer-distributed
- End-to-End LLM Guide☆97Updated 4 months ago
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆47Updated 5 months ago
- A minimal implementation of LLaVA-style VLM with interleaved image & text & video processing ability.☆84Updated 2 months ago
- LLaMA 2 implemented from scratch in PyTorch☆254Updated last year
- From scratch implementation of a vision language model in pure PyTorch☆162Updated 6 months ago
- Prune transformer layers☆64Updated 5 months ago
- A set of scripts and notebooks on LLM finetunning and dataset creation☆93Updated last month
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆82Updated last year
- LoRA and DoRA from Scratch Implementations☆188Updated 8 months ago
- Reference implementation of Mistral AI 7B v0.1 model.☆27Updated 10 months ago
- ☆133Updated 9 months ago
- Starter pack for NeurIPS LLM Efficiency Challenge 2023.☆118Updated last year
- RAGs: Simple implementations of Retrieval Augmented Generation (RAG) Systems☆83Updated 7 months ago
- ring-attention experiments☆97Updated last month
- ☆44Updated last week
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆86Updated last year
- ☆40Updated 6 months ago
- Recreating PyTorch from scratch (C/C++, CUDA, NCCL and Python, with multi-GPU support and automatic differentiation!)☆114Updated 5 months ago
- Set of scripts to finetune LLMs☆36Updated 7 months ago
- Notes on quantization in neural networks☆58Updated 11 months ago
- Notes and commented code for RLHF (PPO)☆38Updated 8 months ago
- Notes on the Mistral AI model☆18Updated 10 months ago
- Collection of autoregressive model implementation☆67Updated this week
- Accelerate Model Training with PyTorch 2.X, published by Packt☆29Updated 5 months ago
- Pretrain Vision and Large Language Models in Python, Published by Packt☆84Updated 10 months ago
- My fork os allen AI's OLMo for educational purposes.☆28Updated this week
- Unofficial implementation of https://arxiv.org/pdf/2407.14679☆36Updated 2 months ago
- Coding a Multimodal (Vision) Language Model from scratch in PyTorch with full explanation: https://www.youtube.com/watch?v=vAmKB7iPkWw☆308Updated 3 months ago
- ☆15Updated last year
- Parameter-efficient finetuning script for Phi-3-vision, the strong multimodal language model by Microsoft.☆54Updated 5 months ago