ChenQian9104 / AM_sim_transformer

Fast additive manufacturing thermal simulation based on transformers (100x faster than finite element analysis)

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