NVIDIA / NeMo
A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
β13,126Updated this week
Alternatives and similar repositories for NeMo:
Users that are interested in NeMo are comparing it to the libraries listed below
- Ongoing research training transformer models at scaleβ11,377Updated this week
- Fast and memory-efficient exact attentionβ15,503Updated this week
- π A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (iβ¦β8,325Updated this week
- Development repository for the Triton language and compilerβ14,406Updated this week
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) trainingβ21,351Updated 6 months ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.β36,746Updated this week
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalitiesβ20,747Updated last week
- Unsupervised text tokenizer for Neural Network-based text generation.β10,585Updated last week
- Transformer related optimization, including BERT, GPTβ6,022Updated 10 months ago
- Large Language Model Text Generation Inferenceβ9,756Updated this week
- Train transformer language models with reinforcement learning.β11,688Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMsβ38,093Updated this week
- [NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.β21,444Updated 6 months ago
- π€ PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.β17,319Updated this week
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterβ¦β13,944Updated 6 months ago
- Accessible large language models via k-bit quantization for PyTorch.β6,658Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.β9,053Updated this week
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"β6,267Updated 4 months ago
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.β28,990Updated this week
- CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an imageβ27,379Updated 6 months ago
- PyTorch extensions for high performance and large scale training.β3,259Updated last month
- End-to-End Speech Processing Toolkitβ8,775Updated last week
- TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that containβ¦β9,422Updated this week
- Build and share delightful machine learning apps, all in Python. π Star to support our work!β36,108Updated this week
- π₯ Fast State-of-the-Art Tokenizers optimized for Research and Productionβ9,378Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and moreβ31,283Updated this week
- Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"β11,290Updated 2 months ago
- π€ Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.β27,531Updated this week
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)β¦β13,160Updated last week
- Fast inference engine for Transformer modelsβ3,592Updated last week