Stars
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
FastAPI framework, high performance, easy to learn, fast to code, ready for production
Models and examples built with TensorFlow
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…
Scrapy, a fast high-level web crawling & scraping framework for Python.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Open source platform for the machine learning lifecycle
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
An open source implementation of CLIP.
Geometric Computer Vision Library for Spatial AI
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
Hackable and optimized Transformers building blocks, supporting a composable construction.
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
Distributed Asynchronous Hyperparameter Optimization in Python
Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams
High-resolution models for human tasks.
NeuralProphet: A simple forecasting package
Denoising Diffusion Probabilistic Models
GLIDE: a diffusion-based text-conditional image synthesis model
A library for transfer learning by reusing parts of TensorFlow models.
Siamese and triplet networks with online pair/triplet mining in PyTorch