hkproj / pytorch-transformer-distributedLinks
Distributed training (multi-node) of a Transformer model
☆72Updated last year
Alternatives and similar repositories for pytorch-transformer-distributed
Users that are interested in pytorch-transformer-distributed are comparing it to the libraries listed below
Sorting:
- minimal GRPO implementation from scratch☆92Updated 4 months ago
- Notes on Direct Preference Optimization☆19Updated last year
- LLaMA 2 implemented from scratch in PyTorch☆337Updated last year
- An extension of the nanoGPT repository for training small MOE models.☆160Updated 4 months ago
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆110Updated last year
- LoRA and DoRA from Scratch Implementations☆206Updated last year
- ☆179Updated 6 months ago
- ☆88Updated 9 months ago
- Prune transformer layers☆69Updated last year
- ☆198Updated 5 months ago
- ☆40Updated last month
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆108Updated 2 months ago
- The official implementation of the paper "What Matters in Transformers? Not All Attention is Needed".☆174Updated 3 months ago
- RL significantly the reasoning capability of Qwen2.5-1.5B-Instruct☆29Updated 4 months ago
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆47Updated last year
- Notes and commented code for RLHF (PPO)☆97Updated last year
- Unofficial implementation of https://arxiv.org/pdf/2407.14679☆45Updated 10 months ago
- Notes on quantization in neural networks☆89Updated last year
- A set of scripts and notebooks on LLM finetunning and dataset creation☆110Updated 9 months ago
- From scratch implementation of a vision language model in pure PyTorch☆227Updated last year
- A simplified implementation for experimenting with RLVR on GSM8K, This repository provides a starting point for exploring reasoning.☆112Updated 5 months ago
- Notes about LLaMA 2 model☆63Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆188Updated last month
- GPU Kernels☆188Updated 2 months ago
- This repository contains an implementation of the LLaMA 2 (Large Language Model Meta AI) model, a Generative Pretrained Transformer (GPT)…☆68Updated last year
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆318Updated 2 months ago
- First-principle implementations of groundbreaking AI algorithms using a wide range of deep learning frameworks, accompanied by supporting…☆174Updated last week
- Welcome to the LLMs Interview Prep Guide! This GitHub repository offers a curated set of interview questions and answers tailored for Dat…☆143Updated last year
- ☆161Updated last year
- Survey: A collection of AWESOME papers and resources on the latest research in Mixture of Experts.☆122Updated 10 months ago