Rich is a Python library for rich text and beautiful formatting in the terminal.

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

You like pytorch? You like micrograd? You love tinygrad! ❤️

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Faker is a Python package that generates fake data for you.

Python composable command line interface toolkit

The official GitHub page for the survey paper "A Survey of Large Language Models".

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Refine high-quality datasets and visual AI models

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

The source code that powers readthedocs.org

Plumb a PDF for detailed information about each char, rectangle, line, et cetera — and easily extract text and tables.

A Python library for the Docker Engine API

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Simple Online Realtime Tracking with a Deep Association Metric

Flow-based visual scripting for Python

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"

Making data higher-quality, juicier, and more digestible for foundation models! 🍎 🍋 🌽 ➡️ ➡️🍸 🍹 🍷为大模型提供更高质量、更丰富、更易”消化“的数据！

Transformer: PyTorch Implementation of "Attention Is All You Need"

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

SSD: Single Shot MultiBox Detector | a PyTorch Tutorial to Object Detection

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

The Python Dict that's better than heroin.

🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch

A Python library to use infix notation in Python

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

Python 3.8+ toolbox for submitting jobs to Slurm

No-code in the front, Python in the back. An open-source framework for creating data apps.