torchtitan

LLM trainer

A native PyTorch library for training large language models using distributed parallelism and optimization techniques.

A native PyTorch Library for large model training

GitHub

3k stars
44 watching
222 forks
Language: Python
last commit: almost 2 years ago
Linked from 1 awesome list


Backlinks from these awesome lists:

Related projects:

RepositoryDescriptionStars
pytorch/torchtuneA PyTorch library for easily authoring and experimenting with large language models4,479
rasbt/llms-from-scratchDeveloping and pretraining a GPT-like Large Language Model from scratch35,405
nvidia/apexTools for streamlined mixed precision and distributed training in PyTorch8,460
ajbrock/biggan-pytorchAn unofficial implementation of PyTorch BigGANs for generating high-fidelity natural images2,874
huggingface/accelerateA tool to simplify training and deployment of PyTorch models on various devices and configurations8,056
nvidia/megatron-lmA framework for training large language models using scalable and optimized GPU techniques10,804
turboderp/exllamaA re-implementation of Llama for efficient use with quantized weights on modern GPUs.2,783
pytorch/igniteA high-level library to help with training and evaluating neural networks in PyTorch4,554
pytorch/pytorchA Python library providing tensors and dynamic neural networks with strong GPU acceleration84,978
ahmedfgad/torchgaTrains PyTorch models using a genetic algorithm96
alpha-vllm/llama2-accessoryAn open-source toolkit for pretraining and fine-tuning large language models2,732
pytorchbearer/torchbearerA PyTorch model fitting library designed to simplify the process of training deep learning models.636
kaiyangzhou/deep-person-reidA PyTorch library for training and retraining deep neural networks for person re-identification in images and videos.4,353
lyhue1991/torchkerasA PyTorch-based model training framework designed to simplify and streamline training workflows by providing a unified interface for various loss functions, optimizers, and validation metrics.1,822
jfzhang95/pytorch-deeplab-xceptionA PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets2,919