Compute in Transit — Presentation 01

Compute in Transit: the Idea and the Budget

What it means to compute on data while it crosses a link, what a photonic link module (arXiv:2608.21536) is reported to do, and the arithmetic that decides what fits: operations per byte at line rate, whether the transform is charged, precision, energy and placement. Five tests, applied to the data movements of LLM training and inference.

Compute in transit Operations per byte Photonic link module Published operation set Placement Five tests
Bytes that must move → Budget: ops / byte → Published operations → Precision → Placement → Candidates
00

Topics We'll Cover

Concepts used here, and where they are explained. Every concept is in the series glossary; concepts the sister series already explain link to their glossaries.

01

Data Movement Is the Bottleneck

LLM work moves large, regular blocks of data between devices. Two of them set the scale for this series: the gradient or pseudo-gradient sync in training, and the KV-cache hand-off in disaggregated inference.

Training: one model-sized update

  • Llama-3-70B, BF16: 141.10 GB per update.
  • A ring all-reduce sends 246.93 GB from each of 8 workers, 277.80 GB from each of 64.
  • Moving it once takes 313.6 ms over NVLink 4, 2,822.1 ms over InfiniBand NDR and 11,288.4 ms over 100 GbE.

Inference: one prompt's KV hand-off

  • Llama-3-8B, 2,048 tokens: 268.4 MB (131,072 bytes per token).
  • 0.60 ms over NVLink 4, 5.38 ms over IB NDR, 21.49 ms over 100 GbE, 85.92 ms over 25 GbE.
  • A 32,768-token Llama-3-70B prompt hands off 10,737.4 MB: 859.01 ms over 100 GbE.
MovementSizeNVLink 4IB NDR 400G100 GbE25 GbE
Llama-3-70B BF16 update141.10 GB313.6 ms2,822.1 ms11,288.4 ms45,153.5 ms
Llama-3-8B BF16 update16.06 GB35.7 ms321.2 ms1,284.8 ms5,139.2 ms
Llama-3-70B KV, 2,048 tokens671.1 MB1.50 ms13.43 ms53.71 ms214.77 ms
Llama-3-8B KV, 2,048 tokens268.4 MB0.60 ms5.38 ms21.49 ms85.92 ms

Source: results.md, sections 3 and 4. Link rates and model shapes come from Disaggregated_Inference_Sim's presets.

02

What Compute in Transit Means

Compute in transit is computation applied in the data path, on data that must move anyway. The data is not fetched to a processor and sent back; it is changed on the way past. Where the computation sits decides what it can see.

GPU (sender) NIC pluggable QSFP module fibre: one flow switch many flows meet here other senders other senders GPU (receiver) NIC 1 2 3 1 endpointits own data, any operation 2 linkone flow 3 switchmany flows, can combine
PlacementSeesCan doCannot do
Endpoint (GPU, NIC)Its own data, before it leaves or after it landsAnything the device's instruction set allows, at the cost of its own time and memory trafficCombine with data it has not received; it is not “free”, since it competes with the device's main work
Link (pluggable transceiver module)One flow, at line ratePer-stream transforms: compress, re-encode, transform, score against a held vectorCombine two streams: a sum of gradients from two workers needs both at one point
SwitchMany flowsCombine flows: in-network aggregation (the sum in an all-reduce)Much per-byte work: the same compute budget is shared by every port

A photonic interconnect that only moves bytes is not compute in transit. The distinction in this series is whether the device changes the data (glossary).

03

The Photonic Link Module

The series starts from one published device. Kundu, Hammond, Todd, Goel, Simpson, Rajendra, Michel, Crawford, Bergamaschi, R. Todd and New describe it in “Programmable Compute-in-Transit using Integrated Photonics” (arXiv:2608.21536, submitted 21 August 2026).

What the paper reports

  • A QSFP56 pluggable module with an integrated optical Fourier transform device.
  • It lists radix-4 FFT and NTT, addition, subtraction, multiplication, MAC, floating-point MAC and modular arithmetic. The modular logic runs in the driver IP on the host FPGA, an AMD Alveo V80 attached through the QSFP56 port.
  • “Tens of GFLOPs of operations on data links”: the authors' claim.
  • Evaluated inside a CKKS bootstrapping flow: ring dimension 65,536, 32-bit words, about 420 GB streamed through the photonics; functional correctness checked against a CPU implementation. The 65,536-point NTTs were split into 256-point transforms, then spread over four optical channels by radix-4 decomposition.
  • The photonic devices run at 14 GBaud, with NRZ data into the module; module power is 9.7 W. No throughput, latency or energy per operation is measured.

What the developer says

  • Compute-in-Transit is “operations performed with light directly in the data path”, and “AI infrastructure has hit the data movement wall” (the developer’s website).
  • Estimated 2.76 pJ/bit link energy on 45 nm silicon photonics, excluding the laser; “Native 384 GOPS / lambda @ 32GSps (GF45nm)” (the developer’s website).
  • Evaluation kit: photonic compute module, AMD Alveo V80 accelerator, QSFP data path and control software.
  • Stated targets: distributed and hyperscale AI, optical interconnect, secure and confidential AI; AI, security, HPC and telecoms (the developer’s website). The paper adds post-quantum cryptography and FHE.
How much weight these figures carry

A third-party review notes that the GFLOPs figure is asserted rather than measured (pith.science). The GOPS and pJ/bit figures are company estimates. This deck treats all three as claims and brackets “tens of GFLOPs” with 10 and 99 GFLOP/s, which is our reading, not a measurement.

Note: the paper facts on this slide were checked against its full text (four pages) on 9 October 2026.

04

The Published Operation Set, and What Is Missing

The paper lists radix-4 FFT and NTT, addition and subtraction, multiplication, MAC, floating-point MAC and modular arithmetic (add, subtract, multiply, MAC, MSUB). Everything a candidate kernel does must be built from them, or it cannot run in the module.

OperationListed?What it gives an LLM kernel
FFT (1×N, radix-4)yesFrequency-domain compression along a sequence or within a chunk
NTT (1×N, radix-4)yesExact transforms over integers mod q: FHE, not float traffic
Modular arithmeticyesInteger rescaling and encoding (runs in the host FPGA’s driver IP)
Addition, subtraction, multiplicationyesSumming streams, scaling, element-wise products
MAC, floating-point MACyesDot products against a held vector, weighted sums
Comparisonnot listedNeeded for absmax scaling, thresholds, max
Rounding to a narrower formatnot listedNeeded to requantise BF16 to FP8
Top-k selectionnot listedNeeded by DeMo-style sparsification and KV block selection
Data-dependent controlnot listedNeeded for any “if this value, then” logic

“Not listed” means only that the public sources do not mention it; the module or its host FPGA may do more.

Consequence

Kernels must be data-independent: the same arithmetic on every value, whatever its size. Keeping a fixed frequency band replaces top-k. On synthetic 64×64 chunks, keeping 1/16 of the DCT coefficients, the fixed band keeps 0.957 of a smooth signal's energy against top-k's 0.995, but only 0.062 of white noise against top-k's 0.325 (results.md, section 5, illustrative).

05

How the Optics Computes

A lens performs a Fourier transform on a coherent light field as it propagates, and two lenses with a mask between them compute a convolution (the 4f system). The transform has no operation count, only the cost of getting data into and out of the light. The background is in Fourier Optics for Inference 01: the lens as a Fourier transformer and the 4f system.

Electrical bits in → Modulators → Passive diffractive OFT device
4 optical channels
→ Detectors → Digital recombination
larger, higher precision

So the native optical transform is small, and long transforms are built by recombination. How much of that recombination is digital, and whether it counts against the module's “tens of GFLOPs”, is not public. That question decides the budget (slide 07).

06

Interactive: the Budget in Operations per Byte

A module that applies C operations per second to a stream arriving at R bytes per second can afford C / R operations per byte. Anything more and the link slows to the module's pace.

LinkLine rate (GB/s)ClaimOperations per byte
QSFP56 200G25low (10 GFLOP/s)0.40
QSFP56 200G25high (99 GFLOP/s)3.96
400G50low (10 GFLOP/s)0.20
400G50high (99 GFLOP/s)1.98
800G100low (10 GFLOP/s)0.10
800G100high (99 GFLOP/s)0.99
Prototype optics (4 × 14 Gb/s NRZ)7low (10 GFLOP/s)1.43
Prototype optics (4 × 14 Gb/s NRZ)7high (99 GFLOP/s)14.14

Source: results.md, section 1. 400G and 800G rows assume the same engine per link. The prototype-optics rate is our arithmetic from the paper, not a figure it states.

Port rate is not the optics' rate

25 GB/s is the QSFP56 port's rate. In the prototype the port connects the module to its host FPGA, and the photonic devices run at 14 GBaud, with NRZ data into four modulators (Kundu et al., section II): about 7 GB/s, 28% of the port, by our arithmetic. At 7 GB/s the same claim gives 1.43 to 14.14 operations per byte, enough for a charged BF16 FFT of length 2,048 at the high end. That is more work per byte, but on a link the prototype could not fill: a module limited to its optics' rate would itself throttle a 200 Gb/s link. 25 GB/s is the rate a product would have to match.

Move the claim and the line rate below. Work per value uses the same conventions as cit_budget.py: FFT 2.5 log2 n per value; 2-D DCT of c×c chunks 2 × 2.5 log2 c; sum 1; dot product 2; FP8 requantisation about 2 (illustrative); top-k about 2 comparisons (illustrative).

99
Budget, ops per byte
—
Budget, ops per value
—
Longest FFT that fits
—
Kernels that fit
—
KernelTransform / byteOther / byteCounted / byteIn setFits

The same ledger, in Python, is analysis/cit_budget.py. At 99 GFLOP/s on 25 GB/s with BF16 and the transform charged, the page reproduces results.md section 2: the longest FFT that fits is length 8.

07

Is the Transform Free?

There are two ways to account for the optical transform, and the answer to the whole question changes with the choice.

Charged

The “tens of GFLOPs” include the transform's FLOPs (or the digital recombination that builds long transforms from short ones). Every FFT costs 2.5 log2 n per value. At the low end no transform fits: even length 2 costs 1.25 per BF16 byte against a budget of 0.40. At the high end, 3.96 per byte, the longest BF16 FFT that fits is length 8.

Free

The optics performs the transform by propagation, and the GFLOPs figure counts only the other arithmetic. Then every transform-based kernel fits, and the remaining arithmetic decides. A fixed-band selection is a gather with no arithmetic, so it fits even at the low end.

KernelTransform / byteOther / byteIn setLow, chargedHigh, chargedLow, free
Sum two gradient streams0.000.50yesnoyesno
1-D FFT, length 8, fixed band3.750.00yesnoyesyes
1-D FFT, length 64, fixed band7.500.00yesnonoyes
1-D FFT, length 1024, fixed band12.500.00yesnonoyes
2-D DCT 16×16, fixed band10.000.00yesnonoyes
2-D DCT 64×64, fixed band15.000.00yesnonoyes
2-D DCT 64×64 + top-k (DeMo)15.001.00nononono
Requantise BF16 to FP80.001.00nonoyesno
Score a KV key (dot product, head dim 128)0.001.00yesnoyesno

Source: results.md, section 2. Low = 0.40, high = 3.96 operations per byte (QSFP56). “Fits” is arithmetic only; “in set” is a separate test.

08

Precision: Noise or Exactness

An optical transform is analogue. Its result carries noise from the light, the detectors and the converters. Whether that matters depends on the data.

Float traffic: a noise level

  • Gradients, activations and KV are floating-point. They already carry their format's rounding error, so the analogue chain need only match it, not be exact.
  • The rule from FOptInf 01, slide 17: relative RMS error = crest × 2−ENOB × 2/√12. Matching BF16 needs about 10.3 ENOB, INT8 8.0, FP8 E4M3 6.3 (FOptInf results.md, section 6).
  • Compressed traffic is lossy anyway. Analogue noise adds to the compression error; it does not need to vanish.

Cryptography: exact results

  • FHE and post-quantum schemes work in integers mod q. A single wrong digit corrupts the ciphertext.
  • The module reaches exactness by decomposition: small, low-precision optical transforms recombined digitally into larger, higher-precision ones (the patents on slide 05).
  • The CKKS bootstrapping evaluation checked functional correctness against a CPU implementation (slide 03).
Why this matters for LLM traffic

Decomposition buys exactness with extra passes and digital work. Float traffic does not need it, so an LLM kernel could, in principle, run the optics at its native precision. The developer claims “Scalable accuracy e.g. FP4->FP64” and the paper projects FP4 MAC on a future link, but no ENOB or error figure is published; the driver IP includes a proprietary error-correction block. The FHE accelerator series works through the exact case (FHESim 04).

09

Energy per Bit

The module’s developer estimates 2.76 pJ/bit link energy on 45 nm silicon photonics, excluding the laser (the developer’s website). For scale, the link energy coefficients used in Brendan's Disaggregated_Inference_Sim:

LinkpJ/bitStatus
Photonic compute-in-transit link2.76Company estimate, 45 nm, excluding laser
Co-packaged optics3Illustrative simulator coefficient
NVLink 45Illustrative simulator coefficient
InfiniBand NDR 400G15Illustrative simulator coefficient
100 GbE15Illustrative simulator coefficient

Simulator presets: disagg_sim/hardware.py, LINKS. They are first-order coefficients for the simulator's energy model, not measurements.

10

Five Tests Any Candidate Must Pass

#TestQuestionWhere it is decided
1BottleneckDoes this traffic bound the system's time or throughput?Simulation of the whole system, not the link alone
2OperationsIs the kernel built only from the listed operations (FFT, NTT, add, multiply, MAC, modular arithmetic)?Slide 04
3BudgetDoes its work per byte fit C / R, with the transform charged or free?Slides 06 and 07
4PrecisionDoes it tolerate analogue noise, or can it be made exact?Slide 08
5PlacementDoes it need one flow (a pluggable) or many (a switch)?Slide 02
Test 1 in Brendan's simulator

Disaggregated_Inference_Sim compresses the KV hand-off either in transit or on the prefill GPU (FOptInf 03, slide 13). Compression helped only where the link bound the system: GQA KV over 25 GbE at 14 req/s, FP8 cut the hand-off p99 from 9,657.8 ms to 166.6 ms and lifted SLO attainment from 41.0% to 100.0%. The GPU did the same compression for the same 166.6 ms (simulator results.md, section 15; transit stage parameters illustrative).

11

Candidates at a Glance

Each data movement placed on two of the tests. The placement is a judgement from this deck's arithmetic, not a measurement; decks 02 and 03 test it.

Needs a switch Strong Weak Fits, little to gain Cross-data-centre gradient sync KV-cache tiering Query-aware KV selection (scoring) KV hand-off on slow links (boundary) In-cluster all-reduce sum of two streams: many flows FHE and encrypted traffic Ring-attention KV blocks (overlapped) Pipeline / tensor-parallel activations Weight streaming fits the published operations: low → high traffic bounds the system: low → high FHE and encrypted traffic: a strong fit for the operations; little to gain for LLM latency.
12

Open Questions for the Module's Developer

Four answers would settle most of this deck's uncertainty. Each changes a verdict, not a detail.

QuestionWhy it mattersIf yesIf no
Is the optical transform charged against the “tens of GFLOPs”?Decides whether any transform-based kernel fitsOnly FFTs of length 8 or less fit at 3.96 ops/byteEvery fixed-band kernel fits; the remaining arithmetic decides
Can the module compare and round?FP8 requantisation, absmax scaling and top-k need themRequantisation joins the candidatesKernels stay data-independent: fixed bands, fixed scales
What is the native transform size and precision?Sets the recombination work and the noise floor (slide 05, 08)A large native size with float-level ENOB suits LLM trafficA small native transform at low ENOB pushes work into digital recombination
Is LLM traffic a target?The stated targets include distributed and hyperscale AI; the published evaluation is FHELLM kernels may get dedicated supportThis series is a feasibility study on a device built for cryptography

A measured throughput figure, in place of the asserted one, would also replace this deck's 10 to 99 GFLOP/s bracket.

13

What to Take Away