pytorch

The collaboration workspace for Machine Learning

Open Source Continuous Inference Benchmark Research Platform — Kimi K3 2.8T, MiniMax M3, DeepSeekv4, GLM5 - GB200 NVL72 vs MI355X vs B200 vs GB300 NVL72 & soon™ TPUv6e/v7/Trainium2/3 | 开源持续推理基准研究平台 — Kimi K2.7-Code、MiniMax M3、DeepSeekv4、GLM5 - GB200 NVL72 vs MI355X vs B200 vs GB300 NVL72，即将推出™ TPUv6e/v7/Trainium2/3

Self-host the powerful Chatterbox TTS model. This server offers a user-friendly Web UI, flexible API endpoints (incl. OpenAI compatible), predefined voices, voice cloning, and large audiobook-scale text processing. Runs accelerated on NVIDIA (CUDA), AMD (ROCm), and CPU.

Korean BERT pre-trained cased (KoBERT)

🛸 Use pretrained transformers like BERT, XLNet and GPT-2 in spaCy

Calculate token/s & GPU memory requirement for any LLM. Supports llama.cpp/ggml/bnb/QLoRA quantization

A group of notebooks and other files which can help you learn AI from scratch.

🩹Editing large language models within 10 seconds⚡

Scalable and memory-optimized training of diffusion models

"Deep Generative Modeling": Introductory Examples

An elegent pytorch implement of transformers

https://wavespeed.ai/ Best inference performance optimization framework for HuggingFace Diffusers on NVIDIA GPUs.

Proximal Policy Optimization (PPO) algorithm for Super Mario Bros

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

Examples of programs built using Modal

Paint by Example: Exemplar-based Image Editing with Diffusion Models

Pocket-Sized Multimodal AI for content understanding and generation across multilingual texts, images, and 🔜 video, up to 5x faster than OpenAI CLIP and LLaVA 🖼️ & 🖋️

Fine-tune the Whisper speech recognition model to support training without timestamp data, training with timestamp data, and training without speech data. Accelerate inference and support Web deployment, Windows desktop deployment, and Android deployment

Neural Network Compression Framework for enhanced OpenVINO™ inference

LLM KV cache compression made easy