Stars
ComfyUI custom nodes for Ovi joint video+audio generation
Simple and efficient DeepSeek V3 SFT using pipeline parallel and expert parallel, with both FP8 and BF16 trainings
text and image to video generation: CogVideoX (2024) and CogVideo (ICLR 2023)
A powerful tool that translates ComfyUI workflows into executable Python code.
[CVPR2024, Highlight] Official code for DragDiffusion
Audio generation using diffusion models, in PyTorch.
Implementation of Video Diffusion Models, Jonathan Ho's new paper extending DDPMs to Video Generation - in Pytorch
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
news-please - an integrated web crawler and information extractor for news that just works
newspaper3k is a news, full-text, and article metadata extraction in Python 3. Advanced docs:
Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.
Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method.
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch
A Mac tool that finds available delivery slots for Amazon's Whole Foods delivery and Amazon Fresh services
A Chrome extension helping users access their web services while traveling outside mainland China
Progressive Growing of GANs for Improved Quality, Stability, and Variation
StyleGAN - Official TensorFlow Implementation
Keras implementation of ArcFace, CosFace, and SphereFace
A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility
Tensorflow implementation of "Generalized End-to-End Loss for Speaker Verification"
https://www.kaggle.com/c/humpback-whale-identification
Rewrite the LOUPE library (https://github.com/antoine77340/LOUPE) into Keras version. Many learnable pooling or differentiable aggregation methods are covered. (NetVLAD, NetRVLAD, SoftDBoW, NetFV, CG)
Keras implementation of Representation Learning with Contrastive Predictive Coding
CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch