// RUN โ€” FITS

Can the AMD Instinct MI325X run NeMO Megatron GPT 20B?

Yes โ€” here's how fast and up to what context, with real measured numbers.

NeMO Megatron GPT 20B on the AMD Instinct MI325X

QuantFits?Max contextSpeedBeyond context
FP16 yes 740k 94.5-115.5 t/s ~0.6-0.8 (slow)
Q8 yes 816k 189-231 t/s ~1.3-1.5 (slow)
Q4 yes 850k 337.5-412.5 t/s ~2.3-2.8 (slow)

Estimate (memory-bound). Beyond "max context" the KV cache spills to system RAM offload (depends on your system RAM) and speed drops to the "beyond" figure.

FAQ

Can the AMD Instinct MI325X run NeMO Megatron GPT 20B?โ–ถ

Yes. At Q4 it generates ~337.5-412.5 tok/s and fits up to 850k context; beyond that it spills to system RAM offload (depends on your system RAM) and slows to ~2.3-2.8 tok/s.

How much context fits?โ–ถ

At Q4, up to about 850k tokens stay in fast memory; longer context spills to system RAM offload (depends on your system RAM) and throughput collapses.