Return to Projects

Frameworks for GenAI Builders
Lightning AI


Lightning AI has been maintaining key open source frameworks widely used today in GenAI and general deep learning. PyTorch Lightning, Lightning Fabric and TorchMetrics are powering a portion of the GenAI ecosystem today (e.g. StableDiffusion, NeMO, TinyLlama).

We are open to collaborations and integration at the level of the Lightning open source stack, which includes lit-gpt and lit-llama (minimalistic, high-performance LLM implementations).

As part of the effort, we are creating open-source resources on the Lightning Platform in the form of fully open-source, reproducible environments (that might or might not use the Lightning stack). These are not locked into the platform: no proprietary SDK is used; everything is fully open source; the platform facilitates consumption, reproducibility, and reuse.