NeMo Framework
Training Frameworks
NeMo Framework

Generative AI framework built for researchers and PyTorch developers working on Large Language Models (LLMs), Multimodal Models (MMs), Automatic Speech Recognition (ASR), Text to Speech (TTS), and Computer Vision (CV) domains.

GitHub
Generative AI framework built for researchers and PyTorch developers working on Large Language Models (LLMs), Multimodal Models (MMs), Automatic Speech Recognition (ASR), Text to Speech (TTS), and Computer Vision (CV) domains.

Relevant Sites

GPT-NeoX 7,128

An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.

Transformer Engine 2,269

A library for accelerating Transformer model training on NVIDIA GPUs.

maxtext 1,651

A simple, performant and scalable Jax LLM!

DeepSpeed 37,391

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

BMTrain 580

Efficient Training for Big Models.

veRL 4,759

veRL is a flexible and efficient RL framework for LLMs.

Leave a Reply

Your email address will not be published. Required fields are marked *