Skip to main content
Loading navigation

Pydantic AI & Baseten Distributed Inference Interconnect Simulator

Network interconnect bottlenecks between host nodes cause high pipeline latency in multi-node tensor parallel clusters. Build a Pydantic AI agent system deployed with Baseten runtime tools to simulate NVLink, NVSwitch, and InfiniBand cluster topology, inject network fault scenarios, and calculate tail latency recovery times.

EngineeringHosted by Vera

What you are building

Develop a distributed inference interconnect simulator using Pydantic AI and Baseten to analyze network bottlenecks and tail latency under node faults.

How work is evaluated

Evaluation

Evaluates distributed interconnect simulator accuracy during hardware link fault injection tests.

Datasets

Public datasets, plus the private datasets you have access to.

How submissions are scored

Dimensions
2 scoring checks
Binary
2 pass or fail dimensions
Ordinal
0 scaled dimensions
Dimension 1

latency_degradation_check

Checks that degraded latency reflects simulated bandwidth constraints.

Binary check

This dimension contributes its full weight only when the submission satisfies the requirement. Partial credit is not awarded.

Dimension 2

simulation_accuracy

Correlation with physical NCCL All-Reduce trace benchmarks • target: 0.95 • range: 0.88-1

Binary check

This dimension contributes its full weight only when the submission satisfies the requirement. Partial credit is not awarded.

Learning goals

  • Define type-safe interconnect models using Pydantic AI models and dependency injection

  • Deploy inference runtime tools on Baseten to profile pipeline and tensor parallel communication

  • Simulate NVLink/NVSwitch 900GB/s and InfiniBand 400Gbps network links

  • Inject packet drop and cable failure events to verify multi-node fault recovery speed

Resources

Dataset notes

NCCL collective benchmark performance logs across varied cluster topologies.

Challenge setup

Evaluates distributed interconnect simulator accuracy during hardware link fault injection tests.

Challenge input

JSON containing cluster_size, topology, failure_event, model_tp_size

Pydantic AI

Structured Python agent framework with strict validation

Baseten

Model serving and system execution runtime

Evaluated output

JSON containing baseline_latency_ms, degraded_latency_ms, recovery_time_ms, throughput_degradation_pct

Evaluation checks
  • Checks that degraded latency reflects simulated bandwidth constraints.
  • Correlation with physical NCCL All-Reduce trace benchmarks • target: 0.95 • range: 0.88-1
Configured evaluation evidence
  • Benchmark: NCCL Performance Tests
  • Simulation Accuracy target: 0.95
  • 1 public reference case
Configured runtime
  • Python execution harness
  • Python sandbox (unavailable on Versalist)

Shows the first task, up to three tools, and up to three evidence items.

View technical recipe

Configured tools

Action Space
  • Pydantic AI · Required
  • Baseten · Optional
  • Anthropic · Optional
Orchestration
  • Pydantic AI · Required

Evaluation contract

  • latency_degradation_check · Weight 1
  • simulation_accuracy · Weight 1

Metrics

  • simulation_accuracy · Target 0.95

Benchmarks

  • NCCL Performance Tests
  • general

Reference cases

  • 1 public reference case

Workflow tasks

  1. Task 1

    Input: JSON containing cluster_size, topology, failure_event, model_tp_size

    Output: JSON containing baseline_latency_ms, degraded_latency_ms, recovery_time_ms, throughput_degradation_pct

    Execution harness: Python

Evaluation tools

  • Pydantic AI · Type-Safe Agent Framework
  • Baseten · Inference Runtime Host

Recipe freeze marker

No recipe freeze time is recorded.

Frequently Asked Questions about Pydantic AI & Baseten Distributed Inference Interconnect Simulator