zuenko

barlowtwins

PyTorch implementation of Barlow Twins.

vissl

VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.

swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

hydra

Hydra is a framework for elegantly configuring complex applications

pytorch-lars

"Layer-wise Adaptive Rate Scaling" in PyTorch

coolgpus

GPU fan control for headless Linux

albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

cl-ica

Code for the paper "Contrastive Learning Inverts the Data Generating Process".

DALI_pytorch_demo

Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia example, namely a completely CPU pipeline & improved memory usage

dali-pytorch

A PyTorch toolkit for extremely fast ImageNet training with NVIDIA DALI.

sam

SAM: Sharpness-Aware Minimization (PyTorch)

nfnets-pytorch

NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/

gpytorch

A highly efficient implementation of Gaussian Processes in PyTorch

pytorch-optimizer

torch-optimizer -- collection of optimizers for Pytorch

category_encoders

A library of sklearn compatible categorical variable encoders

langdetect

Port of Google's language-detection library to Python.

deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications

DiffMorph

Image morphing without reference points by applying warp maps and optimizing over them.

byol-pytorch

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch

tinkoff-api

Python Tinkoff API client for asyncio and humans

Fewshot_Detection

Few-shot Object Detection via Feature Reweighting

Few-Shot-Learning

pytorch-classification-uncertainty

This repo contains a PyTorch implementation of the paper: "Evidential Deep Learning to Quantify Classification Uncertainty"

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

finnhub-python

Finnhub Python API Client. Finnhub API provides institutional-grade financial data to investors, fintech startups and investment firms. We support real-time stock price, global fundamentals, global ETFs holdings and alternative data. https://finnhub.io/docs/api

FinRL

FinRL: Financial Reinforcement Learning. 🔥

FinRL-Trading

For trading. Please star.

SuperGluePretrainedNetwork

SuperGlue: Learning Feature Matching with Graph Neural Networks (CVPR 2020, Oral)

liif

Learning Continuous Image Representation with Local Implicit Image Function, in CVPR 2021 (Oral)

LightAutoML

LAMA - automatic model creation framework