[Jiminator]

Motivation

Split #44747 into reviewable parts. This is the Beam CUDA Docker image part, position 1 of 8 in the merge order below. Its branch starts from #44747's merge base with main. No other part depends on it or includes it.

Modifications

Adds docker/beam/beam.Dockerfile, a Beam serving image on top of lmsysorg/sglang:v0.5.22, following docker/muse_spark/muse-spark.Dockerfile. It clones the upstream beam branch (pin a commit with BEAM_REF) and installs it without dependencies. It then adds the two pins that branch needs on top of the release: OpenAI 2.8.0 for the Responses shell types and nvidia-cuda-tileiras 13.4.92 for the cuTile router GEMM. The Dynamo and grpcio-tools pins moved to #45292.

docker/Dockerfile is not changed by any part. This image is a stopgap: delete it once Beam runs from the main image.

This part changes 1 file (+31 / -0) beyond the parts its branch carries.

Validation

• Cut from [Feat] beam: Add native Beam serving support #44747's head (f7473673dd) without modifying the beam branch, which stays the reference. Every branch starts from [Feat] beam: Add native Beam serving support #44747's merge base with main (5bcaf0f8ed, Oct 13); the parts have not yet been synced with current main.
• Conflicts when merged into main 28448a86d4: none.
• Coverage: [Chore] docker: Add Beam CUDA image (1/8) #45253, [Perf] beam: Add optional cuTile BF16x3 router GEMM #45361 and [Feat] parser: Add Beam reasoning, tool calls and response templates (5/7) #45257 merged, plus the one-line [Chore] deps: Add nvidia-cuda-tileiras for the optional cuTile router #45290 and [Chore] deps: Bump the OpenAI SDK to 2.8.0 for Responses shell tools (2/8) #45291 changes, give exactly the [Feat] beam: Add native Beam serving support #44747 tree, except that streamed LoRA (25 files from sglang-miles) and Beam LoRA serving (10 files plus the LoRA hunks in 3 files) are left out, the five platform pyprojects keep OpenAI 2.6.1, and docker/beam/beam.Dockerfile is added. No other path differs.
• No part except [Chore] deps: Add nvidia-cuda-tileiras for the optional cuTile router #45290 and [Chore] deps: Bump the OpenAI SDK to 2.8.0 for Responses shell tools (2/8) #45291 changes a pyproject, and none changes docker/Dockerfile. environ.py is split by hunk. Every changed Python file in this part parses.
• Not run for this split: pre-commit, registered-test registry parsing and the CPU and GPU suites. docker buildx build --check passes. The image builds on arm64 and amd64, and its exact environment serves Beam on GB300 and H200. Intermediate parts make no claim of standalone Beam serving validation; the recorded accuracy and performance results are in [Feat] beam: Add native Beam serving support #44747.

Merge order

Every part targets main. #44747 stays open, unchanged, as the full reference. Merge in this order:

1. [Chore] docker: Add Beam CUDA image (1/8) #45253 — Beam CUDA Docker image
2. [Chore] deps: Bump the OpenAI SDK to 2.8.0 for Responses shell tools (2/8) #45291 — OpenAI SDK 2.8.0, needed by the Responses shell tools in [Feat] parser: Add Beam reasoning, tool calls and response templates (5/7) #45257
3. [Feat] kernels: Beam model support #45254 — Kernels
4. [Feat] srt: Add attention, memory and DFlash DP support for Beam (4/8) #45255 — Attention, memory and DFlash DP support
5. [Feat] beam: Add Beam model, runtime support and checkpoints (4/7) #45256 — Beam model, checkpoints and tests
6. [Feat] parser: Add Beam reasoning, tool calls and response templates (5/7) #45257 — Beam reasoning, tool calls and response templates (also after [Feat] parser: Enrich chat_parsing streaming events #40478, [Fix] parser: Accept whole-number floats for integer tool arguments #43941 and [Feat] parser: Add checkpoint-driven response-template adapters #40479)
7. [Chore] deps: Add nvidia-cuda-tileiras for the optional cuTile router #45290 — nvidia-cuda-tileiras, needed by [Perf] beam: Add optional cuTile BF16x3 router GEMM #45361
8. [Perf] beam: Add optional cuTile BF16x3 router GEMM #45361 — cuTile BF16x3 router GEMM (optional: without it Beam uses the cuBLAS router)

#45292 (Dynamo and grpcio-tools pins for the Beam image) is not in the merge order for now.

🤖 Generated with Claude Code