Advanced Guides# Advanced Scale up on multiple devices Performance considerations Use Flax NNX and Linen together Model surgery