pytorch

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

huggingface.co/transformers https://github.com/huggingface/transformers

https://github.com/Lightning-AI/pytorch-lightning 🔗 lightning.ai Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

🔧🔗github hf diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

🕸️🔗huggingface.co/docs/diffusers

https://github.com/Lightning-AI/torchmetrics Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications. 🔗 https://lightning.ai/docs/torchmetrics/

https://github.com/Lightning-AI/lightning-thunder Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.

🔧🔗https://github.com/MinishLab/tokenlearn

Pre-train Static Word Embeddings

🔧🔗https://github.com/huggingface/lerobot

🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning

https://github.com/Lightning-AI/lightning-Habana Lightning support for Intel Habana accelerators.

https://github.com/coqui-ai/TTS 🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production

https://github.com/coqui-ai/Trainer 🐸 - A general purpose model trainer, as flexible as it gets

Real-time inference for Stable Diffusion - 0.88s latency. Covers AITemplate, nvFuser, TensorRT, FlashAttention. Join our Discord communty: https://discord.com/invite/TgHXuSJEk6