This document is relevant for: Trn2, Trn3

Framework

Topic	Description
Async execution	Async execution double buffering
Checkpoint loading	Multi-process weight loading from SafeTensors
Model factory	Model registry and factory pattern
Per-graph compiler arguments	Per-graph neuronx-cc arguments through a model method

This document is relevant for: Trn2, Trn3