Models and examples built with TensorFlow

The best ChatGPT that $100 can buy.

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

Open-Sora: Democratizing Efficient Video Production for All

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

gpt-oss-120b and gpt-oss-20b are two open-weight language models by OpenAI

PyTorch implementations of Generative Adversarial Networks.

Machine Learning Engineering Open Book

DALL·E Mini - Generate images from a text prompt

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

🐍 Geometric Computer Vision Library for Spatial AI

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.

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

AutoML library for deep learning

ImageBind One Embedding Space to Bind Them All

Matplotlib styles for scientific plotting

Home of StarCoder: fine-tuning & inference!

Find duplicate files

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams

NeMo Guardrails is an open-source toolkit for easily adding programmable guardrails to LLM-based conversational systems.

Open-source implementation of AlphaEvolve

SwinIR: Image Restoration Using Swin Transformer (official repository)

A data augmentations library for audio, image, text, and video.

Universal and Transferable Attacks on Aligned Language Models

Gemma open-weight LLM library, from Google DeepMind

A Social Media Enumeration & Correlation Tool by Jacob Wilkin(Greenwolf)

Collection of generative models in Tensorflow