i-mein

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

automl

Google Brain AutoML

tensortrade

An open source reinforcement learning framework for training, evaluating, and deploying robust trading agents.

alpha_vantage

A python wrapper for Alpha Vantage API for financial data.

hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.

shapash

🔅 Shapash: User-friendly Explainability and Interpretability to Develop Reliable and Transparent Machine Learning Models

koila

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

grad-cam

[ICCV 2017] Torch code for Grad-CAM

COVID-Net

COVID-Net Open Source Initiative

lofo-importance

Leave One Feature Out Importance

python3-krakenex

REST Exchange API for Kraken.com, Python 3

tph-yolov5

bigbird

Transformers for Longer Sequences

Transformer_Time_Series

Enhancing the Locality and Breaking the Memory Bottleneck of Transformer on Time Series Forecasting (NeurIPS 2019)

convit

Code for the Convolutional Vision Transformer (ConViT)

rexnet

Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models

Holocron

PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix

SAB-cnn-audio-denoiser

Tensorflow 2.0 implementation of the paper: A Fully Convolutional Neural Network for Speech Enhancement

HugsVision

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

SIIM-COVID19-Detection

1st place solution for SIIM-FISABIO-RSNA COVID-19 Detection Challenge

ds_bowl_2018

Kaggle Data Science Bowl 2018

ceecnet

source code for the task of semantic change detection (built with mxnet)

Cryptocurrency_TimeSeries_Analysis

Code used in the crypto price forecasting project (see medium post linked here)

HubMap

Kaggle HubMap

open-images-2019-instance-segmentation

7th place solution to the Open Images 2019 Instance Segmentation competition

TAUP-PyTorch

Task Agnostic Unsupervised Learning

ifl_unsup_anom_det

GWC_YOLOv5

Customized YOLOv5 for Global Wheat Challenge 2021

EnhancedLSTM4RUL

In this enhanced deep LSTM approach, clustering is performed for all collected sensors data and operational monitoring information, using Gaussian distributions. These Gaussian mixtures clustering over the entire collected data leads to an adequate and specific deep neural network for RUL prediction.

Kaggle_TGBR

Kaggle-TensorFlow - Help Protect the Great Barrier Reef