Self Supervised Learning Python Packages

pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

lightly

A python library for self-supervised learning on images.

lightly-utils

A python library for self-supervised learning on images.

dipy

DIPY is the paragon 3D/4D+ medical imaging library in Python. Contains generic methods for spatial normalization, signal processing, machine learning, statistical analysis and visualization of medical images. Additionally, it contains specialized methods for computational anatomy including diffusion, perfusion and structural imaging.

s3prl

Self-Supervised Speech Pre-training and Representation Learning Toolkit

mmcls

OpenMMLab Pre-training Toolbox and Benchmark

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

info-nce-pytorch

PyTorch implementation of the InfoNCE loss for self-supervised learning.

stepcount

Improved Step Counting via Foundation Models for Wrist-Worn Accelerometers

minerva

Minerva is a framework for training machine learning models for researchers.

paddlespeech

Easy-to-use Speech Toolkit including Self-Supervised Learning model, SOTA/Streaming ASR with punctuation, Streaming TTS with text frontend, Speaker Verification System, End-to-End Speech Translation and Keyword Spotting. Won NAACL2022 Best Demo Award.

byol-pytorch

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

biapy

Open source Python library for building bioimage analysis pipelines

actinet

An activity classification model based on self-supervised learning for wrist-worn accelerometer data.

gigaam

Foundational Model for Speech Recognition Tasks

libauc

LibAUC: A Deep Learning Library for X-Risk Optimization

contrastive-learner

A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network

otx

Build computer vision models in a fraction of the time and with less data.

novae

Graph-based foundation model for spatial transcriptomics data. Zero-shot spatial domain inference, batch-effect correction, and many other features.

paddlespeech-feat

Easy-to-use Speech Toolkit including Self-Supervised Learning model, SOTA/Streaming ASR with punctuation, Streaming TTS with text frontend, Speaker Verification System, End-to-End Speech Translation and Keyword Spotting. Won NAACL2022 Best Demo Award.

tabular-transformer

An end-to-end deep learning framework based on Transformer model, adapted for tabular data domain, support both supervised and self-supervised learning, used for regression, classification tasks.

paddleaudio

Easy-to-use Speech Toolkit including Self-Supervised Learning model, SOTA/Streaming ASR with punctuation, Streaming TTS with text frontend, Speaker Verification System, End-to-End Speech Translation and Keyword Spotting. Won NAACL2022 Best Demo Award.

accelbrainbeat

The purpose of this repository is to make prototypes as case study in the context of proof of concept(PoC) and research and development(R&D) that I have written in my website. The main research topics are Auto-Encoders in relation to the representation learning, the statistical machine learning for energy-based models, adversarial generation networks(GANs), Deep Reinforcement Learning such as Deep Q-Networks, semi-supervised learning, and neural network language model for natural language processing.

deepaudio-x

A python library to train Deep Neural Networks on various audio tasks using Self-Supervised backbones.