CoreNet: A library for training deep neural networks

Official Code for "SVD-LLM: Truncation-aware Singular Value Decomposition for Large Language Model Compression"

An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.

[ICLR 2024] Fine-tuning LLaMA to follow Instructions within 1 Hour and 1.2M Parameters

PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.

This repository contains small projects related to Neural Networks and Deep Learning in general. Subjects are closely linekd with articles I publish on Medium. I encourage you both to read as well …

Free, ultrafast Copilot alternative for Vim and Neovim

A helpful 5-page machine learning cheatsheet to assist with exam reviews, interview prep, and anything in-between.

Geom_GCN re-implementation using only pytorch. ( Not using DGL)

[TPAMI 2022] "Bag of Tricks for Training Deeper Graph Neural Networks A Comprehensive Benchmark Study" by Tianlong Chen*, Kaixiong Zhou*, Keyu Duan, Wenqing Zheng, Peihao Wang, Xia Hu, Zhangyang Wang

Pytorch Implementation of Rethinking Graph Neural Architecture Search from Message-passing (CVPR2021)

Object Classification Training Framework

Code for our ICML'2020 paper "Stabilizing Differentiable Architecture Search via Perturbation-based Regularization"

[ACM'MM 2020] "MM-Hand: 3D-Aware Multi-Modal Guided Hand Generative Network for 3D Hand Pose Synthesis" Zhenyu Wu, Duc Hoang, Shih-Yao Lin, Yusheng Xie, Liangjian Chen, Yen-Yu Lin, Zhangyang Wang, …

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

starter from "How to Train a GAN?" at NIPS2016

Code of single-view depth prediction algorithm on Internet Photos described in "MegaDepth: Learning Single-View Depth Prediction from Internet Photos, Z. Li and N. Snavely, CVPR 2018".

PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis

Depth-Aware Video Frame Interpolation (CVPR 2019)