hkproj / pytorch-transformer-distributedLinks
Distributed training (multi-node) of a Transformer model
☆85Updated 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☆98Updated 7 months ago
- Notes on Direct Preference Optimization☆23Updated last year
- An extension of the nanoGPT repository for training small MOE models.☆202Updated 7 months ago
- RL significantly the reasoning capability of Qwen2.5-1.5B-Instruct☆30Updated 8 months ago
- ☆45Updated 5 months ago
- Prune transformer layers☆69Updated last year
- ☆209Updated 9 months ago
- LoRA and DoRA from Scratch Implementations☆211Updated last year
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆48Updated last year
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆117Updated 2 years ago
- LLaMA 2 implemented from scratch in PyTorch☆358Updated 2 years ago
- The official implementation of the paper "What Matters in Transformers? Not All Attention is Needed".☆177Updated 7 months ago
- Unofficial implementation of https://arxiv.org/pdf/2407.14679☆49Updated last year
- ☆224Updated last week
- Notes and commented code for RLHF (PPO)☆111Updated last year
- A set of scripts and notebooks on LLM finetunning and dataset creation☆110Updated last year
- Survey: A collection of AWESOME papers and resources on the latest research in Mixture of Experts.☆135Updated last year
- This repository contains an implementation of the LLaMA 2 (Large Language Model Meta AI) model, a Generative Pretrained Transformer (GPT)…☆72Updated 2 years ago
- ☆98Updated last year
- GPU Kernels☆203Updated 6 months ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆193Updated 4 months ago
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆121Updated 5 months ago
- ☆45Updated 5 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆300Updated 2 months ago
- A simplified implementation for experimenting with RLVR on GSM8K, This repository provides a starting point for exploring reasoning.☆136Updated 8 months ago
- Notes on quantization in neural networks☆104Updated last year
- ☆42Updated last year
- making the official triton tutorials actually comprehensible☆57Updated 2 months ago
- Reproduction of DeepSeek-R1☆240Updated 6 months ago
- Repo hosting codes and materials related to speeding LLMs' inference using token merging.☆36Updated 2 weeks ago