Owner

Lightning-AI

13 indexed repositories · View on GitHub

• pytorch-lightning

  Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

  Python · 31351 stars

• litgpt

  20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

  Python · 13670 stars

• lit-llama

  Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad...

  Python · 6082 stars

• LitServe

  A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.

  Python · 3942 stars

• torchmetrics

  Machine learning metrics for distributed, scalable PyTorch applications.

  Python · 2473 stars

• lightning-thunder

  PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri...

  Python · 1473 stars

• litData

  Speed up model training by fixing data loading.

  Python · 614 stars

• utilities

  Common Python utilities and GitHub Actions in Lightning Ecosystem

  Python · 62 stars

• litAI

  LLM router + minimal agent framework in one. Call any LLM API with OpenAI format. Unified billing, tools, retries, fallback, logging. Bui...

  Python · 52 stars

• LitModels

  Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-gr...

  Python · 40 stars

• LitLogger

  A minimal Python logger that tracks everything you try when building AI - metrics, prompts, models, etc, so you can see what changed and ...

  Python · 37 stars

• sdk

  SDK and CLI for lightning.ai platform. Develop and train AI with ease.

  Python · 10 stars

• Lightning-multinode-templates

  Multinode templates for Pytorch Lightning

  Python · 9 stars