Skip to content

Production-tested AI infrastructure tools for efficient AGI development and community-driven innovation

License

Notifications You must be signed in to change notification settings

deepseek-ai/open-infra-index

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 

Repository files navigation

DeepSeek-Open-Infra

Hello, DeepSeek Open Infra!

202502 Open-Source Week

We're a tiny team @deepseek-ai pushing our limits in AGI exploration.

Starting this week , Feb 24, 2025 we'll open-source 5 repos – one daily drop – not because we've made grand claims, but simply as developers sharing our small-but-sincere progress with full transparency.

These are humble building blocks of our online service: documented, deployed, and battle-tested in production. No vaporware, just sincere code that moved our tiny yet ambitious dream forward.

Why? Because every line shared becomes collective momentum that accelerates the journey. Daily unlocks begin soon. No ivory towers - just pure garage-energy and community-driven innovation πŸ”§

Stay tuned – let's geek out in the open together.

Day 1 - FlashMLA

Efficient MLA Decoding Kernel for Hopper GPUs
Optimized for variable-length sequences, battle-tested in production

πŸ”— FlashMLA GitHub Repo
βœ… BF16 support
βœ… Paged KV cache (block size 64)
⚑ Performance: 3000 GB/s memory-bound | BF16 580 TFLOPS compute-bound on H800

Day 2 - DeepEP

Excited to introduce DeepEP - the first open-source EP communication library for MoE model training and inference.

πŸ”— DeepEP GitHub Repo
βœ… Efficient and optimized all-to-all communication
βœ… Both intranode and internode support with NVLink and RDMA
βœ… High-throughput kernels for training and inference prefilling
βœ… Low-latency kernels for inference decoding
βœ… Native FP8 dispatch support
βœ… Flexible GPU resource control for computation-communication overlapping

Day 3 - DeepGEMM

Introducing DeepGEMM - an FP8 GEMM library that supports both dense and MoE GEMMs, powering V3/R1 training and inference.

πŸ”— DeepGEMM GitHub Repo
⚑ Up to 1350+ FP8 TFLOPS on Hopper GPUs
βœ… No heavy dependency, as clean as a tutorial
βœ… Fully Just-In-Time compiled
βœ… Core logic at ~300 lines - yet outperforms expert-tuned kernels across most matrix sizes
βœ… Supports dense layout and two MoE layouts

Day 4 - Optimized Parallelism Strategies

βœ… DualPipe - a bidirectional pipeline parallelism algorithm for computation-communication overlap in V3/R1 training.
πŸ”— GitHub Repo

βœ… EPLB - an expert-parallel load balancer for V3/R1.
πŸ”— GitHub Repo

πŸ“Š Analyze computation-communication overlap in V3/R1.
πŸ”— GitHub Repo

Day 5 - 3FS, Thruster for All DeepSeek Data Access

Fire-Flyer File System (3FS) - a parallel file system that utilizes the full bandwidth of modern SSDs and RDMA networks.

⚑ 6.6 TiB/s aggregate read throughput in a 180-node cluster
⚑ 3.66 TiB/min throughput on GraySort benchmark in a 25-node cluster
⚑ 40+ GiB/s peak throughput per client node for KVCache lookup
🧬 Disaggregated architecture with strong consistency semantics
βœ… Training data preprocessing, dataset loading, checkpoint saving/reloading, embedding vector search & KVCache lookups for inference in V3/R1

πŸ“₯ 3FS β†’ πŸ”—GitHub Repo
β›² Smallpond - data processing framework on 3FS β†’ πŸ”—GitHub Repo

Day 6 - One More Thing: DeepSeek-V3/R1 Inference System Overview

Optimized throughput and latency via:
πŸ”§ Cross-node EP-powered batch scaling
πŸ”„ Computation-communication overlap
βš–οΈ Load balancing

Production data of V3/R1 online services:
⚑ 73.7k/14.8k input/output tokens per second per H800 node
πŸš€ Cost profit margin 545%

Cost And Theoretical Income.jpg

πŸ’‘ We hope this week's insights offer value to the community and contribute to our shared AGI goals.

πŸ“– Deep Dive: πŸ”—Day 6 - One More Thing: DeepSeek-V3/R1 Inference System Overview
πŸ“– δΈ­ζ–‡η‰ˆ: πŸ”—DeepSeek-V3 / R1 ζŽ¨η†η³»η»Ÿζ¦‚θ§ˆ

2024 AI Infrastructure Paper (SC24)

Fire-Flyer AI-HPC: A Cost-Effective Software-Hardware Co-Design for Deep Learning

πŸ“„ Paper Link
πŸ“„ Arxiv Paper Link

About

Production-tested AI infrastructure tools for efficient AGI development and community-driven innovation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published