AXLearn: Modular Large Model Training on Heterogeneous Infrastructure