Stars
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Models and examples built with TensorFlow
The Python micro framework for building web applications.
scikit-learn: machine learning in Python
Rich is a Python library for rich text and beautiful formatting in the terminal.
Fine-tuning & Reinforcement Learning for LLMs. 🦥 Train OpenAI gpt-oss, DeepSeek-R1, Qwen3, Gemma 3, TTS 2x faster with 70% less VRAM.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
💫 Industrial-strength Natural Language Processing (NLP) in Python
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.
🗃 Open source self-hosted web archiving. Takes URLs/browser history/bookmarks/Pocket/Pinboard/etc., saves HTML, JS, PDFs, media, and more...
Image-to-Image Translation in PyTorch
Open standard for machine learning interoperability
Magenta: Music and Art Generation with Machine Intelligence
A TTS model capable of generating ultra-realistic dialogue in one pass.
Accelerate your web app development | Build fast. Run fast.
Luigi is a Python module that helps you build complex pipelines of batch jobs. It handles dependency resolution, workflow management, visualization etc. It also comes with Hadoop support built in.
Official repository for IPython itself. Other repos in the IPython organization contain things like the website, documentation builds, etc.
StyleGAN - Official TensorFlow Implementation
Python package built to ease deep learning on graph, on top of existing DL frameworks.
Ongoing research training transformer models at scale
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
CTF framework and exploit development library