Stars
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
Fully open reproduction of DeepSeek-R1
Faster R-CNN (Python implementation) -- see https://github.com/ShaoqingRen/faster_rcnn for the official MATLAB version
The GitHub repository for the paper "Informer" accepted by AAAI 2021.
Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams
Robust recipes to align language models with human and AI preferences
Open-source, accurate and easy-to-use video speech recognition & clipping tool, LLM based AI clipping intergrated.
PyTorch implementation of DQN, AC, ACER, A2C, A3C, PG, DDPG, TRPO, PPO, SAC, TD3 and ....
Evolutionary Scale Modeling (esm): Pretrained language models for proteins
PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
[CVPR 2023] DepGraph: Towards Any Structural Pruning
Using modified BiSeNet for face parsing in PyTorch
micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-…
A toolkit for reproducible reinforcement learning research.
Author's PyTorch implementation of TD3 for OpenAI gym tasks
Master Federated Learning in 2 Hours—Run It on Your PC!
[ICLR 2023 Oral] Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Model
[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression
[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".
[NeurIPS 2024] SimPO: Simple Preference Optimization with a Reference-Free Reward
A library with extensible implementations of DPO, KTO, PPO, ORPO, and other human-aware loss functions (HALOs).
Federated Optimization in Heterogeneous Networks (MLSys '20)
A repository in preparation for open-sourcing lottery ticket hypothesis code.
🧬 Nucleotide Transformer: Building and Evaluating Robust Foundation Models for Human Genomics