Machine Learning Awesome

• LLM
  • LLM Awesome
• Diffusion
  • Diffusion Awesome
• NLP
  • NLP Awesome
• GAN
• img2img
• Framework
  • PyTorch
    • by Meta
    • tinygrad/tinygrad
    • minitorch/minitorch
    • micrograd
  • Lightning-AI/pytorch-lightning
    • Apache-2.0, Python
  • JAX
    • by Google
    • google/maxtext
      • simple, performant and scalable Jax LLM
  • apple/corenet
  • Paddle
    • by 百度
  • OpenCV
  • Tensorflow
  • Keras
  • AutoKeras
  • FastAI
  • XGBoost
  • LightGBM
  • MegEngine
  • Scikit-Learn
  • microsoft/SynapseML
    • MIT, Scala
    • Distributed Machine Learning
  • caffe
  • flashlight/flashlight
    • C++ standalone library for machine learning
    • from Facebook AI Research Speech team, creators of Torch and Deep Speech
  • arrayfire/arrayfire
    • BSD-3, C++
    • general purpose GPU library
    • Binding: Python, Rust, Julia, NIM
    • WIP: .NET, Go, Java, Lua, JS, R, Ruby
  • google/jax
    • Apache-2.0, Python, C++
    • Autograd and XLA
    • 基础计算框架
    • google/flax
      • neural network
    • deepmind/rlax
      • reinforcement learning
    • deepmind/optax
      • gradient processing and optimization
    • deepmind/dm-haiku
      • neural network
    • deepmind/chex
• Fineturning
  • unslothai/unsloth
    • Apache-2.0, Python
    • Finetune Llama 3.1, Mistral, Phi & Gemma LLMs 2-5x faster with 80% less memory
• Labeling
  • CVAT
  • Label Studio
  • CVHub520/X-AnyLabeling
    • GPLv3, Python
• Deploy/Inference/Serving/Mobile/Optimization
  • openvinotoolkit/openvino
    • Apache-2.0, C++, Python
    • optimizing, deploying AI inference
    • by Intel
  • hpcaitech/ColossalAI
    • Apache-2.0, Python
  • microsoft/DeepSpeed
    • Apache-2.0, Python, C++
    • optimization library that makes distributed training and inference easy, efficient, and effective
  • Tencent/ncnn
    • BSD-3, C++, C
    • neural network inference framework optimized for the mobile platform
  • johnolafenwa/deepstack
    • Apache-2.0, Go, Python
    • Cross Platform AI Engine for Edge Devices
• MLOps
  • wandb
    • MIT, Python, Go
• Voice assistant/voice agent/voice chat/TTS
  • mezbaul-h/june
  • PromtEngineer/Verbi
  • bolna-ai/bolna
• VToonify
• music
  • AI-Guru/music-generation-research
• upscale
  • IBM/MAX-Image-Resolution-Enhancer
    • Apache-2.0, Python
    • Docker
  • upscayl/upscayl
    • AGPL-3.0
    • 需要 GPU
    • App 方式
    • Real-ESRGAN
  • philz1337x/clarity-upscaler
    • AGPLv3, Python
    • Magnific Alternative
  • Araxeus/PNG-Upscale
    • MIT, Java
  • 商业
    • https://magnific.ai/
• image
  • https://huggingface.co/datasets/vivym/midjourney-messages
  • openai/CLIP
    • MIT
    • CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image
• video
  • blakeblackshear/frigate
    • MIT, Python, TS
    • NVR with realtime local object detection for IP cameras
• text to image
  • Yutong-Zhou-cv/Awesome-Text-to-Image
• vision
  • facebookresearch/dinov2
    • Apache-2.0, PyTorch
    • 2023-04-14
    • DINOv2: Learning Robust Visual Features without Supervision
    • 用于特征提取，无监督
  • facebookresearch/dino
    • Apache-2.0, PyTorch
    • 2021
    • ViT
    • XCiT
    • DINO - Self-Supervised Vision Transformers
    • PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
• audio/music/speech/voice/tts
  • microsoft/muzic
  • yl4579/StyleTTS2
    • MIT, Python
    • HN
• diagram/Visualizer
  • lutzroeder/netron
  • https://alexlenail.me/NN-SVG/LeNet.html
  • HarisIqbal88/PlotNeuralNet
    • Latex code for making neural networks diagrams
  • https://www.tensorflow.org/tensorboard/graphs
  • stared/keras-sequential-ascii
    • ASCII
  • https://medium.com/inbrowserai/simple-diagrams-of-convoluted-neural-networks-39c097d2925b
• RT-DETR - Real-Time Detection Transformer
  • by Baidu
• SAM/Segment Anything Model
  • by Meta
  • facebookresearch/segment-anything
    • https://segment-anything.com
  • MobileSAM
  • FastSAM
• Visual recognition/object detection/segmentation
  • facebookresearch/detectron2
  • open-mmlab/mmdetection
  • google-research/big_vision
  • Yolo - You Only Look Once
    • YOLO-NAS - Neural Architecture Search
    • WongKinYiu/yolov7
    • YOLOv7 Breakdown
• lastmile-ai/aiconfig
  • MIT, Python
  • config-based framework to build generative AI applications
• Dataset
  • https://annas-archive.org/llm
  • https://www.opendatanetwork.com/
  • https://datasetsearch.research.google.com/
  • datumaro
    • 数据集管理
  • OpenOrca
    • SlimOrca
• Foundation
  • https://lfaidata.foundation/
  • https://ai.meta.com/
  • facebookresearch
  • https://ai.google/
  • https://labs.google/
  • https://ai.tencent.com/
  • NVlabs
    • NVIDIA Research
  • THUDM
    • 清华 KEG & 数据挖掘
• apps/ios/android
  • niedev/RTranslator
    • Apache-2.0, C++, Java
• PKU-YuanGroup/Video-LLaVA
• GokuMohandas/MadeWithML
• josephmisiti/awesome-machine-learning
• wunderwuzzi23/mlattacks
• tangramdotdev/tangram
  • HN
• facebookresearch/moco-v3
• facebookresearch/ParlAI
  • BlenderBot 2.0: An open source chatbot that builds long-term memory and searches the internet
  • InfoQ BlenderBot 2.0 Chatbot
• JianshuZhang/WAP
• salesforce/warp-drive
• isl-org/MiDaS
• NVlabs/stylegan3
• PaddlePaddle/Knover
• google-research-datasets/timedial
  • https://arxiv.org/abs/2106.04571
• google-research-datasets/disfl-qa
  • https://arxiv.org/abs/2106.04016
• vijaydwivedi75/gnn-lspe
• microsoft/onnxruntime
• PyTorch 1.10 HN
• facebookresearch/salina
• facebookresearch/ppuda
• wuhaozhe/style_avatar
• facebook/prophet
  • producing high quality forecasts for time series data
• facebookresearch/Kats
  • analyze time series data
• deepinsight/insightface
  • 2D and 3D Face Analysis
• VowpalWabbit/vowpal_wabbit
  • frontier of machine learning
• wav2vec
• https://nv-tlabs.github.io/editGAN/
• bryandlee/animegan2-pytorch AnimeGANv2 HN
• yeemachine/kalidokit
• google-research/ibc
  • Implicit Behavioral Cloning
• bhky/opennsfw2
  • Yahoo Open-NSFW model
• eugeneyan/applied-ml
• PeterL1n/RobustVideoMatting
• yuval-alaluf/hyperstyle
• SysCV/pcan
• tinyfool/VideoRemoveBackground
• open-mmlab/mmhuman3d
• BaltiApps/Pixelify-Google-Photos
• sedthh/pyxelate
• facebookresearch/theseus
• onion-liu/aahq-dataset
  • Artstation-Artistic-face-HQ Dataset (AAHQ)
• parrt/tensor-sensor
• mattbradley/dash
• openai/whisper
  • HN
  • https://openai.com/blog/whisper/
• https://banmo-www.github.io/
• https://nn-512.com/
• https://hypernerf.github.io/
• babysor/MockingBird
  • 5 秒内克隆您的声音并生成任意语音内容
  • HN
• https://mlconsole.com/
• AminRezaei0x443/memory-efficient-attention
  • Jax, PyTorch
• Perceiver IO: a scalable, fully-attentional model that works on any modality
• mchong6/JoJoGAN
• facebookresearch/SLIP
  • Self-supervision meets Language-Image Pre-training
• alibaba/DeepRec
  • recommendation engine
• open-mmlab/mmdeploy
• handtracking-io/yoha
• facebookresearch/Detic
• Machine-Learning-Tokyo/Interactive_Tools
• ashishpatel26/500-AI-Machine-learning-Deep-learning-Computer-vision-NLP-Projects-with-code
• kmario23/deep-learning-drizzle
• academic/awesome-datascience
• trekhleb/homemade-machine-learning
• donnemartin/data-science-ipython-notebooks
• https://github.com/facebookresearch/mae
  • Masked Autoencoders
• https://github.com/naver-ai/c3-gan
• kaegi/alass
  • Automatic Language-Agnostic Subtitle Synchronization
• SubSync: Subtitle Speech Synchronizer
  • HN
• gnes-ai/gnes
• apache/tvm
• facebookresearch/ConvNeXt
• NVlabs/instant-ngp
• lucidrains/RETRO-pytorch
• libffcv/ffcv
• johnnyzn/DW-GAN
  • CAPTCHA
• Kubasinska/MI-EEG-1D-CNN
• google-research/circuit_training
• Justin62628/Squirrel-RIFE
• automl/auto-sklearn
• open-mmlab/mmrotate
  • Rotated Object Detection
• google-research/frame-interpolation
  • Frame Interpolation for Large Motion
• patrick-kidger/diffrax
• dynamite-ready/movie-parser
• Nixtla/neuralforecast
  • forecasting algorithms for time series data
• ouhenio/stylegan3-projector
  • StyleGAN3 + Inversion
• victordibia/handtrack.js
  • HN
• facebookincubator/gloo
• horovod/horovod
  • Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet
• facebookresearch/contriever
• https://bishopfox.com/blog/unredacter-tool-never-pixelation
  • unmask
• jokenox/Goopt
  • 文本内容生成
• fastai/fastcore
• GroupViT: Semantic Segmentation Emerges from Text Supervision
• TorchStudio/torchstudio
• replicate/cog Containers for machine learning
• https://www.nvidia.com/en-us/studio/canvas/
• https://pimeyes.com/
• Parti: Pathways Autoregressive Text-to-Image Model
  • HN
• kuprel/min-dalle
  • DALL E Mini PyTorch
• borisdayma/dalle-mini
• Keytap3: check if your keyboard can be eavesdropped through a microphone
• iperov/DeepFaceLive
• TencentARC/GFPGAN
  • Real-world Face Restoration
• google-research/multinerf
  • Mip-NeRF 360, Ref-NeRF, and RawNeRF
• Running your own A.I. Image Generator with Latent-Diffusion
  • HN
• Anjok07/ultimatevocalremovergui
  • 移除人声
• NVIDIAGameWorks/kaolin-wisp
  • PyTorch library powered by NVIDIA Kaolin Core to work with neural fields
  • NeRFs, NGLOD, instant-ngp and VQAD
• nnaisense/evotorch
• Musico: AI Generated Music
  • HN
• Adventure game graphics with DALL-E 2
  • HN

---

• ml-tooling/best-of-ml-python
• TheAlgorithms/Python
• freeCodeCamp/freeCodeCamp

UI/Desktop/GUI/WebUI

• omnimodel
• Yonom/assistant-ui
  • React Components for AI Chat
• lobehub/lobe-chat
  • Apache-2.0, TypeScript
• JetBrains/KotlinDL
  • Kotlin DSL for ML

Library

• js/ts/nodejs
  • dexaai/openai-fetch
  • vercel/ai

Learn

• google-research/google-research
  • goemotions
• https://ai.googleblog.com/
• https://www.kdnuggets.com/
• https://stanford.edu/~shervine/teaching/
• DEEP LEARNING COURSE https://fleuret.org/dlc/
  • HN
• https://github.com/y33-j3T/Coursera-Deep-Learning
• 中文
  • d2l-ai/d2l-zh
    • https://zh.d2l.ai

---

Books

• Probabilistic Machine Learning: An Introduction
• An Introduction to Applied Bayesian Modeling https://www.bayesrulesbook.com/
  • https://xianblog.wordpress.com/2022/07/05/bayes-rules-book-review/

---

• Practical Deep Learning for Coders 2022
  • https://news.ycombinator.com/item?id=32186647
• dair-ai/ML-YouTube-Courses
• ML and NLP Research Highlights of 2021
• Machine Learning Algorithms Cheat Sheet
  • HN
• https://probml.github.io/pml-book/
• https://scikit-learn.org/stable/tutorial/machine_learning_map/index.html
• Neural network from scratch
  • HN
  • https://www.3blue1brown.com/topics/neural-networks
• Distributional Reinforcement Learning
• A visual introduction to machine learning
• Evaluating Syntactic Abilities of Language Models
• RLDS: An Ecosystem to Generate, Share, and Use Datasets in Reinforcement Learning
• https://learnaifromscratch.github.io/ai.html
• microsoft/ML-For-Beginners
• microsoft/Data-Science-For-Beginners
• LabML Neural Networks
  • PyTorch
• Reinforcement Learning: Theory and Algorithms
• How to replace estimations and guesses with a Monte Carlo simulation HN
• Neural Networks from Scratch
  • HN
• How to train large deep learning models as a startup HN
• Red Hot: The 2021 Machine Learning, AI and Data (MAD) Landscape
• Guide for building an End-to-End Logistic Regression Model
• SELF-PARKING CAR IN 500 LINES OF CODE
• facebookresearch/minihack
• Using Machine Learning to Denoise Images for Better OCR Accuracy
• Introduction to AutoEncoder and Variational AutoEncoder
• SimVLM: Simple Visual Language Model Pre-training with Weak Supervision
• Training a DCGAN in PyTorch
• Self-Supervised Reversibility-Aware Reinforcement Learning
• https://arxiv.org/pdf/2109.02869.pdf
  • https://www.youtube.com/watch?v=m8POyrJ7Pgg&t=1939s
  • https://attentionneuron.github.io/
• https://arxiv.org/abs/2106.01345
  • https://www.youtube.com/watch?v=83QN9S-0I84
  • https://github.com/kzl/decision-transformer
• http://proceedings.mlr.press/v139/vicol21a.html
• https://arxiv.org/abs/2010.01412
  • https://github.com/google-research/sam
  • https://github.com/davda54/sam
• Machine Learning Research Papers Released In 2021
• Interesting Algorithms Released By Google AI In 2021
• A visual introduction to machine learning
• https://brilliant.org/courses/intro-neural-networks/introduction-65/

Framework

• tensorflow/lingvo
  • building sequence models neural networks in Tensorflow
  • ASR, MT
• dlib
• clab/dynet
• CNTK
• mlpack
• SHARK
• Armadillo
• Faisis
• OpenNN
• FANN
• bennylp/awesome-cpp-ml
• Boosting
  • XGBoost
  • ThunderGBM
  • LightGBM
  • CatBoost
• Web
  • BrainJS/brain.js
    • MIT, TS
    • GPU accelerated Neural networks in JavaScript for Browsers and Node.js
  • @tensorflow/tfjs
    • https://www.tensorflow.org/js/
    • tfjs-vis
    • @tensorflow/tfjs-node
    • @tensorflow/tfjs-node-gpu - Linux
    • tensorflow/tfjs-models
  • spencermountain/compromise
    • MIT, JS
    • modest NLP
  • ml5js/ml5-library
    • 基于 TensorFlow.js
    • Blue Oak Model License 1.0.0 modified
  • NaturalNode/natural
    • MIT, JS
    • Tokenizer
    • String Distance
    • Stemmer
    • Bayesian & Logistic Regression Classifier
    • Maximum Entropy Classifier
    • Sentiment Analysis
    • WordNet - moos/wordnet-db
    • 无 中文 支持
    • NaturalNode/node-sylvester
      • vector, matrix, geometry for JS
    • NaturalNode/node-nltools
  • retextjs/retext
  • linonetwo/segmentit
    • 中文分词
  • wagenaartje/neataptic
    • neuro-evolution & backpropagation
    • 不在维护
  • cazala/synaptic
    • 不在维护
• POS Tagger - part-of-speech tagger
  • Eric Brill

---

• PyTorch vs TensorFlow in 2022
  • HN

Intrested

• ZHKKKe/MODNet
  • 背景消除
• tonybeltramelli/pix2code
  • GUI Screenshot -> Code

Models

• BERT
• Tacotron
• Wavenet/Waveglow/WaveRNN
• Eesen, Espresso, Kaldi, Wav2letter, NeMo
• VGG’16
• VGG’19
• ResNet50
• ResNet101
• ResNet152
• ResNet50v2
• ResNet101v2
• ResNet152v2
• MobileNet
• MobileNetv2
• https://modelplace.ai/models
• OpenBMB/BMList

STT

• snakers4/silero-models
• alphacep/vosk-api
  • Offline speech recognition API
  • Python, Java, C#, Node
  • 支持中文
  • alphacep/vosk-asterisk
    • res-speech-vosk - Asterisk 集成
• alphacep/vosk-android-demo
  • Offline speech recognition for Android with Vosk library
• kaldi-asr/kaldi
  • Speech Recognition Toolkit
• julius-speech/julius
  • Open-Source Large Vocabulary Continuous Speech Recognition Engine
• daanzu/kaldi-active-grammar
  • Python Kaldi speech recognition with grammars that can be set active/inactive dynamically at decode-time
• espnet/espnet
  • End-to-End Speech Processing Toolkit
• flashlight/wav2letter
  • Facebook AI Research's Automatic Speech Recognition Toolkit
• Nvidia/NeMo
  • toolkit for conversational AI
  • ASR, NLP, TTS
• PaddlePaddle/PaddleSpeech
  • ASR toolkit
  • 百度 Deep Speech: Scaling up end-to-end speech recognition
• mozilla/DeepSpeech
  • 基于 Tensorflow
• arjo129/uSpeech
  • Speech recognition toolkit for the arduino
• coqui-ai
  • TTS
  • STT - 没有中文模型
• synesthesiam/voice2json
  • 命令行工具
  • 中文模型基于 pocketsphinx
  • HN
• NATSpeech/NATSpeech
  • Non-Autoregressive Text-to-Speech (NAR-TTS) framework
• cmusphinx
  • 工作已经开始转移到 Kaldi, Vosk
  • cmusphinx/pocketsphinx

Hardware Platform

• RTX
• Colab Pro
• Paperspace Pro