AMR-Transformer: Enabling Efficient Long-range Interaction for Complex Neural Fluid Simulation

Figure 1. AMR tokenizer results in high-resolution simulation. Visualization of the AMR tokenizer applied to 1024 × 1024 shock wave and explosion simulations. The AMR tokenizer captures fine-scale structures while reducing token count. Right panels compare the regular 512 × 512 grid with the AMR partitioning, each right panel displays the total cell count and partitioning scheme (bottom-right), along with the mean squared error (MSE) relative to the 1024 × 1024 ground truth (top-left).