TorchFEM is a compact finite element library based on PyTorch taking advantage of the auto-differentiation of PyTorch to speed up the traditional finite element pipeline, instead of train a black-box model to "solve" PDE. The code base is modified based on the popular MATLAB package iFEM by Dr. Long Chen.
scaomath/torch-afem
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|