Skip to content

feat: add DGX Spark adaptation for concurrent Gemma 4 agents (Fixes #399)#405

Open
Paramveersingh-S wants to merge 1 commit into
google-gemma:mainfrom
Paramveersingh-S:feat/spark-concurrent-dgx
Open

feat: add DGX Spark adaptation for concurrent Gemma 4 agents (Fixes #399)#405
Paramveersingh-S wants to merge 1 commit into
google-gemma:mainfrom
Paramveersingh-S:feat/spark-concurrent-dgx

Conversation

@Paramveersingh-S

Copy link
Copy Markdown

Description

This PR addresses Issue #399 by providing a guide and implementation for running the apps/concurrent multi-agent orchestrator on an NVIDIA DGX Spark cluster.

Key Changes

  • Created apps/concurrent_spark directory.
  • Added spark_orchestrator.py which demonstrates utilizing PySpark mapInPandas to distribute the specialist agent task payloads across the DGX Spark Executors.
  • Added an architectural README.md explaining how to map the agent architecture to a high-speed CX-7 ring topology via distributed LLM serving (e.g. vLLM binding to NCCL/NVLink on the worker nodes).

Fixes #399.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Gemma 4 Concurrent on DGX Spark cluster

1 participant