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.
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
PyTorch Tutorial for Deep Learning Researchers
Data science Python notebooks: Deep learning (TensorFlow, Theano, Caffe, Keras), scikit-learn, Kaggle, big data (Spark, Hadoop MapReduce, HDFS), matplotlib, pandas, NumPy, SciPy, Python essentials,…
A cross-platform command-line utility that creates projects from cookiecutters (project templates), e.g. Python package projects, C projects.
The open source developer platform to build AI agents and models with confidence. Enhance your AI applications with end-to-end tracking, observability, and evaluations, all in one integrated platform.
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Graph Neural Network Library for PyTorch
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.
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.
Datasets, Transforms and Models specific to Computer Vision
1 Line of code data quality profiling & exploratory data analysis for Pandas and Spark DataFrames.
A paper list of object detection using deep learning.
A logical, reasonably standardized, but flexible project structure for doing and sharing data science work.
A PyTorch implementation of the Transformer model in "Attention is All You Need".
Deep universal probabilistic programming with Python and PyTorch
tensorboard for pytorch (and chainer, mxnet, numpy, ...)
Most popular metrics used to evaluate object detection algorithms.
PyTorch deep learning projects made easy.
Simple examples to introduce PyTorch
An easy to use PyTorch to TensorRT converter
A flask template with Bootstrap, asset bundling+minification with webpack, starter templates, and registration/authentication. For use with cookiecutter.
Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.
Model summary in PyTorch similar to `model.summary()` in Keras
An all-in-one Docker image for deep learning. Contains all the popular DL frameworks (TensorFlow, Theano, Torch, Caffe, etc.)
Semantic Segmentation Architectures Implemented in PyTorch
A PyTorch implementation of the YOLO v3 object detection algorithm
A library for Multilingual Unsupervised or Supervised word Embeddings
Siamese and triplet networks with online pair/triplet mining in PyTorch
