Challenge

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
Challenge brief

What you are building

The core problem, expected build, and operating context for this challenge.

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

Delivery guide

How work is evaluated

Evaluation

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

Datasets

Shared data for this challenge

Review public datasets and any private uploads tied to your build.

Evaluation rubric

How submissions are scored

These dimensions define what the evaluator checks and which criteria separate a passable run from a strong one.

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

What you should walk away with

  • 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 and assets

Reference links and supporting material

Dataset notes

NCCL collective benchmark performance logs across varied cluster topologies.

How this agent runs

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

Checks for
  • Checks that degraded latency reflects simulated bandwidth constraints.
  • Correlation with physical NCCL All-Reduce trace benchmarks • target: 0.95 • range: 0.88-1
Proof of success
  • Benchmark: NCCL Performance Tests
  • Simulation Accuracy target: 0.95
  • 1 public reference case
Runtime evidence
  • Python execution harness
  • Python sandbox (unavailable on Versalist)
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

Recipe state

This is a preview. The configuration can change before the evaluation recipe is locked.

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