meta-pytorch / torchforgeLinks
PyTorch-native post-training at scale
☆479Updated last week
Alternatives and similar repositories for torchforge
Users that are interested in torchforge are comparing it to the libraries listed below
Sorting:
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆301Updated this week
- Simple & Scalable Pretraining for Neural Architecture Research☆298Updated last week
- Scalable toolkit for efficient model reinforcement☆1,009Updated this week
- Load compute kernels from the Hub☆316Updated last week
- Checkpoint-engine is a simple middleware to update model weights in LLM inference engines☆804Updated this week
- Dion optimizer algorithm☆379Updated last week
- ☆877Updated this week
- A scalable asynchronous reinforcement learning implementation with in-flight weight updates.☆270Updated last week
- ☆225Updated 2 weeks ago
- Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…☆353Updated 10 months ago
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆441Updated 2 weeks ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆270Updated this week
- PyTorch building blocks for the OLMo ecosystem☆314Updated this week
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆344Updated 6 months ago
- Async RL Training at Scale☆734Updated last week
- Memory optimized Mixture of Experts☆69Updated 3 months ago
- ☆218Updated 9 months ago
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆241Updated this week
- Physics of Language Models, Part 4☆255Updated 3 months ago
- ☆545Updated last month
- Normalized Transformer (nGPT)☆192Updated 11 months ago
- An extension of the nanoGPT repository for training small MOE models.☆207Updated 7 months ago
- Training API☆202Updated 3 weeks ago
- PyTorch Single Controller☆869Updated this week
- Scalable and Performant Data Loading☆331Updated this week
- LLM KV cache compression made easy☆674Updated this week
- Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support☆147Updated this week
- Reverse Engineering Gemma 3n: Google's New Edge-Optimized Language Model☆250Updated 5 months ago
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA (+ more DSLs)☆642Updated this week
- Open-source framework for the research and development of foundation models.☆574Updated last week