NVIDIA-NeMo / AutomodelLinks
Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support
☆266Updated this week
Alternatives and similar repositories for Automodel
Users that are interested in Automodel are comparing it to the libraries listed below
Sorting:
- Training library for Megatron-based models with bidirectional Hugging Face conversion capability☆400Updated this week
- Megatron's multi-modal data loader☆315Updated last week
- Memory optimized Mixture of Experts☆72Updated 6 months ago
- Load compute kernels from the Hub☆389Updated this week
- ArcticInference: vLLM plugin for high-throughput, low-latency inference☆384Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆279Updated 2 months ago
- ☆132Updated 8 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆334Updated 3 months ago
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆272Updated last week
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆137Updated last year
- KV cache compression for high-throughput LLM inference☆151Updated last year
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆219Updated last week
- Efficient LLM Inference over Long Sequences☆394Updated 7 months ago
- ☆230Updated 2 months ago
- Block Diffusion for Ultra-Fast Speculative Decoding☆459Updated this week
- QeRL enables RL for 32B LLMs on a single H100 GPU.☆481Updated 2 months ago
- A unified library for building, evaluating, and storing speculative decoding algorithms for LLM inference in vLLM☆220Updated this week
- Official implementation for Training LLMs with MXFP4☆118Updated 9 months ago
- The evaluation framework for training-free sparse attention in LLMs☆114Updated last week
- Accelerating MoE with IO and Tile-aware Optimizations☆563Updated 2 weeks ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆229Updated 7 months ago
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆288Updated 2 months ago
- LM engine is a library for pretraining/finetuning LLMs☆113Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆93Updated last year
- TPU inference for vLLM, with unified JAX and PyTorch support.☆223Updated this week
- Triton-based implementation of Sparse Mixture of Experts.☆263Updated 4 months ago
- A framework to study AI models in Reasoning, Alignment, and use of Memory (RAM).☆344Updated last month
- PyTorch-native post-training at scale☆605Updated this week
- 🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation…☆116Updated 2 months ago
- A tool to configure, launch and manage your machine learning experiments.☆215Updated last week