Popular GitHub Repositories for Sharpness Aware Minimization

The code for the paper 'Heterogeneous Risk Minimization' of ICML2021.

Context aware, pluggable and customizable data protection and de-identification SDK for text and images

A Streamlit-powered Chat Assistant utilizing Ollama's DeepSeek-R1 (1.5B) model for intelligent, context-aware conversations. Designed with an interactive chat interface, adjustable model parameters. It leverages Ollama for local model serving and Streamlit for a responsive UI for text-based interactions.

Sparsity-aware deep learning inference runtime for CPUs

High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.

Jupyter extensions that help you write code faster: Context aware AI Chat, Autocomplete, and Spreadsheet

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-Arithmetic-Only Inference)、Low-Bit(≤2b)/Ternary and Binary(TWN/BNN/XNOR-Net); post-training-quantization(PTQ), 8-bit(tensorrt); 2、 pruning: normal、regular and group convolutional channel pruning; 3、 group convolution structure; 4、batch-normalization fuse for quantization. deploy: tensorrt, fp32/fp16/int8(ptq-calibration)、op-adapt(upsample)、dynamic_shape

GANimation: Anatomically-aware Facial Animation from a Single Image (ECCV'18 Oral) [PyTorch]

Ultra Fast Structure-aware Deep Lane Detection (ECCV 2020)

Bi-directional Attention Flow (BiDAF) network is a multi-stage hierarchical process that represents context at different levels of granularity and uses a bi-directional attention flow mechanism to achieve a query-aware context representation without early summarization.

? An Open Source Texas Hold'em AI

[ICLR 2019] ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardware

[ICCV 2023] StableVideo: Text-driven Consistency-aware Diffusion Video Editing

KGAT: Knowledge Graph Attention Network for Recommendation, KDD2019

R-KV: Redundancy-aware KV Cache Compression for Reasoning Models

Official code for CVPR2022 paper: Depth-Aware Generative Adversarial Network for Talking Head Video Generation

A library with extensible implementations of DPO, KTO, PPO, ORPO, and other human-aware loss functions (HALOs).

Empower Large Language Models (LLM) using Knowledge Graph based Retrieval-Augmented Generation (KG-RAG) for knowledge intensive tasks

Empower Sequence Labeling with Task-Aware Language Model

Official PyTorch implementation of "Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB Image", ICCV 2019