AM simulation tool based on transformers (ViT & ViViT) Notes: The sliced geometry is encoded by spatil- and temporal- transformer A MLP is used to predict the temperature at (x, y, z, t) inspired by NeRF Usage: python3 multiGPUs.py