Repository Index

NVR with realtime local object detection for IP cameras

• rtsp
• realtime
• tensorflow
• google-coral

Rembg is a tool to remove images background

• image-processing
• background-removal
• python

Ongoing research training transformer models at scale

• large-language-models
• model-para
• transformers

Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 600+ LLMs (Qwen3.8, DeepSeek-V4, GLM-5.1, InternLM3, Llama4, ...) and 300+ MLLMs (Qwen3-VL, Qwen3-Omni, InternVL3.5, Ovis2.5, GLM5.3, Gemma4, Llava, Phi4, ...) (AAAI 2025).

• llm
• lora
• sft
• multimodal

Collection of scripts to aggregate image data for the purposes of training an NSFW Image Classifier

• nsfw-classifier
• nsfw
• deep-learning
• content-moderation

Implementation of Nougat Neural Optical Understanding for Academic Documents

Pytorch implementation of convolutional neural network visualization techniques

• grad-cam
• cam
• pytorch
• guided-backpropagation

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

• deepspeed-library
• gpt-3
• transformers
• language-model

[ICCV 2023] ProPainter: Improving Propagation and Transformer for Video Inpainting

• object-removal
• video-completion
• video-inpainting
• video-outpainting

The repository host gives no description.

• research

Official Tensorflow implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation (ICLR 2020)

The central nervous system and backend orchestrator for Fractal. It handles multi-tenant task scheduling, real-time TFLOPs budgeting, and federated weight aggregation using FedAvg.

• backend
• fedavg
• federated-learning
• flask

Language model tokenization at GB/s

• llm
• nlp
• tokenization
• tokenizer

Recommendation Algorithm大规模推荐算法库，包含推荐系统经典及最新算法LR、Wide&Deep、DSSM、TDM、MIND、Word2Vec、Bert4Rec、DeepWalk、SSR、AITM，DSIN，SIGN，IPREC、GRU4Rec、Youtube_dnn、NCF、GNN、FM、FFM、DeepFM、DCN、DIN、DIEN、DLRM、MMOE、PLE、ESMM、ESCMM, MAML、xDeepFM、DeepFEFM、NFM、AFM、RALM、DMR、GateNet、NAML、DIFM、Deep Crossing、PNN、BST、AutoInt、FGCNN、FLEN、Fibinet、ListWise、DeepRec、ENSFM，TiSAS，AutoFIS等，包含经典推荐系统数据集criteo 、movielens等

• ple
• deepfm
• mmoe
• word2vec

🔥🔥High-Performance Face Recognition Library on PaddlePaddle & PyTorch🔥🔥

• pytorch
• face-recognition
• face-detection
• face-alignment

Graph Attention Networks (https://arxiv.org/abs/1710.10903)

• graph-attention-networks
• attention-mechanism
• self-attention
• tensorflow

DeepFill v1/v2 with Contextual Attention and Gated Convolution, CVPR 2018, and ICCV 2019 Oral

• deepfill
• image-inpainting
• generative-adversarial-network
• attention-model

An open-source project dedicated to tracking and segmenting any objects in videos, either automatically or interactively. The primary algorithms utilized include the Segment Anything Model (SAM) for key-frame segmentation and Associating Objects with Transformers (AOT) for efficient tracking and propagation purposes.

• interactive-segmentation
• segment-anything
• segment-anything-model
• video-object-segmentation

Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)

• graph-attention-networks
• attention-mechanism
• self-attention
• pytorch

Image to prompt with BLIP and CLIP

• clip
• pytorch

A comprehensive set of fairness metrics for datasets and machine learning models, explanations for these metrics, and algorithms to mitigate bias in datasets and models.

• ai
• fairness-ai
• fairness
• fairness-testing

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

• gat
• graph-attention-networks
• attention-mechanism
• self-attention

Official PyTorch implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation

simple neural network library in C99

• backpropagation
• genetic-algorithm
• ann
• neural-network