Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Models and examples built with TensorFlow
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
⚡ A Fast, Extensible Progress Bar for Python and CLI
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Image-to-Image Translation in PyTorch
LabelImg is now part of the Label Studio community. The popular image annotation tool created by Tzutalin is no longer actively being developed, but you can check out Label Studio, the open source …
Repository to track the progress in Natural Language Processing (NLP), including the datasets and the current state-of-the-art for the most common NLP tasks.
Graph Neural Network Library for PyTorch
Magenta: Music and Art Generation with Machine Intelligence
OpenAI Baselines: high-quality implementations of reinforcement learning algorithms
Image augmentation for machine learning experiments.
Python Implementation of Reinforcement Learning: An Introduction
Machine Learning Engineering Open Book
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
Bayesian Modeling and Probabilistic Programming in Python
Deep universal probabilistic programming with Python and PyTorch
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
A Python implementation of global optimization with gaussian processes.
Efficiently computes derivatives of NumPy code.
A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv
Count the MACs / FLOPs of your PyTorch model.