Official Pytorch Implementation of PhyMPGN: Physics-encoded Message Passing Graph Network for spatiotemporal PDE systems

Wave propagation modules for PyTorch.

A machine learning compiler for GPUs, CPUs, and ML accelerators

Official PyTorch implementation of PSE/PSRN: Fast and efficient symbolic expression discovery through parallelized symbolic enumeration. Evaluates millions of expressions simultaneously on GPU with…