Simple and efficient DeepSeek V3 SFT using pipeline parallel and expert parallel, with both FP8 and BF16 trainings

Various AI scripts. Mostly Stable Diffusion stuff.

Hallo: Hierarchical Audio-Driven Visual Synthesis for Portrait Image Animation

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

Audio generation using diffusion models, in PyTorch.

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀

Code release for ConvNeXt model

1st place solution for SIIM-FISABIO-RSNA COVID-19 Detection Challenge

NeurIPS 2020, Debiased Contrastive Learning

Data augmentation for NLP

news-please - an integrated web crawler and information extractor for news that just works.

News, full-text, and article metadata extraction in Python 3. Advanced docs:

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

Contrastive Language-Image Pretraining

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding

五个一航班查询app

Distributed, mixed-precision training with PyTorch

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

YOLOv5 in PyTorch > ONNX > CoreML > iOS