On-device AI across mobile, embedded and edge for PyTorch

Generative Self-Supervised Learning for Medical Image Classification (ACCV 2024)

Profiling and inspecting memory in pytorch

[CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".

Official Repository for "HydraViT: Stacking Heads for a Scalable ViT" (NeurIPS'24)

Code release for NeurIPS 2020 paper "Co-Tuning for Transfer Learning"

Official implementation of Dynamic Perceiver

Fast and memory-efficient exact attention

[ACCV2024] Official PyTorch code for Neural Substitution for Branch-level Network Re-parameterization

Distributed preprocessing and data loading for language datasets

Ongoing research training transformer models at scale

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"

PyCIL: A Python Toolbox for Class-Incremental Learning

Continual learning baselines and strategies from popular papers, using Avalanche. We include EWC, SI, GEM, AGEM, LwF, iCarl, GDumb, and other strategies.

Continual Learning tutorials and demo running on Google Colaboratory.

Spring Boot로 구성된 러너스(runus) 앱 서버 repo입니다.

(CVPR2024)RMT: Retentive Networks Meet Vision Transformer

High-Resolution Image Synthesis with Latent Diffusion Models

This is a pytorch implementation of k-means clustering algorithm

Fast Open-Source Search & Clustering engine × for Vectors & 🔜 Strings × in C++, C, Python, JavaScript, Rust, Java, Objective-C, Swift, C#, GoLang, and Wolfram 🔍

A library for efficient similarity search and clustering of dense vectors.

Learning from synthetic data - code and models

[CVPR 2023] DepGraph: Towards Any Structural Pruning

Neuron Merging: Compensating for Pruned Neurons (NeurIPS 2020)

PruningBench: A Comprehensive Benchmark of Structural Pruning

Official Code of Paper "Reversible Column Networks" "RevColv2"