MiaAI-Lab/Qwen3.8-Flash-Next-Dual-DGX-Sparks
Qwen3.8-Flash-Next-NVFP4-vLLM-Dual-DGX-Spark
What it solves
This project provides a specialized deployment framework for running the Qwen3.8-Flash-Next-NVFP4 large language model across two NVIDIA DGX Spark nodes. It solves the complexity of multi-node inference by automating weight distribution, patching model kernels for compatibility, and managing distributed memory resources to maximize KV cache capacity.
How it works
It uses Docker containers to orchestrate a vLLM-based inference server across two nodes connected via high-speed RoCE/IB networking. The project includes scripts to download model weights, synchronize them between the head and worker nodes (either via rsync or a shared NFS volume), and apply specific patches to the model's layers and kernels to ensure they run correctly on the Spark hardware. It also manages memory by evicting page caches to prevent out-of-memory errors and supports advanced configurations like FP8 KV caching to significantly increase the number of tokens the model can handle in its context window.
Who it’s for
This project is for infrastructure engineers and AI researchers who have access to multi-node DGX Spark hardware and need to deploy high-performance, large-context Qwen models. It is intended for users comfortable with Linux, Docker, and distributed systems who require a turn-key solution for multi-node inference.
Highlights
- Multi-node Optimization: Specifically tuned for dual-DGX Spark setups using Tensor Parallelism (TP2), Expert Parallelism (EP), and Multi-Token Prediction (MTP3).
- High-Capacity KV Cache: Enables up to 3.65 million tokens of KV cache capacity using FP8 quantization.
- Automated Patching: Includes automatic patching for FlashInfer kernels and model layers to ensure compatibility with the NVFP4 architecture.
- Flexible Deployment: Supports multiple checkpoint types, including stock NVFP4, official FP8, and abliterated checkpoints, with options for NFS-based or rsync-based weight distribution.
Related
- Project
- Project
- Project
- Dispatch
- Project