PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.
Dépôts
Dépôts de Lightning-AI
(1 460 stars) (114 forks) (13 issues indexées) (13 good first issues ouvertes)
Lightning-AI/lit-llamaPython
Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.
(5 533 stars) (473 forks) (2 issues indexées) (2 good first issues ouvertes)
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
(26 687 stars) (3 233 forks) (121 issues indexées) (120 good first issues ouvertes)
Lightning-AI/utilitiesPython
Common Python utilities and GitHub Actions in Lightning Ecosystem
(62 stars) (23 forks) (1 issue indexée) (1 good first issue ouverte)