PagedWeight cuts GPU memory for expert language models by up to 72%
A runtime method compresses expert-model weights while keeping FP16-level accuracy, easing memory-bound serving. Also: robot-frame inputs that handle moving cameras, and Ray 2.56.1 fixes for data conversions.
The business desk's take → Bezos backs CuspAI as it launches an AI materials foundry for chips30-second briefing
- PagedWeight introduces runtime dynamic quantization for MoE serving.
- Enables FP16-level quality with major memory savings.
- Watch integrations into MoE stacks and reproducible benchmarks.
One-Line Summary
Serving-time precision and representation tweaks are squeezing more from today’s compute: dynamic weight quantization for expert models, robot-frame inputs that stabilize control across views, and a Ray update that smooths data pipelines.
Research Papers
PagedWeight compresses expert LLM weights on the fly for memory-bound serving
PagedWeight is a serving-side method that helps expert-based language models fit into limited memory by compressing weights at runtime and adapting to how much memory the key–value cache needs. Mixture-of-Experts (MoE) large language models (LLMs) route tokens to a small set of expert sub-networks for efficiency, but in long conversations the key–value (KV) cache can dominate memory and squeeze out model weights. PagedWeight dynamically quantizes expert weights at runtime and explicitly balances expert precision with KV cache size to keep quality, memory, and latency in check. 1
In tests across memory-sensitive MoE serving scenarios, the authors report FP16-equivalent accuracy with up to 72.0% graphics processing unit (GPU) memory savings and as much as 1.94× higher throughput. Compared with existing weight-quantization baselines under a similar memory budget, PagedWeight improves output quality by up to 39.3% while incurring at most a 4.1% throughput loss. 1
The system acts a bit like operating-system paging for precision: when the KV cache grows, it lowers precision for some experts; when memory pressure eases, it raises precision again. For operators running long-context MoE assistants, this exposes a tunable curve between answer quality and serving cost rather than a fixed tradeoff; the paper details how different settings shift that curve under varied workloads. 1
Robot-centric pointmaps help models follow instructions across camera views
This paper gives instruction-following robots a picture of the scene in the robot’s own coordinate frame so actions line up with what the camera sees even when the viewpoint changes. Vision-Language-Action (VLA) models observe images in the camera frame but must execute in the robot frame, which creates a mismatch that grows as datasets aggregate demonstrations from many camera setups. 2
The method builds images whose pixels store 3D coordinates of scene points in the robot frame, preserving the usual H×W grid so pre-trained 2D VLAs can use them with minimal architectural change. On RoboCasa, pointmaps improve both pi0.5 and SmolVLA and outperform representative camera-viewpoint and 3D-aware baselines; on a real robot, their advantage over an RGB-only policy widens when the camera is moved to an unseen placement. 2
Open Source & Repos
Ray 2.56.1 fixes data conversion regressions and stability issues
Ray is a distributed compute engine for AI workloads; the 2.56.1 release highlights multiple Ray Data fixes around converting datasets to pandas, including an opt-out flag (RAY_DATA_ENABLE_ARROW_BACKED_PANDAS_CONVERSION) for Arrow-backed conversion, a concat crash from int64/double[pyarrow] overflow, and a TensorDtype.from_arrow crash. 3
For teams building training and inference pipelines atop Ray Data, these changes reduce failure risk when handing off to pandas and give a clear escape hatch if Arrow-backed behavior breaks downstream code. If you upgraded to 2.56 and hit regressions, 2.56.1 provides targeted stability without rolling back the series. 3
Why It Matters
Serving-time precision scheduling for expert LLMs (PagedWeight) and robot-frame inputs that keep control robust across views both target the boundary between model capability and system constraints—memory pressure and frame mismatch—so you can keep quality while using the same hardware and data. 1
On the infrastructure side, surgical fixes in core libraries like Ray prevent pipeline stalls and rollbacks, which often yields more real throughput than adding hardware when teams are bottlenecked on data preparation and conversion. 3
Terms in This Article
- Quantization
Quantization is a neural‑network compression‑and‑acceleration technique that represents weights and activations with low‑bit integers by selecting scale and offset from calibrated value ranges and mapping values onto a uniform grid with rounding, reducing storage and memory traffic while leveraging low‑precision hardware for higher throughput.
- MoE
Mixture of Experts (MoE) is a sparse conditional-computation architecture in which a gating/router network selects a small subset of expert feed-forward subnets for each input token—typically replacing the Transformer’s FFN—so total parameter capacity can grow via many experts while per-token compute remains bounded.
- KV Cache
A KV cache is the inference-time memory structure that stores previously computed attention key/value tensors in an autoregressive Transformer. It avoids recomputing the same prefix during decoding, trading additional memory pressure for lower first-token latency and higher serving throughput.
Two desks on this story
PagedWeight cuts GPU memory for expert language models by up to 72%
Bezos backs CuspAI as it launches an AI materials foundry for chips
CuspAI raises $450 million at a $2.6 billion valuation and launches an AI Materials Foundry with 45+ partners to accelerate materials discovery for chips and energy. The coalition signals a push to cut rare-metal dependencies and speed lab-to-fab pipelines with shared compute and expertise.
Comments (0)